probes, Fail on Login error

The following functions do not fail when loginTo function fails: - StartTCPServer - StartHTTPServer - StartHTTPServerWithSourceIp Adding Expect accordingly. Signed-off-by: Ram Lavi <[email protected]>
attlee-wang · Jun 8, 2022 · affa2dd · affa2dd
1 parent b44232e
commit affa2dd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/vmi_servers.go b/tests/vmi_servers.go
@@ -23,17 +23,17 @@ func (s server) composeNetcatServerCommand(port int, extraArgs ...string) string
 }
 
 func StartTCPServer(vmi *v1.VirtualMachineInstance, port int, loginTo console.LoginToFunction) {
-	loginTo(vmi)
+	ExpectWithOffset(1, loginTo(vmi)).To(Succeed())
 	TCPServer.Start(vmi, port)
 }
 
 func StartHTTPServer(vmi *v1.VirtualMachineInstance, port int, loginTo console.LoginToFunction) {
-	loginTo(vmi)
+	ExpectWithOffset(1, loginTo(vmi)).To(Succeed())
 	HTTPServer.Start(vmi, port)
 }
 
 func StartHTTPServerWithSourceIp(vmi *v1.VirtualMachineInstance, port int, sourceIP string, loginTo console.LoginToFunction) {
-	loginTo(vmi)
+	ExpectWithOffset(1, loginTo(vmi)).To(Succeed())
 	HTTPServer.Start(vmi, port, fmt.Sprintf("-s %s", sourceIP))
 }