test(action): add timeout to SSH server start

Signed-off-by: AtomicFS <vojtech.vesely@9elements.com>

action/container/ssh_test.go

@@ -9,6 +9,7 @@ import (
 	"fmt"
 	"os"
 	"testing"
+	"time"
 
 	"dagger.io/dagger"
 	"github.com/stretchr/testify/assert"
@@ -101,6 +102,13 @@ func TestOpenSSH(t *testing.T) {
 			myContainer, err := Setup(ctx, client, &tc.opts, "")
 			assert.NoError(t, err)
 
+			// Set timeout for opening the SSH server
+			timeout := make(chan bool, 5)
+			go func() {
+				time.Sleep(1 * time.Second)
+				timeout <- true
+			}()
+
 			// Open the SSH
 			go func() {
 				err := OpenSSH(ctx, client, myContainer, tc.opts.WorkdirContainer, tc.optsSSH)
@@ -110,7 +118,13 @@ func TestOpenSSH(t *testing.T) {
 				assert.NoError(t, err)
 			}()
 			// Wait until SSH server is ready
-			<-tc.optsSSH.TunnelReady
+			select {
+			case <-tc.optsSSH.TunnelReady:
+				// the SSH tunnel is ready
+			case <-timeout:
+				// the SSH tunnel is not ready yet, and has timed out
+				t.Fatal("Opening SSH server has taken too long")
+			}
 
 			// Connect with client
 			config := &ssh.ClientConfig{