From 64593574d88acf82f9861e2dc39fe24b4822126e Mon Sep 17 00:00:00 2001
From: Charles Weng <hueborg@proton.me>
Date: Wed, 12 Jun 2024 06:59:01 -0700
Subject: [PATCH] test update and get connected hosts (#9136)

---
 .../test/java/com/cloud/agent/AgentShellTest.java    | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/agent/src/test/java/com/cloud/agent/AgentShellTest.java b/agent/src/test/java/com/cloud/agent/AgentShellTest.java
index f7151779f58..4126692546f 100644
--- a/agent/src/test/java/com/cloud/agent/AgentShellTest.java
+++ b/agent/src/test/java/com/cloud/agent/AgentShellTest.java
@@ -350,4 +350,16 @@ public class AgentShellTest {
 
         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());
+    }
 }