robocar-base/testtools/testtools.go

package testtools

import (
	"github.com/cyrilix/robocar-base/mqttdevice"
	mqtt "github.com/eclipse/paho.mqtt.golang"
	"github.com/golang/protobuf/proto"
	log "github.com/sirupsen/logrus"
	"sync"
)


func NewFakePublisher() *FakePublisher {
	return &FakePublisher{msg: make(map[string]mqttdevice.MqttValue)}
}

type FakePublisher struct {
	muMsg sync.Mutex
	msg   map[string]mqttdevice.MqttValue
}

func (f *FakePublisher) Publish(topic string, payload mqttdevice.MqttValue) {
	f.muMsg.Lock()
	defer f.muMsg.Unlock()
	f.msg[topic] = payload
}

func (f *FakePublisher) PublishedEvent(topic string) mqttdevice.MqttValue {
	f.muMsg.Lock()
	defer f.muMsg.Unlock()
	return f.msg[topic]
}

type fakeMessage struct {
	qos     byte
	topic   string
	payload []byte
	acked   bool
}

func (f *fakeMessage) Duplicate() bool {
	return false
}

func (f *fakeMessage) Qos() byte {
	return f.qos
}

func (f *fakeMessage) Retained() bool {
	return false
}

func (f *fakeMessage) Topic() string {
	return f.topic
}

func (f *fakeMessage) MessageID() uint16 {
	return 1234
}

func (f *fakeMessage) Payload() []byte {
	return f.payload
}

func (f *fakeMessage) Ack() {
	f.acked = true
}

func NewFakeMessage(topic string, payload []byte) mqtt.Message {
	return &fakeMessage{
		qos:     0,
		topic:   topic,
		payload: payload,
		acked:   false,
	}
}

func NewFakeMessageFromProtobuf(topic string, msg proto.Message) mqtt.Message{
	payload, err := proto.Marshal(msg)
	if err  != nil {
		log.Errorf("unable to marshal protobuf message %T: %v", msg, err)
		return nil
	}
	return NewFakeMessage(topic, payload)
}