add wallet mode removal test

admin-ch · Feb 11, 2022 · 8acf752 · 8acf752
1 parent 8fe4bba
commit 8acf752
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/...ag/covidcertificate/backend/verifier/ws/controller/VerificationRulesControllerV2Test.java b/...ag/covidcertificate/backend/verifier/ws/controller/VerificationRulesControllerV2Test.java
@@ -72,6 +72,7 @@ public void verificationRulesTest() throws Exception {
                 boolean modeExists = false;
                 var iter = expected.get("modeRules").get("activeModes").iterator();
                 var verifierIter = expected.get("modeRules").get("verifierActiveModes").iterator();
+                var walletIter = expected.get("modeRules").get("walletActiveModes").iterator();
                 while (iter.hasNext()) {
                     JsonNode mode = iter.next();
                     if (disabledMode.equals(mode.get("id").asText())) {
@@ -86,6 +87,13 @@ public void verificationRulesTest() throws Exception {
                         verifierIter.remove();
                     }
                 }
+                while (walletIter.hasNext()) {
+                    JsonNode mode = walletIter.next();
+                    if (disabledMode.equals(mode.get("id").asText())) {
+                        modeExists = true;
+                        walletIter.remove();
+                    }
+                }
                 if (!modeExists) {
                     throw new IllegalArgumentException(
                             "JSON doesn't seem to have the mode we're testing for. Edit test case or JSON");