fix tests

Signed-off-by: Frank Jogeleit <frank.jogeleit@web.de>
kyverno · Sep 5, 2024 · db44606 · db44606
1 parent dea2c73
commit db44606
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/pkg/listener/cleanup_test.go b/pkg/listener/cleanup_test.go
@@ -14,7 +14,7 @@ func Test_CleanupListener(t *testing.T) {
 		c := &client{}
 
 		slistener := listener.NewCleanupListener(ctx, target.NewCollection(&target.Target{Client: c}))
-		slistener(report.LifecycleEvent{Type: report.Added, PolicyReport: preport1})
+		slistener(report.LifecycleEvent{Type: report.Deleted, PolicyReport: preport1})
 
 		assert.True(t, c.cleanupCalled, "expected cleanup method was called")
 	})

diff --git a/pkg/target/securityhub/securityhub.go b/pkg/target/securityhub/securityhub.go
@@ -112,7 +112,7 @@ func (c *client) mapFindings(polr v1alpha2.ReportInterface, results []v1alpha2.P
 }
 
 func (c *client) Send(result v1alpha2.PolicyReportResult) {
-	c.BatchSend(nil, []v1alpha2.PolicyReportResult{result})
+	c.BatchSend(&v1alpha2.PolicyReport{}, []v1alpha2.PolicyReportResult{result})
 }
 
 func (c *client) BatchSend(polr v1alpha2.ReportInterface, results []v1alpha2.PolicyReportResult) {

diff --git a/pkg/target/securityhub/securityhub_test.go b/pkg/target/securityhub/securityhub_test.go
@@ -42,7 +42,8 @@ func (c *client) GetFindings(ctx context.Context, params *hub.GetFindingsInput,
 }
 
 func (c *client) BatchUpdateFindings(ctx context.Context, params *hub.BatchUpdateFindingsInput, optFns ...func(*hub.Options)) (*hub.BatchUpdateFindingsOutput, error) {
-	c.fetched = true
+	c.batched = true
+
 	return &hub.BatchUpdateFindingsOutput{}, nil
 }