mirror of
https://github.com/kevin-DL/services.git
synced 2026-01-12 03:05:14 +00:00
b8877f8312
* fixup streams stuff * use cache instead of pg in streams * hash out issuer streams test * fix compile error
44 lines
947 B
Go
44 lines
947 B
Go
package handler_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/micro/micro/v3/service/events"
|
|
"github.com/micro/micro/v3/service/store/memory"
|
|
"github.com/micro/services/pkg/cache"
|
|
"github.com/micro/services/streams/handler"
|
|
)
|
|
|
|
func testHandler(t *testing.T) *handler.Streams {
|
|
h := &handler.Streams{
|
|
Cache: cache.New(memory.NewStore()),
|
|
Events: new(eventsMock),
|
|
Time: func() time.Time {
|
|
return time.Unix(1612787045, 0)
|
|
},
|
|
}
|
|
return h
|
|
}
|
|
|
|
type eventsMock struct {
|
|
PublishCount int
|
|
PublishTopic string
|
|
PublishMessage interface{}
|
|
|
|
ConsumeTopic string
|
|
ConsumeChan <-chan events.Event
|
|
}
|
|
|
|
func (e *eventsMock) Publish(topic string, msg interface{}, opts ...events.PublishOption) error {
|
|
e.PublishCount++
|
|
e.PublishTopic = topic
|
|
e.PublishMessage = msg
|
|
return nil
|
|
}
|
|
|
|
func (e *eventsMock) Consume(topic string, opts ...events.ConsumeOption) (<-chan events.Event, error) {
|
|
e.ConsumeTopic = topic
|
|
return e.ConsumeChan, nil
|
|
}
|