test update and get connected hosts (#9136)

This commit is contained in:
Charles Weng 2024-06-12 06:59:01 -07:00 committed by GitHub
parent 66f7f76263
commit 64593574d8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -350,4 +350,16 @@ public class AgentShellTest {
Mockito.verify(agentShellSpy).setHosts(expected); Mockito.verify(agentShellSpy).setHosts(expected);
} }
@Test
public void updateAndGetConnectedHost() {
String expected = "test";
AgentShell shell = new AgentShell();
shell.setHosts("test");
shell.getNextHost();
shell.updateConnectedHost();
Assert.assertEquals(expected, shell.getConnectedHost());
}
} }