modified: watchpid_test.go

immortal · Jul 22, 2016 · fe1f800 · fe1f800
1 parent fc33119
commit fe1f800
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/watchpid_test.go b/watchpid_test.go
@@ -2,6 +2,7 @@ package immortal
 
 import (
 	"os/exec"
+	"syscall"
 	"testing"
 	"time"
 )
@@ -29,11 +30,14 @@ func TestWatchPidGetpid(t *testing.T) {
 		D.watchPid(pid, ch)
 		ch <- cmd.Wait()
 	}()
-
-	err := <-ch
-	if err != nil {
-		if err.Error() != "EXIT" {
-			t.Error(err)
+	select {
+	case <-time.After(time.Millisecond):
+		syscall.Kill(pid, syscall.SIGTERM)
+	case err := <-ch:
+		if err != nil {
+			if err.Error() != "EXIT" {
+				t.Error(err)
+			}
 		}
 	}
 }