Test that assertion generation is allowed in the initial and PARTIAL_…

…AUDIT_INFO_SET states, and not in any other states.
DemocracyDevelopers · Aug 23, 2024 · a92a738 · a92a738
1 parent 55f48d7
commit a92a738
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 8 deletions.
diff --git a/...t/src/test/java/au/org/democracydevelopers/corla/endpoint/GenerateAssertionsAPITests.java b/...t/src/test/java/au/org/democracydevelopers/corla/endpoint/GenerateAssertionsAPITests.java
@@ -48,7 +48,6 @@
 import us.freeandfair.corla.model.AuditReason;
 import us.freeandfair.corla.model.Choice;
 import us.freeandfair.corla.model.ContestResult;
-import us.freeandfair.corla.model.DoSDashboard;
 import us.freeandfair.corla.persistence.Persistence;
 
 import javax.transaction.Transactional;
@@ -186,8 +185,8 @@ public void closeMocks() {
    */
   @Test
   @Transactional
-  void assertionGenerationBlockedWhenInWrongASMStateOrComparisonAuditsPresent() {
-    testUtils.log(LOGGER, "assertionGenerationBlockedWhenInWrongASMStateOrComparisonAuditsPresent");
+  void assertionGenerationBlockedWhenInWrongASMState() {
+    testUtils.log(LOGGER, "assertionGenerationBlockedWhenInWrongASMState");
 
     // Mock the main class; mock its auth as the mocked state admin auth.
     try (MockedStatic<Main> mockedMain = Mockito.mockStatic(Main.class);

diff --git a/server/eclipse-project/src/test/java/au/org/democracydevelopers/corla/util/testUtils.java b/server/eclipse-project/src/test/java/au/org/democracydevelopers/corla/util/testUtils.java
@@ -22,12 +22,10 @@
 package au.org.democracydevelopers.corla.util;
 import au.org.democracydevelopers.corla.model.ContestType;
 import org.apache.log4j.Logger;
-import us.freeandfair.corla.model.Choice;
-import us.freeandfair.corla.model.Contest;
-import us.freeandfair.corla.model.ContestResult;
-import us.freeandfair.corla.model.County;
+import us.freeandfair.corla.model.*;
 
-import java.util.List;
+import java.math.BigDecimal;
+import java.util.*;
 
 /**
  * This class contains utilities and default data for use in testing.