Added test cases

Author: samujjwaal

src/test/java/com/DesignPatternTest.java

@@ -0,0 +1,41 @@
+package com;
+
+import com.squareup.javapoet.JavaFile;
+import com.structural.Flyweight;
+import org.junit.jupiter.api.Test;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class DesignPatternTest {
+    String[] classNames = {"Flyweight", "ConcreteFlyweight","UnsharedConcreteFlyweight","FlyweightFactory"};
+    String packageName = "com.test_generateCode.flyweight";
+
+    @Test
+    void test_generateCode() throws IOException {
+
+        // Test Case to verify  generateCode() returns JavaFile[] object
+
+        assertEquals(JavaFile[].class,new Flyweight(0).generateCode(classNames, packageName).getClass());
+    }
+
+    @Test
+    void test_writeJavaFiles() throws IOException {
+
+        // Test Case to verify the generated java files are written successfully in the output folder
+        int mode = 0;
+        Flyweight test = new Flyweight(mode);
+        JavaFile[] test_files = test.generateCode(classNames, packageName);
+        test.writeJavaFiles(test_files);
+        Path path = Paths.get("outputs/com/test_generateCode/flyweight");
+        assertTrue(Files.exists(path));
+
+
+    }
+
+}

src/test/java/com/Hw1DesignPatternGeneratorTest.java

@@ -0,0 +1,41 @@
+package com;
+
+import com.structural.Flyweight;
+import org.junit.jupiter.api.Test;
+
+import java.io.IOException;
+
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class Hw1DesignPatternGeneratorTest {
+
+
+    @Test
+    void test_getInstance() {
+
+        // Test Case to verify only 1 instance of Hw1DesignPatternGenerator is created each time
+
+        Hw1DesignPatternGenerator firstInstance = Hw1DesignPatternGenerator.getInstance();
+        Hw1DesignPatternGenerator secondInstance = Hw1DesignPatternGenerator.getInstance();
+
+        assertEquals(firstInstance,secondInstance);
+    }
+
+
+    @Test
+    void chooseDesignPattern() throws IOException {
+
+        //Test Case to verify correct design pattern file is executed when user inputs the choice of design pattern
+        // Here Flyweight( 11 ) is selected.
+
+        DesignPattern design_pattern2 = Hw1DesignPatternGenerator.getInstance().chooseDesignPattern(11,0);
+        assertEquals(Flyweight.class,design_pattern2.getClass());
+
+        // Test Case to check null is returned on selecting incorrect design pattern choice
+        int choice = 75;
+        DesignPattern design_pattern = Hw1DesignPatternGenerator.getInstance().chooseDesignPattern(choice,0);
+        assertNull(design_pattern);
+
+    }
+}