[Enhancement kbss-cvut/termit-ui#520] Fix removePublishesEventAndDrop…

…sGraph test
lukaskabc · Nov 5, 2024 · cb5c227 · cb5c227
1 parent bd29968
commit cb5c227
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/test/java/cz/cvut/kbss/termit/persistence/dao/VocabularyDaoTest.java b/src/test/java/cz/cvut/kbss/termit/persistence/dao/VocabularyDaoTest.java
@@ -30,6 +30,7 @@
 import cz.cvut.kbss.termit.event.AssetPersistEvent;
 import cz.cvut.kbss.termit.event.AssetUpdateEvent;
 import cz.cvut.kbss.termit.event.RefreshLastModifiedEvent;
+import cz.cvut.kbss.termit.event.VocabularyEvent;
 import cz.cvut.kbss.termit.event.VocabularyWillBeRemovedEvent;
 import cz.cvut.kbss.termit.model.Glossary;
 import cz.cvut.kbss.termit.model.Model;
@@ -761,10 +762,14 @@ void removePublishesEventAndDropsGraph() {
 
         transactional(() -> sut.remove(vocabulary));
 
-        ArgumentCaptor<VocabularyWillBeRemovedEvent> eventCaptor = ArgumentCaptor.forClass(VocabularyWillBeRemovedEvent.class);
-        verify(eventPublisher).publishEvent(eventCaptor.capture());
+        ArgumentCaptor<VocabularyEvent> eventCaptor = ArgumentCaptor.forClass(VocabularyWillBeRemovedEvent.class);
+        verify(eventPublisher, atLeastOnce()).publishEvent(eventCaptor.capture());
 
-        VocabularyWillBeRemovedEvent event = eventCaptor.getValue();
+        VocabularyWillBeRemovedEvent event = (VocabularyWillBeRemovedEvent) eventCaptor
+                .getAllValues().stream()
+                .filter(e -> e instanceof VocabularyWillBeRemovedEvent)
+                .findAny().orElseThrow();
+
         assertNotNull(event);
 
         assertEquals(event.getVocabularyIri(), vocabulary.getUri());