From 187a8494750ceceef05188df74bcb4b5779b8175 Mon Sep 17 00:00:00 2001
From: Mario <m4r10.php@gmail.com>
Date: Wed, 11 May 2022 22:58:33 +0200
Subject: [PATCH] Improve tracer_test.go

---
 tracer/tracer_test.go | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 tracer/tracer_test.go

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)
+}