update test to reflect change to the queried ssh host

jesseduffield · Sep 7, 2024 · 9449434 · 9449434
1 parent e036f01
commit 9449434
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/commands/ssh/ssh_test.go b/pkg/commands/ssh/ssh_test.go
@@ -64,7 +64,7 @@ func TestSSHHandlerHandleSSHDockerHost(t *testing.T) {
 
 			startCmdCount := 0
 			startCmd := func(cmd *exec.Cmd) error {
-				assert.EqualValues(t, []string{"ssh", "-L", "/tmp/lazydocker-ssh-tunnel-12345/dockerhost.sock:/var/run/docker.sock", "192.168.5.178", "-N"}, cmd.Args)
+				assert.EqualValues(t, []string{"ssh", "-L", "/tmp/lazydocker-ssh-tunnel-12345/dockerhost.sock:/var/run/docker.sock", s.envVarValue, "-N"}, cmd.Args)
 
 				startCmdCount++
 
@@ -91,7 +91,7 @@ func TestSSHHandlerHandleSSHDockerHost(t *testing.T) {
 				setenv:      setenv,
 			}
 
-			_, err := handler.HandleSSHDockerHost()
+			_, err := handler.HandleSSHDockerHost(s.envVarValue)
 			assert.NoError(t, err)
 
 			assert.Equal(t, s.expectedDialContextCount, dialContextCount)