NOISSUE - Add mutex to WebSocket service mock (#294)

ws/api/transport_test.go

@@ -60,8 +60,7 @@ func handshake(tsURL string, chanID uint64, token string, addHeader bool) (*webs
 		header.Add("Authorization", token)
 	}
 	url := makeURL(tsURL, chanID, token, addHeader)
-	conn, resp, err := websocket.DefaultDialer.Dial(url, header)
+	return websocket.DefaultDialer.Dial(url, header)
-	return conn, resp, err
 }
 
 func TestHandshake(t *testing.T) {

ws/mocks/messages.go

@@ -1,6 +1,8 @@
 package mocks
 
 import (
+	"sync"
+
 	"github.com/mainflux/mainflux"
 	"github.com/mainflux/mainflux/ws"
 )
@@ -10,25 +12,30 @@ var _ ws.Service = (*mockService)(nil)
 type mockService struct {
 	subscriptions map[uint64]ws.Channel
 	pubError      error
+	mutex         sync.Mutex
 }
 
 // NewService returns mock message publisher.
 func NewService(subs map[uint64]ws.Channel, pubError error) ws.Service {
-	return mockService{subs, pubError}
+	return &mockService{subs, pubError, sync.Mutex{}}
 }
 
-func (svc mockService) Publish(msg mainflux.RawMessage) error {
+func (svc *mockService) Publish(msg mainflux.RawMessage) error {
 	if len(msg.Payload) == 0 {
 		return svc.pubError
 	}
+	svc.mutex.Lock()
 	svc.subscriptions[msg.Channel].Messages <- msg
+	svc.mutex.Unlock()
 	return nil
 }
 
-func (svc mockService) Subscribe(chanID uint64, channel ws.Channel) error {
+func (svc *mockService) Subscribe(chanID uint64, channel ws.Channel) error {
 	if _, ok := svc.subscriptions[chanID]; !ok {
 		return ws.ErrFailedSubscription
 	}
+	svc.mutex.Lock()
 	svc.subscriptions[chanID] = channel
+	svc.mutex.Unlock()
 	return nil
 }