server.go

// Package mesgtest is a testing package for MESG service.
// Use this package while unit testing your programs.
package mesgtest

import (
	"encoding/json"

	service "github.com/mesg-foundation/go-service/proto"
	uuid "github.com/satori/go.uuid"
)

// Server is a test server.
type Server struct {
	service *serviceServer
	socket  *Socket
}

// NewServer creates a new test server.
func NewServer() *Server {
	return &Server{
		service: newServiceServer(),
		socket:  newSocket(),
	}
}

// Start starts the test server.
func (s *Server) Start() error {
	return s.socket.listen(s.service)
}

// Socket returns a in-memory socket for client application.
func (s *Server) Socket() *Socket {
	return s.socket
}

// LastEmit returns the chan that receives last emitted event's info.
func (s *Server) LastEmit() chan *Event {
	return s.service.emitC
}

// Execute executes a task with data.
func (s *Server) Execute(task string, data interface{}) (string, *Execution, error) {
	bytes, err := json.Marshal(data)
	if err != nil {
		return "", nil, err
	}

	uuidV4, err := uuid.NewV4()
	if err != nil {
		return "", nil, err
	}
	id := uuidV4.String()

	taskData := &service.TaskData{
		ExecutionID: id,
		TaskKey:     task,
		InputData:   string(bytes),
	}

	select {
	case <-s.service.closingC:
		return "", nil, ErrConnectionClosed{}
	case s.service.taskC <- taskData:
	}

	select {
	case <-s.service.closingC:
		return "", nil, ErrConnectionClosed{}
	case resp := <-s.service.submitC:
		return id, resp, nil
	}
}

// ListenToken returns the token of service that started listening for tasks.
func (s *Server) ListenToken() string {
	return s.service.token
}

// Close closes test server.
func (s *Server) Close() error {
	return s.socket.close()
}

// ErrConnectionClosed returned when the connection closed between go-service and server.
type ErrConnectionClosed struct{}

func (e ErrConnectionClosed) Error() string {
	return "connection closed"
}