2 files changed, 79 insertions, 0 deletions
diff --git a/appc-client/code-generator/src/test/java/org/openecomp/appc/tools/generator/api/CLITest.java b/appc-client/code-generator/src/test/java/org/openecomp/appc/tools/generator/api/CLITest.java
new file mode 100644
index 000000000..3d6eb1bf3
--- /dev/null
+++ b/appc-client/code-generator/src/test/java/org/openecomp/appc/tools/generator/api/CLITest.java
@@ -0,0 +1,57 @@
+package org.openecomp.appc.tools.generator.api;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+public class CLITest {
+    @Test
+    public void missingSourceFileTest()  {
+        CLI cli = new CLI();
+        try {
+            String[] input = new String[1];
+            cli.main(input);
+        } catch (Exception e) {
+            Assert.assertEquals("Source file is missing. Please add argument 'client <source file> <destination file> <model template>'",e.getMessage());
+        }
+    }
+    @Test
+    public void missingDestinationFileTest()  {
+        CLI cli = new CLI();
+        try {
+            String[] input = {"sourceFilePath",null};
+            cli.main(input);
+        } catch (Exception e) {
+            Assert.assertEquals("Destination file name is missing. Please add argument 'client sourceFilePath <destination> <model template> <builder> <conf file>'",e.getMessage());
+        }
+    }
+    @Test
+    public void missingTemplateFileTest()  {
+        CLI cli = new CLI();
+        try {
+            String[] input = {"sourceFilePath","destinationPath",null};
+            cli.main(input);
+        } catch (Exception e) {
+            Assert.assertEquals("template file name is missing. Please add argument 'client sourceFilePath destinationPath <model template> <builder> <conf file>'",e.getMessage());
+        }
+    }
+    @Test
+    public void missingBuilderNameTest()  {
+        CLI cli = new CLI();
+        try {
+            String[] input = {"sourceFilePath","destinationPath","templateFileName",null};
+            cli.main(input);
+        } catch (Exception e) {
+            Assert.assertEquals("builder FQDN is missing. Please add argument 'client sourceFilePath destinationPath templateFileName <builder> <conf file>'",e.getMessage());
+        }
+    }
+    @Test
+    public void missingContextConfFileNameTest()  {
+        CLI cli = new CLI();
+        try {
+            String[] input = {"sourceFilePath","destinationPath","templateFileName","builderFQDN",null};
+            cli.main(input);
+        } catch (Exception e) {
+            Assert.assertEquals(e.getMessage(),"context conf file is missing. Please add argument 'client sourceFilePath destinationPath templateFileName builderFQDN <conf file>'");
+        }
+    }
+}
diff --git a/appc-client/code-generator/src/test/java/org/openecomp/appc/tools/generator/impl/CodeGenWriterTest.java b/appc-client/code-generator/src/test/java/org/openecomp/appc/tools/generator/impl/CodeGenWriterTest.java
new file mode 100644
index 000000000..84cca9337
--- /dev/null
+++ b/appc-client/code-generator/src/test/java/org/openecomp/appc/tools/generator/impl/CodeGenWriterTest.java
@@ -0,0 +1,22 @@
+package org.openecomp.appc.tools.generator.impl;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.io.IOException;
+
+
+public class CodeGenWriterTest {
+    @Test
+    public void writeTest() throws IOException {
+        CodeGenWriter codeGenWriter = new CodeGenWriter("destination");
+        char[] cbuf = {'t','e','s','t'};
+        int off = 1;
+        int len = 3;
+        codeGenWriter.write(cbuf,off,len);
+        codeGenWriter.flush();
+        codeGenWriter.close();
+        Assert.assertNotNull(codeGenWriter);
+    }
+
+}