about summary refs log tree commit diff
path: root/src/payload-processor/processor/processor_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'src/payload-processor/processor/processor_test.go')
-rw-r--r--src/payload-processor/processor/processor_test.go8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/payload-processor/processor/processor_test.go b/src/payload-processor/processor/processor_test.go
index 4a55e48..d7849d9 100644
--- a/src/payload-processor/processor/processor_test.go
+++ b/src/payload-processor/processor/processor_test.go
@@ -19,20 +19,20 @@ func TestProcessMessage(t *testing.T) {
 	}
 
 	message := []byte(`
-	{
+	[{
 		"absolute_path": "/tmp/file.txt",
 		"contents": "hello world",
 		"timestamp": "2021-01-01T00:00:00Z"
-	}
+	}]
 	`)
 
 	ctx := context.Background()
 
-	mockdb.EXPECT().InsertFile(ctx, lib.File{
+	mockdb.EXPECT().InsertFiles(ctx, []lib.File{{
 		Absolute_path: "/tmp/file.txt",
 		Contents:      "hello world",
 		Timestamp:     time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC),
-	}, "USER_ID").Return(nil)
+	}}, "USER_ID").Return(nil)
 
 	err := processor.handleMessage(ctx, kafka.Message{Value: message}, "USER_ID")