test: update for signature change

Signed-off-by: Mike Beaumont <[email protected]>
kumahq · Jul 13, 2023 · 9d4425d · 9d4425d
1 parent 2e4afdf
commit 9d4425d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/pkg/hds/tracker/healthcheck_generator_test.go b/pkg/hds/tracker/healthcheck_generator_test.go
@@ -47,7 +47,7 @@ var _ = Describe("HDS Snapshot generator", func() {
 			generator := NewSnapshotGenerator(resourceManager, given.hdsConfig, 9901)
 
 			// when
-			snapshot, err := generator.GenerateSnapshot(&envoy_config_core_v3.Node{Id: "mesh-1.dp-1"})
+			snapshot, err := generator.GenerateSnapshot(context.Background(), &envoy_config_core_v3.Node{Id: "mesh-1.dp-1"})
 
 			// then
 			Expect(err).ToNot(HaveOccurred())

diff --git a/pkg/util/watchdog/watchdog_test.go b/pkg/util/watchdog/watchdog_test.go
@@ -1,6 +1,7 @@
 package watchdog_test
 
 import (
+	"context"
 	"fmt"
 	"time"
 
@@ -37,7 +38,7 @@ var _ = Describe("SimpleWatchdog", func() {
 					C: timeTicks,
 				}
 			},
-			OnTick: func() error {
+			OnTick: func(context.Context) error {
 				onTickCalls <- struct{}{}
 				return nil
 			},
@@ -82,7 +83,7 @@ var _ = Describe("SimpleWatchdog", func() {
 					C: timeTicks,
 				}
 			},
-			OnTick: func() error {
+			OnTick: func(context.Context) error {
 				return expectedErr
 			},
 			OnError: func(err error) {
@@ -121,7 +122,7 @@ var _ = Describe("SimpleWatchdog", func() {
 					C: timeTicks,
 				}
 			},
-			OnTick: func() error {
+			OnTick: func(context.Context) error {
 				panic("xyz")
 			},
 			OnError: func(err error) {