Improve previous TestPreviewChangesCoD fix

Evolveum · Apr 16, 2024 · 22cb18a · 22cb18a
1 parent 18057f1
commit 22cb18a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/...st/src/test/java/com/evolveum/midpoint/model/intest/simulation/TestPreviewChangesCoD.java b/...st/src/test/java/com/evolveum/midpoint/model/intest/simulation/TestPreviewChangesCoD.java
@@ -21,8 +21,6 @@
 
 import com.evolveum.midpoint.test.TestObject;
 
-import com.evolveum.midpoint.util.exception.CommonException;
-
 import org.springframework.test.annotation.DirtiesContext;
 import org.springframework.test.context.ContextConfiguration;
 import org.testng.AssertJUnit;
@@ -449,8 +447,8 @@ private Map<Class<? extends ObjectType>, Integer> collectCounts(Task task, Opera
                 continue;
             }
 
-            if (!isNativeRepository() && PolicyType.class.isAssignableFrom(clazz)) {
-                continue; // FIXME temporary hack to make this test pass on generic repository
+            if (!repositoryService.supports(clazz)) {
+                continue;
             }
 
             int count = repositoryService.countObjects(clazz, null, null, result);