diff --git a/tracer/tracer_test.go b/tracer/tracer_test.go new file mode 100644 index 0000000..a41e25c --- /dev/null +++ b/tracer/tracer_test.go @@ -0,0 +1,35 @@ +package tracer + +import ( + "github.com/stretchr/testify/assert" + "testing" +) + +func TestInit(t *testing.T) { + mockStrategy := func(event Event) {} + + tracer := Init(mockStrategy) + + assert.NotNil(t, tracer.strategy) +} + +func TestTraceEvent(t *testing.T) { + eventCalled := Event{} + + mockStrategy := func(event Event) { + eventCalled = event + } + + tracer := Init(mockStrategy) + + tracer.TraceEvent(Event{ + ID: "mockID", + Protocol: HTTP, + Status: Stateless, + }) + + assert.NotNil(t, eventCalled.ID) + assert.Equal(t, eventCalled.ID, "mockID") + assert.Equal(t, eventCalled.Protocol, HTTP) + assert.Equal(t, eventCalled.Status, Stateless) +}