feat: add-back and re-factor secondary code check

bcgov · Sep 15, 2023 · a770708 · a770708
1 parent cf535b5
commit a770708
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/api/src/test/java/ca/bc/gov/educ/api/edx/service/EdxUsersServiceTests.java b/api/src/test/java/ca/bc/gov/educ/api/edx/service/EdxUsersServiceTests.java
@@ -162,6 +162,30 @@ void findPrimaryEdxActivationCodeOutOfManyForDistrict() {
     assertThat(found.getIsPrimary()).isTrue();
   }
 
+  @Test
+  void findPrimaryEdxActivationCodeOnlyReturnsPrimaryEdxActivationCodeForSchool() {
+    EdxActivationCodeEntity mockActivationCode = this.createEdxActivationCodeEntity(
+      UUID.randomUUID().toString(), false, true, UUID.randomUUID(), 0, UUID.randomUUID(), null);
+    EdxActivationCodeEntity secondaryActivationCode = this.edxActivationCodeRepository.save(mockActivationCode);
+
+    assertThrows(EntityNotFoundException.class, () -> {
+      this.service
+        .findPrimaryEdxActivationCode(InstituteTypeCode.SCHOOL, secondaryActivationCode.getSchoolID().toString());
+    });
+  }
+
+  @Test
+  void findPrimaryEdxActivationCodeOnlyReturnsPrimaryEdxActivationCodeForDistrict() {
+    EdxActivationCodeEntity mockActivationCode = this.createEdxActivationCodeEntity(
+      UUID.randomUUID().toString(), false, true, UUID.randomUUID(), 0, null, UUID.randomUUID());
+    EdxActivationCodeEntity secondaryActivationCode = this.edxActivationCodeRepository.save(mockActivationCode);
+
+    assertThrows(EntityNotFoundException.class, () -> {
+      this.service
+        .findPrimaryEdxActivationCode(InstituteTypeCode.DISTRICT, secondaryActivationCode.getDistrictID().toString());
+    });
+  }
+
   @Test
   void findPrimaryEdxActivationCodeCannotFindNonExistingEdxActivationCodeForSchool() {
     String uuid = UUID.randomUUID().toString();