mirror of
https://github.com/planetmint/planetmint-go.git
synced 2025-03-30 15:08:28 +00:00
e99b614e4a
* added a bit of logging * added update messages to mqtt (every second with a timestamp) * added isConnectionOpen to the interface and analysis this during execution * added connection loss handler and status check * put some global objects into an objects to avoid inter service testing issues * TestingLocker Mutex to RWMutex (increase performance) * termintationMutex became a sync.RWMutex * added logging to the mqtt mock client * made monitor Mutex a RWMutex * added Mutex protection to the numberOfElements varialbe * added another Waiting block to the machine attestation methods (CI tests) * had to adjust the test cases to the impact of that change. Signed-off-by: Jürgen Eckel <juergen@riddleandcode.com>
140 lines
3.2 KiB
Go
140 lines
3.2 KiB
Go
package mocks
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
mqtt "github.com/eclipse/paho.mqtt.golang"
|
|
)
|
|
|
|
// MockMQTTClient is the mock mqtt client
|
|
type MockMQTTClient struct {
|
|
ConnectFunc func() mqtt.Token
|
|
DisconnectFunc func(quiesce uint)
|
|
PublishFunc func(topic string, qos byte, retained bool, payload interface{}) mqtt.Token
|
|
SubscribeFunc func(topic string, qos byte, callback mqtt.MessageHandler) mqtt.Token
|
|
UnsubscribeFunc func(topics ...string) mqtt.Token
|
|
IsConnectedFunc func() bool
|
|
IsConnectionOpenFunc func() bool
|
|
connected bool
|
|
connectedMutex sync.Mutex
|
|
}
|
|
|
|
// GetConnectFunc fetches the mock client's `Connect` func
|
|
func GetConnectFunc() mqtt.Token {
|
|
token := mqtt.DummyToken{}
|
|
return &token
|
|
}
|
|
|
|
// GetDisconnectFunc fetches the mock client's `Disconnect` func
|
|
func GetDisconnectFunc(_ uint) {
|
|
// not implemented at this point in time
|
|
}
|
|
|
|
// GetPublishFunc fetches the mock client's `Publish` func
|
|
func GetPublishFunc(_ string, _ byte, _ bool, _ interface{}) mqtt.Token {
|
|
token := mqtt.DummyToken{}
|
|
return &token
|
|
}
|
|
|
|
type message struct {
|
|
duplicate bool
|
|
qos byte
|
|
retained bool
|
|
topic string
|
|
messageID uint16
|
|
payload []byte
|
|
ack func()
|
|
once sync.Once
|
|
}
|
|
|
|
func (m *message) Duplicate() bool {
|
|
return m.duplicate
|
|
}
|
|
|
|
func (m *message) Qos() byte {
|
|
return m.qos
|
|
}
|
|
|
|
func (m *message) Retained() bool {
|
|
return m.retained
|
|
}
|
|
|
|
func (m *message) Topic() string {
|
|
return m.topic
|
|
}
|
|
|
|
func (m *message) MessageID() uint16 {
|
|
return m.messageID
|
|
}
|
|
|
|
func (m *message) Payload() []byte {
|
|
return m.payload
|
|
}
|
|
|
|
func (m *message) Ack() {
|
|
m.once.Do(m.ack)
|
|
}
|
|
|
|
func SendCallbackResponse(topic string, callback mqtt.MessageHandler) {
|
|
time.Sleep(100 * time.Millisecond)
|
|
var opt mqtt.ClientOptions
|
|
client := mqtt.NewClient(&opt)
|
|
var msg message
|
|
msg.topic = topic
|
|
var msgInter mqtt.Message = &msg
|
|
|
|
callback(client, msgInter)
|
|
}
|
|
|
|
func GetSubscribeFunc(topic string, _ byte, callback mqtt.MessageHandler) mqtt.Token {
|
|
go SendCallbackResponse(topic, callback)
|
|
token := mqtt.DummyToken{}
|
|
return &token
|
|
}
|
|
|
|
func GetUnsubscribeFunc(_ ...string) mqtt.Token {
|
|
token := mqtt.DummyToken{}
|
|
return &token
|
|
}
|
|
|
|
// Connect is the mock client's `Disconnect` func
|
|
func (m *MockMQTTClient) Connect() mqtt.Token {
|
|
m.connectedMutex.Lock()
|
|
m.connected = true
|
|
m.connectedMutex.Unlock()
|
|
return GetConnectFunc()
|
|
}
|
|
|
|
// Disconnect is the mock client's `Disconnect` func
|
|
func (m *MockMQTTClient) Disconnect(quiesce uint) {
|
|
GetDisconnectFunc(quiesce)
|
|
}
|
|
|
|
// Publish is the mock client's `Publish` func
|
|
func (m *MockMQTTClient) Publish(topic string, qos byte, retained bool, payload interface{}) mqtt.Token {
|
|
return GetPublishFunc(topic, qos, retained, payload)
|
|
}
|
|
|
|
func (m *MockMQTTClient) Subscribe(topic string, qos byte, callback mqtt.MessageHandler) mqtt.Token {
|
|
return GetSubscribeFunc(topic, qos, callback)
|
|
}
|
|
|
|
func (m *MockMQTTClient) Unsubscribe(topics ...string) mqtt.Token {
|
|
return GetUnsubscribeFunc(topics...)
|
|
}
|
|
|
|
func (m *MockMQTTClient) IsConnected() bool {
|
|
m.connectedMutex.Lock()
|
|
connected := m.connected
|
|
m.connectedMutex.Unlock()
|
|
return connected
|
|
}
|
|
|
|
func (m *MockMQTTClient) IsConnectionOpen() bool {
|
|
m.connectedMutex.Lock()
|
|
connected := m.connected
|
|
m.connectedMutex.Unlock()
|
|
return connected
|
|
}
|