add unit test DcmToolTest

Author: oldratlee

tool/src/main/java/com/alibaba/dcm/tool/DcmTool.java

@@ -19,6 +19,8 @@
  */
 public class DcmTool {
     static final String DCM_AGENT_SUCCESS_MARK_LINE = "!!DCM SUCCESS!!";
+    static final String DCM_TOOLS_TMP_FILE = "DCM_TOOLS_TMP_FILE";
+    static final String DCM_TOOLS_AGENT_JAR = "DCM_TOOLS_AGENT_JAR";
 
     static List<String> actionList = new ArrayList<String>();
 
@@ -35,10 +37,9 @@ public class DcmTool {
         actionList.add("getNegativePolicy");
     }
 
-
     public static void main(String[] args) throws Exception {
-        final String tmpFile = System.getenv("DCM_TOOLS_TMP_FILE");
-        final String agentJar = System.getenv("DCM_TOOLS_AGENT_JAR");
+        final String tmpFile = getConfig(DCM_TOOLS_TMP_FILE);
+        final String agentJar = getConfig(DCM_TOOLS_AGENT_JAR);
         if (tmpFile == null || tmpFile.trim().isEmpty() || agentJar == null || agentJar.trim().isEmpty()) {
             throw new IllegalStateException("blank tmp file " + tmpFile + ", or blank agent jar file " + agentJar);
         }
@@ -110,4 +111,12 @@ public static void main(String[] args) throws Exception {
             exit(1);
         }
     }
+
+    static String getConfig(String name) {
+        String var = System.getenv(name);
+        if (var == null) {
+            var = System.getProperty(name);
+        }
+        return var;
+    }
 }

tool/src/test/java/com/alibaba/dcm/tool/DcmToolTest.java

@@ -0,0 +1,77 @@
+package com.alibaba.dcm.tool;
+
+import org.apache.commons.io.FileUtils;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.File;
+import java.lang.management.ManagementFactory;
+import java.util.Iterator;
+
+import static org.junit.Assert.*;
+
+/**
+ * @author oldratlee
+ */
+public class DcmToolTest {
+    File outputFile;
+    String outputFilePath;
+
+    String agentFilePath;
+
+    @Before
+    public void setUp() throws Exception {
+        outputFile = new File("target/output.log");
+        FileUtils.deleteQuietly(outputFile);
+        FileUtils.touch(outputFile);
+        assertTrue(outputFile.length() == 0);
+        System.out.println("Prepared output file: " + outputFile.getAbsolutePath());
+
+        outputFilePath = outputFile.getAbsolutePath();
+
+
+        File dcmLibProjectDir = new File("../library");
+        if (!dcmLibProjectDir.exists()) {
+            dcmLibProjectDir = new File("library");
+
+            if (!dcmLibProjectDir.exists()) {
+                fail("Not found dcm lib project!");
+            }
+        }
+        File dcmLibProjectTargetDir = new File(dcmLibProjectDir, "target");
+        if (!dcmLibProjectTargetDir.exists()) {
+            fail("Not found dcm lib project target dir!");
+        }
+        System.out.println("Found target dir: " + dcmLibProjectTargetDir);
+
+        final Iterator<File> fileIterator = FileUtils.iterateFiles(dcmLibProjectTargetDir, new String[]{"jar"}, false);
+        while (fileIterator.hasNext()) {
+            final File next = fileIterator.next();
+            final String fileName = next.getName();
+            final String absolutePath = next.getAbsolutePath();
+
+            if (fileName.startsWith("dns-cache-manipulator")) {
+                final String replaced = fileName.replace("dns-cache-manipulator-", "").replace("-SNAPSHOT", "");
+                if (!replaced.contains("-"))
+                    agentFilePath = absolutePath;
+            }
+        }
+
+        assertNotNull(agentFilePath);
+        System.out.println("Found agent file: " + agentFilePath);
+    }
+
+    @Test
+    public void test_main() throws Exception {
+        System.setProperty(DcmTool.DCM_TOOLS_TMP_FILE, outputFilePath);
+        System.setProperty(DcmTool.DCM_TOOLS_AGENT_JAR, agentFilePath);
+
+        DcmTool.main(new String[]{"-p", pid(), "getPolicy"});
+    }
+
+    static String pid() {
+        final String name = ManagementFactory.getRuntimeMXBean().getName();
+        final int idx = name.indexOf("@");
+        return name.substring(0, idx);
+    }
+}

tool/src/test/java/com/alibaba/dcm/tool/TestMainForAgentInject.java

@@ -9,9 +9,9 @@
 public class TestMainForAgentInject {
     public static void main(String[] args) throws Exception {
         while (true) {
-            System.out.println(new Date());
-            System.out.printf("\tbaidu.com: %s\n\n",
+            System.out.printf("%s: baidu.com: %s\n", new Date(),
                     InetAddress.getByName("baidu.com").getHostAddress());
+
             Thread.sleep(1000);
         }
     }