diff --git a/internal/agentupdate/coverage_test.go b/internal/agentupdate/coverage_test.go index 8abf3924e3..1215785ded 100644 --- a/internal/agentupdate/coverage_test.go +++ b/internal/agentupdate/coverage_test.go @@ -168,21 +168,6 @@ func TestRunDownloadedBinarySelfTestPropagatesFailureWithoutTokenArg(t *testing. } } -func TestRestartProcess(t *testing.T) { - orig := execFn - t.Cleanup(func() { execFn = orig }) - - execFn = func(string, []string, []string) error { return nil } - if err := restartProcess("/bin/true"); err != nil { - t.Fatalf("expected nil error, got %v", err) - } - - execFn = func(string, []string, []string) error { return errors.New("boom") } - if err := restartProcess("/bin/false"); err == nil { - t.Fatalf("expected error") - } -} - func TestDetermineArchOverrides(t *testing.T) { origOS, origArch, origUname := runtimeGOOS, runtimeGOARCH, unameCommand t.Cleanup(func() { diff --git a/internal/agentupdate/restart_unix_test.go b/internal/agentupdate/restart_unix_test.go new file mode 100644 index 0000000000..d74f1d0092 --- /dev/null +++ b/internal/agentupdate/restart_unix_test.go @@ -0,0 +1,23 @@ +//go:build !windows + +package agentupdate + +import ( + "errors" + "testing" +) + +func TestRestartProcess(t *testing.T) { + orig := execFn + t.Cleanup(func() { execFn = orig }) + + execFn = func(string, []string, []string) error { return nil } + if err := restartProcess("/bin/true"); err != nil { + t.Fatalf("expected nil error, got %v", err) + } + + execFn = func(string, []string, []string) error { return errors.New("boom") } + if err := restartProcess("/bin/false"); err == nil { + t.Fatal("expected error") + } +}