75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
// Copyright (c) Mainflux
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package mocks
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"github.com/mainflux/mainflux/pkg/messaging"
|
|
"github.com/mainflux/mainflux/ws"
|
|
)
|
|
|
|
var _ messaging.PubSub = (*mockPubSub)(nil)
|
|
|
|
type MockPubSub interface {
|
|
Publish(string, messaging.Message) error
|
|
Subscribe(string, string, messaging.MessageHandler) error
|
|
Unsubscribe(string, string) error
|
|
SetFail(bool)
|
|
SetConn(*websocket.Conn)
|
|
Close() error
|
|
}
|
|
|
|
type mockPubSub struct {
|
|
fail bool
|
|
conn *websocket.Conn
|
|
}
|
|
|
|
// NewPubSub returns mock message publisher-subscriber
|
|
func NewPubSub() MockPubSub {
|
|
return &mockPubSub{false, nil}
|
|
}
|
|
func (pubsub *mockPubSub) Publish(s string, msg messaging.Message) error {
|
|
if pubsub.conn != nil {
|
|
data, err := json.Marshal(msg)
|
|
if err != nil {
|
|
fmt.Println("can't marshall")
|
|
return ws.ErrFailedMessagePublish
|
|
}
|
|
return pubsub.conn.WriteMessage(websocket.BinaryMessage, data)
|
|
}
|
|
if pubsub.fail {
|
|
return ws.ErrFailedMessagePublish
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (pubsub *mockPubSub) Subscribe(string, string, messaging.MessageHandler) error {
|
|
if pubsub.fail {
|
|
return ws.ErrFailedSubscription
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (pubsub *mockPubSub) Unsubscribe(string, string) error {
|
|
if pubsub.fail {
|
|
return ws.ErrFailedUnsubscribe
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (pubsub *mockPubSub) SetFail(fail bool) {
|
|
pubsub.fail = fail
|
|
}
|
|
|
|
func (pubsub *mockPubSub) SetConn(c *websocket.Conn) {
|
|
pubsub.conn = c
|
|
}
|
|
|
|
func (pubsub *mockPubSub) Close() error {
|
|
return nil
|
|
}
|