Initial commit

clients/mock_broker.go

@@ -0,0 +1,35 @@
+package clients
+
+import (
+	"btclock/broker"
+
+	"github.com/stretchr/testify/mock"
+)
+
+// MockEventBroker implements broker.EventBroker for testing
+type MockEventBroker struct {
+	mock.Mock
+	realBroker *broker.EventBroker
+}
+
+func NewMockEventBroker() *broker.EventBroker {
+	mock := &MockEventBroker{
+		realBroker: broker.NewEventBroker(),
+	}
+	return mock.realBroker
+}
+
+func (m *MockEventBroker) Register(eventType string, handler broker.EventHandler) {
+	m.Called(eventType, handler)
+	m.realBroker.Register(eventType, handler)
+}
+
+func (m *MockEventBroker) Publish(eventType string, event broker.Event) {
+	m.Called(eventType, event)
+	m.realBroker.Publish(eventType, event)
+}
+
+func (m *MockEventBroker) Unregister(eventType string, handler broker.EventHandler) {
+	m.Called(eventType, handler)
+	m.realBroker.Unregister(eventType, handler)
+}