#7: Implement fallback for empty ExtensionContext

src/main/java/rocks/limburg/cdimock/CdiMockExtension.java

@@ -53,21 +53,27 @@
 public class CdiMockExtension implements Extension {
 
     private static ThreadLocal<ExtensionContext> beforeAllContext = new ThreadLocal<ExtensionContext>();
+    private static ExtensionContext latestExtensionContext;
 
     private Set<Class<?>> excludedClasses;
     private Set<Class<? extends Annotation>> excludingAnnotationTypes;
     private Map<Class<?>, Object> mockitoBeans;
 
     public static void beforeAll(ExtensionContext context) {
         beforeAllContext.set(context);
+        latestExtensionContext = context;
     }
 
     public static void afterAll(ExtensionContext context) {
+        ExtensionContext beforeAllContext = beforeAllContext.get();
         beforeAllContext.remove();
+        if (beforeAllContext == latestExtensionContext) {
+            latestExtensionContext = null;
+        }
     }
 
     public Optional<ExtensionContext> getExtensionContext() {
-        return ofNullable(beforeAllContext.get());
+        return ofNullable(beforeAllContext.get()).or(() -> ofNullable(latestExtensionContext));
     }
 
     public void excludeClasses(@Observes ProcessAnnotatedType<?> event) {