fix dashboard host for ssh test

bisohns · Nov 25, 2022 · d0503f9 · d0503f9
1 parent 27f1cdd
commit d0503f9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/.github/workflows/test-ssh.yml b/.github/workflows/test-ssh.yml
@@ -23,5 +23,4 @@ jobs:
       - name: Test
         run: | 
           go mod tidy
-          ls /home/runner/work/saido/saido/ssh-key/ci/  
           go test -v ./...
diff --git a/integration/integration_test.go b/integration/integration_test.go
@@ -35,11 +35,18 @@ func NewSSHForTest() driver.Driver {
 	yamlPath := fmt.Sprintf("%s/%s", workingDir, "config-test.yaml")
 	conf := config.LoadConfig(yamlPath)
 	dashboardInfo := config.GetDashboardInfoConfig(conf)
+	var host config.Host
+	for ind := range dashboardInfo.Hosts {
+		if dashboardInfo.Hosts[ind].Address == "0.0.0.0" {
+			host = dashboardInfo.Hosts[ind]
+		}
+	}
+
 	return &driver.SSH{
-		User:            dashboardInfo.Hosts[0].Connection.Username,
-		Host:            dashboardInfo.Hosts[0].Address,
-		Port:            int(dashboardInfo.Hosts[0].Connection.Port),
-		KeyFile:         dashboardInfo.Hosts[0].Connection.PrivateKeyPath,
+		User:            host.Connection.Username,
+		Host:            host.Address,
+		Port:            int(host.Connection.Port),
+		KeyFile:         host.Connection.PrivateKeyPath,
 		KeyPass:         "",
 		CheckKnownHosts: false,
 	}