Exception in missing tests view

The content provider doesn't deal with Java files not containing a type, like package-info.java
MoreUnit · Dec 11, 2023 · bd9a502 · bd9a502
1 parent bb5b328
commit bd9a502
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/org.moreunit.plugin/src/org/moreunit/elements/MissingClassTreeContentProvider.java b/org.moreunit.plugin/src/org/moreunit/elements/MissingClassTreeContentProvider.java
@@ -16,10 +16,10 @@
 
 public class MissingClassTreeContentProvider implements ITreeContentProvider
 {
-    
+
     public MissingClassTreeContentProvider()
     {
-        
+
     }
 
     public Object[] getChildren(Object arg0)
@@ -57,9 +57,14 @@ public Object[] getElements(Object inputElement)
                             ICompilationUnit[] compilationUnits = ((IPackageFragment) javaPackage).getCompilationUnits();
                             for (ICompilationUnit compilationUnit : compilationUnits)
                             {
-                                ClassTypeFacade classTypeFacade = new ClassTypeFacade(compilationUnit);
-                                if(!TypeFacade.isTestCase(compilationUnit) && !classTypeFacade.hasTestCase())
-                                    elements.add(compilationUnit);
+                                if(compilationUnit.findPrimaryType() != null)
+                                {
+                                    ClassTypeFacade classTypeFacade = new ClassTypeFacade(compilationUnit);
+                                    if(! TypeFacade.isTestCase(compilationUnit) && ! classTypeFacade.hasTestCase())
+                                    {
+                                        elements.add(compilationUnit);
+                                    }
+                                }
                             }
                         }
                     }