Add timeout when pinging Docker daemon (#10)

* Add timeout to ping * Tidy code * Handle timeout with correct panic message * Check for context.DeadlineExceeded * Tidy code * Update main.go Co-authored-by: Sergei Egorov <[email protected]>
testcontainers · May 20, 2020 · c70f43e · c70f43e
1 parent ef4ff30
commit c70f43e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -24,7 +24,10 @@ func main() {
 
 	cli, err := client.NewEnvClient()
 	if err == nil {
-		_, err = cli.Ping(context.Background())
+		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+		defer cancel()
+
+		_, err = cli.Ping(ctx)
 	}
 
 	if err != nil {