Add timeout

duck8823 · Nov 28, 2018 · 56542ff · 56542ff
1 parent 3fd0d1d
commit 56542ff
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/application/service/runner/task_runner_test.go b/application/service/runner/task_runner_test.go
@@ -10,6 +10,7 @@ import (
 	"os"
 	"path"
 	"testing"
+	"time"
 )
 
 func TestDockerTaskRunner_Run(t *testing.T) {
@@ -45,8 +46,12 @@ func TestDockerTaskRunner_Run(t *testing.T) {
 		Docker: mockDocker,
 	}
 
+	// and
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
 	// when
-	err := sut.Run(context.Background(), dir, runner.RunOptions{})
+	err := sut.Run(ctx, dir, runner.RunOptions{})
 
 	// then
 	if err != nil {