Add "already restarting" test case

skycoin · Dec 24, 2019 · b8e2cd0 · b8e2cd0
1 parent a926401
commit b8e2cd0
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/pkg/restart/restart_test.go b/pkg/restart/restart_test.go
@@ -66,6 +66,25 @@ func TestContext_Restart(t *testing.T) {
 
 		assert.Equal(t, ErrMalformedArgs, cc.Restart())
 	})
+
+	t.Run("already restarting", func(t *testing.T) {
+		cc.args = nil
+
+		cmd := "touch"
+		path := "/tmp/test_restart"
+		args := []string{cmd, path}
+		cc.args = args
+
+		ch := make(chan error, 1)
+		go func() {
+			ch <- cc.Restart()
+		}()
+
+		assert.NoError(t, cc.Restart())
+		assert.NoError(t, os.Remove(path))
+
+		assert.Equal(t, ErrAlreadyRestarting, <-ch)
+	})
 }
 
 func TestContext_SetCheckDelay(t *testing.T) {