Fix test

Author: Hélios GILLES

org.moreunit.plugin/src/org/moreunit/elements/TestmethodCreator.java

@@ -3,6 +3,7 @@
 import static java.util.Collections.addAll;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 import java.util.ListIterator;
 
@@ -419,9 +420,18 @@ private IMethod createMethod(String methodName, String methodString, IMethod sib
     {
         if(doesMethodExist(methodName))
             return null;
-
         try
         {
+            String testImport = TestTypeConstants.TEST_ANNOTATION.get(testType);
+            if(testImport != null && Arrays.stream(testCaseCompilationUnit.getImports()).noneMatch(i -> i.getElementName().equals(testImport)))
+            {
+                testCaseCompilationUnit.createImport(testImport, null, null);
+            }
+            String staticImportBaseClass = TestTypeConstants.STATIC_IMPORT_BASE_CLASS.get(testType);
+            if(staticImportBaseClass != null && Arrays.stream(testCaseCompilationUnit.getImports()).noneMatch(i -> i.getElementName().equals(staticImportBaseClass)))
+            {
+                testCaseCompilationUnit.createImport(staticImportBaseClass, null, null);
+            }
             return testCaseCompilationUnit.findPrimaryType().createMethod(format(methodString), sibling, true, null);
         }
         catch (JavaModelException e)

org.moreunit.swtbot.test/src/org/moreunit/create/ClassCreationTest.java

@@ -48,11 +48,10 @@ public void should_create_test_with_default_package_for_junit5() throws JavaMode
             public boolean test() throws Exception
             {
                 try {
-                    context.getCompilationUnit("testing.TheWorldTest");
+                    return context.getCompilationUnit("testing.TheWorldTest").findPrimaryType().getMethods().length > 0;
                 } catch (IllegalArgumentException ex) {
-                    return false;
                 }
-                return true;
+                return false;
             }
 
             @Override