Add agent system info test

Author: yunheli

platform-agent/src/main/java/com/flow/platform/agent/CmdManager.java

@@ -212,6 +212,7 @@ public void run() {
             log.setCategory(Category.SYSTEM_INFO);
             logListener.onLog(log);
             logListener.onFinish();
+            cmd.setStatus(CmdStatus.EXECUTED);
         }
 
         // kill current running proc

platform-agent/src/test/java/com/flow/platform/agent/test/CmdManagerTest.java

@@ -21,6 +21,7 @@
 import com.flow.platform.cmd.ProcListener;
 import com.flow.platform.domain.Cmd;
 import com.flow.platform.domain.CmdResult;
+import com.flow.platform.domain.CmdStatus;
 import com.flow.platform.domain.CmdType;
 import java.io.IOException;
 import java.util.Map;
@@ -213,4 +214,56 @@ public void onException(CmdResult result) {
         Assert.assertNotNull(result.getExitValue());
         Assert.assertEquals(CmdResult.EXIT_VALUE_FOR_KILL, result.getExitValue());
     }
+
+
+    @Test
+    public void should_success_run_sys_cmd() throws InterruptedException {
+
+        Cmd cmd = new Cmd("zone1", "agent1", CmdType.RUN_SHELL, resourcePath);
+        cmd.setId(UUID.randomUUID().toString());
+        CountDownLatch finishCountDownLatch = new CountDownLatch(1);
+        CountDownLatch startCountDownLatch = new CountDownLatch(1);
+
+        cmdManager.getExtraProcEventListeners().add(new ProcListener() {
+            @Override
+            public void onStarted(CmdResult result) {
+                startCountDownLatch.countDown();
+                try {
+                    Thread.sleep(3000);
+                } catch (Throwable e) {
+                }
+            }
+
+            @Override
+            public void onLogged(CmdResult result) {
+                finishCountDownLatch.countDown();
+            }
+
+            @Override
+            public void onExecuted(CmdResult result) {
+
+            }
+
+            @Override
+            public void onException(CmdResult result) {
+
+            }
+        });
+
+        // when: start and kill task immediately
+        cmdManager.execute(cmd);
+
+        startCountDownLatch.await();
+        Assert.assertEquals(1, cmdManager.getRunning().size());
+
+        Cmd cmdSys = new Cmd("zone1", "agent1", CmdType.SYSTEM_INFO, "");
+        cmdSys.setId(UUID.randomUUID().toString());
+
+        cmdManager.execute(cmdSys);
+        Assert.assertEquals(CmdStatus.EXECUTED, cmdSys.getStatus());
+
+        finishCountDownLatch.await();
+        Assert.assertEquals(1, cmdManager.getFinished().size());
+
+    }
 }