helper: embed context on creation

Signed-off-by: Ophestra <cat@gensokyo.uk>

1 files changed, 4 insertions, 4 deletions

diff --git a/helper/direct_test.go b/helper/direct_test.go
index 67b86827..e7bda797 100644
--- a/helper/direct_test.go
+++ b/helper/direct_test.go
@@ -11,16 +11,16 @@ import (
 
 func TestDirect(t *testing.T) {
 	t.Run("start non-existent helper path", func(t *testing.T) {
-		h := helper.New(argsWt, "/nonexistent", argF)
+		h := helper.New(context.Background(), argsWt, "/nonexistent", argF)
 
-		if err := h.Start(context.Background(), false); !errors.Is(err, os.ErrNotExist) {
+		if err := h.Start(false); !errors.Is(err, os.ErrNotExist) {
 			t.Errorf("Start: error = %v, wantErr %v",
 				err, os.ErrNotExist)
 		}
 	})
 
 	t.Run("valid new helper nil check", func(t *testing.T) {
-		if got := helper.New(argsWt, "fortify", argF); got == nil {
+		if got := helper.New(context.TODO(), argsWt, "fortify", argF); got == nil {
 			t.Errorf("New(%q, %q) got nil",
 				argsWt, "fortify")
 			return
@@ -28,6 +28,6 @@ func TestDirect(t *testing.T) {
 	})
 
 	t.Run("implementation compliance", func(t *testing.T) {
-		testHelper(t, func() helper.Helper { return helper.New(argsWt, "crash-test-dummy", argF) })
+		testHelper(t, func(ctx context.Context) helper.Helper { return helper.New(ctx, argsWt, "crash-test-dummy", argF) })
 	})
 }