Add unit test for with/without Kafka executor message key

distribworks · Oct 9, 2021 · 3d91bcd · 3d91bcd
1 parent 4339456
commit 3d91bcd
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/builtin/bins/dkron-executor-kafka/kafka_test.go b/builtin/bins/dkron-executor-kafka/kafka_test.go
@@ -7,12 +7,13 @@ import (
 	dktypes "github.com/distribworks/dkron/v3/plugin/types"
 )
 
-func TestProduceExecute(t *testing.T) {
+func TestProduceExecuteWithKey(t *testing.T) {
 	pa := &dktypes.ExecuteRequest{
-		JobName: "testJob",
+		JobName: "testJobWithKey",
 		Config: map[string]string{
 			"topic":         "test",
 			"brokerAddress": "testaddress",
+			"key":           "testkey",
 			"message":       "{\"hello\":11}",
 			"debug":         "true",
 		},
@@ -25,3 +26,22 @@ func TestProduceExecute(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestProduceExecuteWithoutKey(t *testing.T) {
+	pa := &dktypes.ExecuteRequest{
+		JobName: "testJobWithoutKey",
+		Config: map[string]string{
+			"topic":         "test",
+			"brokerAddress": "testaddress",
+			"message":       "{\"hello\":11}",
+			"debug":         "true",
+		},
+	}
+	kafka := &Kafka{}
+	output, err := kafka.Execute(pa, nil)
+	fmt.Println(string(output.Output))
+	fmt.Println(err)
+	if err != nil {
+		t.Fatal(err)
+	}
+}