about summary refs log tree commit diff
path: root/src/payload-processor/processor/processor_test.go
diff options
context:
space:
mode:
authorBaitinq <manuelpalenzuelamerino@gmail.com>2024-05-09 00:15:13 +0200
committerBaitinq <manuelpalenzuelamerino@gmail.com>2024-05-11 01:32:43 +0200
commit0a7be3e1ca6e651759fdb853e83e639f476e0c47 (patch)
tree0a53188b22edf971380290fefc3b6075a6a4e629 /src/payload-processor/processor/processor_test.go
parentk8s: set default number of kafka topic partitions to 2 (diff)
downloadfs-tracer-backend-0a7be3e1ca6e651759fdb853e83e639f476e0c47.tar.gz
fs-tracer-backend-0a7be3e1ca6e651759fdb853e83e639f476e0c47.tar.bz2
fs-tracer-backend-0a7be3e1ca6e651759fdb853e83e639f476e0c47.zip
payloads-processor: connect to DB and insert dummy data
Diffstat (limited to 'src/payload-processor/processor/processor_test.go')
-rw-r--r--src/payload-processor/processor/processor_test.go12
1 files changed, 10 insertions, 2 deletions
diff --git a/src/payload-processor/processor/processor_test.go b/src/payload-processor/processor/processor_test.go
index ee2ef5f..5e954c4 100644
--- a/src/payload-processor/processor/processor_test.go
+++ b/src/payload-processor/processor/processor_test.go
@@ -1,18 +1,26 @@
 package processor
 
 import (
+	"context"
 	"testing"
 
 	"github.com/segmentio/kafka-go"
 	"github.com/stretchr/testify/require"
+	gomock "go.uber.org/mock/gomock"
 )
 
 func TestProcessMessage(t *testing.T) {
-	processor := Processor{}
+	ctrl := gomock.NewController(t)
+	mockdb := NewMockDB(ctrl)
+	processor := Processor{
+		db: mockdb,
+	}
 
 	message := []byte("test")
 
-	err := processor.handleMessage(kafka.Message{Value: message})
+	mockdb.EXPECT().TestInsert(gomock.Any(), string(message)).Return(nil)
+
+	err := processor.handleMessage(context.Background(), kafka.Message{Value: message})
 
 	require.NoError(t, err)
 }