mockChainWriter *mockBatchChainByteWriter
mockChainer *mockBatchCounterChainer
)
BeforeEach(func() {
mockBatcher = newMockEventBatcher()
mockChainWriter = newMockBatchChainByteWriter()
mockChainer = newMockBatchCounterChainer()
testhelpers.AlwaysReturn(mockBatcher.BatchCounterOutput, mockChainer)
testhelpers.AlwaysReturn(mockChainer.SetTagOutput, mockChainer)
})
JustBeforeEach(func() {
logger := gosteno.NewLogger("TestLogger")
marshaller = eventmarshaller.New(mockBatcher, logger)
marshaller.SetWriter(mockChainWriter)
})
Describe("Write", func() {
var envelope *events.Envelope
Context("with a nil writer", func() {
BeforeEach(func() {
envelope = &events.Envelope{
Origin: proto.String("The Negative Zone"),
EventType: events.Envelope_LogMessage.Enum(),
}
})
JustBeforeEach(func() {
marshaller.SetWriter(nil)
var (
marshaller *eventmarshaller.EventMarshaller
sender *fake.FakeMetricSender
writer *mockByteWriter
)
BeforeEach(func() {
writer = newMockByteWriter()
sender = fake.NewFakeMetricSender()
metrics.Initialize(sender, metricbatcher.New(sender, time.Millisecond*10))
})
JustBeforeEach(func() {
logger := gosteno.NewLogger("TestLogger")
marshaller = eventmarshaller.New(logger)
marshaller.SetWriter(writer)
})
Describe("Write", func() {
var envelope *events.Envelope
Context("with a nil writer", func() {
BeforeEach(func() {
envelope = &events.Envelope{
Origin: proto.String("The Negative Zone"),
EventType: events.Envelope_LogMessage.Enum(),
}
})
JustBeforeEach(func() {
marshaller.SetWriter(nil)