fix: integration test

Signed-off-by: Vladislav Sukhin <vladislav@kubeshop.io>
kubeshop · Jul 3, 2024 · c6c4c5c · c6c4c5c
1 parent be2e7c9
commit c6c4c5c
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/cmd/testworkflow-toolkit/commands/artifacts_test.go b/cmd/testworkflow-toolkit/commands/artifacts_test.go
@@ -9,6 +9,7 @@ import (
 	"testing"
 
 	"github.com/kubeshop/testkube/pkg/cloud/data/testworkflow"
+	"github.com/kubeshop/testkube/pkg/mapper/cdevents"
 	"github.com/kubeshop/testkube/pkg/utils/test"
 
 	"github.com/golang/mock/gomock"
@@ -37,6 +38,10 @@ func TestRun_Integration(t *testing.T) {
 			w.WriteHeader(http.StatusOK)
 			return
 		}
+		if r.Method == http.MethodPost {
+			w.WriteHeader(http.StatusOK)
+			return
+		}
 		http.NotFound(w, r)
 	})
 
@@ -66,9 +71,15 @@ func TestRun_Integration(t *testing.T) {
 		return filesystem.NewMockFile(path[1:], b), nil
 	})
 	postProcessor := artifacts.NewJUnitPostProcessor(mockFs, mockClient, "/", "")
-	handler := artifacts.NewHandler(uploader, processor, artifacts.WithPostProcessor(postProcessor))
+	handler := artifacts.NewHandler(uploader, processor, artifacts.WithPostProcessor(postProcessor),
+		artifacts.WithCDEventsTarget(server.URL), artifacts.WithCDEventsArtifactParameters(cdevents.CDEventsArtifactParameters{
+			Id:           "1",
+			Name:         "test-1",
+			WorkflowName: "test",
+			ClusterID:    "12345",
+		}))
 
 	run(handler, walker, testDataFixtures)
 
-	assert.Equal(t, 2, httpRequestCount)
+	assert.Equal(t, 4, httpRequestCount)
 }