updated and new junit tests

src/test/java/de/monticore/lang/monticar/cnnarch/caffe2generator/GenerationTest.java

@@ -22,6 +22,7 @@ package de.monticore.lang.monticar.cnnarch.caffe2generator;
 
 import de.se_rwth.commons.logging.Log;
 import freemarker.template.TemplateException;
+import org.apache.commons.cli.ParseException;
 import org.junit.Before;
 import org.junit.Rule;
 import org.junit.Test;
@@ -31,6 +32,7 @@ import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.*;
 
+import org.junit.contrib.java.lang.system.Assertion;
 import org.junit.contrib.java.lang.system.ExpectedSystemExit;
 import static junit.framework.TestCase.assertTrue;
 
@@ -62,26 +64,33 @@ public class GenerationTest extends AbstractSymtabTest{
     }
 
     @Test
-    public void testCifar10Classifier() throws IOException, TemplateException {
+    public void testUnsupportedLayersCifar10Classifier() throws IOException, TemplateException {
         Log.getFindings().clear();
         String[] args = {"-m", "src/test/resources/valid_tests", "-r", "CifarClassifierNetwork", "-o", "./target/generated-sources-cnnarch/"};
         exit.expectSystemExit();
+        exit.checkAssertionAfterwards(new Assertion() {
+            public void checkAssertion() {
+                assertTrue(Log.getFindings().size() == 2);
+            }
+        });
         CNNArch2Caffe2Cli.main(args);
-        assertTrue(Log.getFindings().size() == 2);
     }
 
     @Test
-    public void testAlexnetGeneration() throws IOException, TemplateException {
+    public void testUnsupportedLayersAlexnet() throws IOException, TemplateException {
         Log.getFindings().clear();
         String[] args = {"-m", "src/test/resources/architectures", "-r", "Alexnet", "-o", "./target/generated-sources-cnnarch/"};
         exit.expectSystemExit();
+        exit.checkAssertionAfterwards(new Assertion() {
+            public void checkAssertion() {
+                assertTrue(Log.getFindings().size() == 2);
+            }
+        });
         CNNArch2Caffe2Cli.main(args);
-        assertTrue(Log.getFindings().size() == 2);
-
     }
 
     @Test
-    public void testGeneratorVGG16() throws IOException, TemplateException {
+    public void testGeneratorVGG16Generation() throws IOException, TemplateException {
         Log.getFindings().clear();
         String[] args = {"-m", "src/test/resources/architectures", "-r", "VGG16", "-o", "./target/generated-sources-cnnarch/"};
         CNNArch2Caffe2Cli.main(args);
@@ -102,17 +111,25 @@ public class GenerationTest extends AbstractSymtabTest{
         Log.getFindings().clear();
         String[] args = {"-m", "src/test/resources/architectures", "-r", "ThreeInputCNN_M14"};
         exit.expectSystemExit();
+        exit.checkAssertionAfterwards(new Assertion() {
+            public void checkAssertion() {
+                assertTrue(Log.getFindings().size() == 2);
+            }
+        });
         CNNArch2Caffe2Cli.main(args);
-        assertTrue(Log.getFindings().size() == 2);
     }
 
     @Test
-    public void testResNeXtGeneration() throws IOException, TemplateException {
+    public void testUnsupportedLayersResNeXt() throws IOException, TemplateException {
         Log.getFindings().clear();;
         String[] args = {"-m", "src/test/resources/architectures", "-r", "ResNeXt50"};
         exit.expectSystemExit();
+        exit.checkAssertionAfterwards(new Assertion() {
+            public void checkAssertion() {
+                assertTrue(Log.getFindings().size() == 2);
+            }
+        });
         CNNArch2Caffe2Cli.main(args);
-        assertTrue(Log.getFindings().size() == 2);
     }
 
     @Test
@@ -190,7 +207,9 @@ public class GenerationTest extends AbstractSymtabTest{
         String rootModelName = "alexnet";
         CNNArch2Caffe2 generator = new CNNArch2Caffe2();
         generator.setGenerationTargetPath("./target/generated-sources-cnnarch");
-        generator.generateCMake(rootModelName);
+        if(generator.isCMakeRequired()){
+            generator.generateCMake(rootModelName);
+        }
 
         assertTrue(Log.getFindings().isEmpty());
 
@@ -207,4 +226,17 @@ public class GenerationTest extends AbstractSymtabTest{
                         "FindArmadillo.cmake"));
     }
 
+    @Test
+    public void testWrongArgumentCNNArch2Caffe2Cli() throws ParseException {
+        Log.getFindings().clear();
+        String[] args = {"-x", "src/test/resources/architectures", "-y", "LeNet", "-z", "./target/generated-sources-cnnarch/"};
+        exit.expectSystemExit();
+        exit.checkAssertionAfterwards(new Assertion() {
+            public void checkAssertion() {
+                assertTrue(Log.getFindings().size() == 2);
+            }
+        });
+        CNNArch2Caffe2Cli.main(args);
+    }
+
 }