update unit tests

crowdcode-de · Oct 17, 2013 · 825b58f · 825b58f
1 parent 5045205
commit 825b58f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/...mple-java/src/test/java/de/crowdcode/kissmda/cartridges/simplejava/EnumGeneratorTest.java b/...mple-java/src/test/java/de/crowdcode/kissmda/cartridges/simplejava/EnumGeneratorTest.java
@@ -19,6 +19,7 @@
 package de.crowdcode.kissmda.cartridges.simplejava;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
@@ -85,8 +86,12 @@ public void testGeneratePackage() {
 
 		enumGenerator.generatePackage(clazz, ast, cu);
 
-		assertEquals("package de.crowdcode.kissmda.testapp.components;\n",
-				cu.toString());
+		String onlyPackage = cu.toString();
+		String expectedResult = "package de.crowdcode.kissmda.testapp.components;\n";
+
+		boolean isInside = onlyPackage.contains(expectedResult);
+
+		assertTrue(isInside);
 	}
 
 	@Test

diff --git a/...java/src/test/java/de/crowdcode/kissmda/cartridges/simplejava/InterfaceGeneratorTest.java b/...java/src/test/java/de/crowdcode/kissmda/cartridges/simplejava/InterfaceGeneratorTest.java
@@ -19,6 +19,7 @@
 package de.crowdcode.kissmda.cartridges.simplejava;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
@@ -93,8 +94,12 @@ public void testGeneratePackage() {
 
 		interfaceGenerator.generatePackage(clazz, ast, cu);
 
-		assertEquals("package de.crowdcode.kissmda.testapp.components;\n",
-				cu.toString());
+		String onlyPackage = cu.toString();
+		String expectedResult = "package de.crowdcode.kissmda.testapp.components;\n";
+
+		boolean isInside = onlyPackage.contains(expectedResult);
+
+		assertTrue(isInside);
 	}
 
 	@SuppressWarnings("unchecked")