mock_tracker.go

package rex

import "encoding/json"

type MockTracker struct {
	BaseTracker
	Messages map[string][][]byte
}

func NewMockTracker(metadata *EventMetadata) *MockTracker {
	self := &MockTracker{}
	self.EventMetadata = metadata
	self.Messages = make(map[string][][]byte)
	return self
}

func (self *MockTracker) Get(topic string, idx int) (map[string]interface{}, error) {
	var r map[string]interface{}
	if len(self.Messages) == 0 {
		return nil, nil
	}
	msgs := self.Messages[topic]
	if msgs == nil || len(msgs) < idx {
		return nil, nil
	}
	err := json.Unmarshal(msgs[idx], &r)
	return r, err
}

func (self *MockTracker) Close() {
}

func (self *MockTracker) FastMessage(topic string, message []byte) error {
	self.Messages[topic] = append(self.Messages[topic], message)
	return nil
}

func (self *MockTracker) FastEvent(topic string, e EventBase, full bool) error {
	self.AddMetadata(e, full)
	return self.FastMessage(topic, self.Encode(e))
}

func (self *MockTracker) FastEventMap(topic string, event map[string]interface{}, full bool) error {
	self.AddMetadataMap(event, full)
	return self.FastMessage(topic, self.EncodeMap(event))
}

func (self *MockTracker) SafeMessage(topic string, message []byte) error {
	self.Messages[topic] = append(self.Messages[topic], message)
	return nil
}

func (self *MockTracker) SafeEvent(topic string, e EventBase, full bool) error {
	self.AddMetadata(e, full)
	return self.SafeMessage(topic, self.Encode(e))
}

func (self *MockTracker) SafeEventMap(topic string, event map[string]interface{}, full bool) error {
	self.AddMetadataMap(event, full)
	return self.SafeMessage(topic, self.EncodeMap(event))
}