Simplify tests for /RegistryDataFormConfigurationMappingCustomizer (u…

…se containsEntry)
croz-ltd · Jul 13, 2022 · 804040c · 804040c
1 parent eb96cad
commit 804040c
Showing 1 changed file with 5 additions and 20 deletions.
diff --git a/...oz/nrich/registry/data/customizer/RegistryDataFormConfigurationMappingCustomizerTest.java b/...oz/nrich/registry/data/customizer/RegistryDataFormConfigurationMappingCustomizerTest.java
@@ -51,11 +51,8 @@ void shouldResolveRegistryFormConfigurationForEntityWithoutCreateRequest() {
         // when
         formConfigurationMappingCustomizer.customizeConfigurationMapping(formConfiguration);
 
-        // and when
-        Class<?> registryCreateClass = formConfiguration.get(formId);
-
         // then
-        assertThat(registryCreateClass).isEqualTo(RegistryTestEntity.class);
+        assertThat(formConfiguration).containsEntry(formId, RegistryTestEntity.class);
     }
 
     @Test
@@ -68,11 +65,8 @@ void shouldResolveRegistryFormConfigurationForEntityWithoutUpdateRequest() {
         // when
         formConfigurationMappingCustomizer.customizeConfigurationMapping(formConfiguration);
 
-        // and when
-        Class<?> registryCreateClass = formConfiguration.get(formId);
-
         // then
-        assertThat(registryCreateClass).isEqualTo(RegistryTestEntityUpdateRequest.class);
+        assertThat(formConfiguration).containsEntry(formId, RegistryTestEntityUpdateRequest.class);
     }
 
     @Test
@@ -85,11 +79,8 @@ void shouldNotAddFormConfigurationForExistingCreateRequest() {
         // when
         formConfigurationMappingCustomizer.customizeConfigurationMapping(formConfiguration);
 
-        // and when
-        Class<?> registryCreateClass = formConfiguration.get(formId);
-
         // then
-        assertThat(registryCreateClass).isEqualTo(Object.class);
+        assertThat(formConfiguration).containsEntry(formId, Object.class);
     }
 
     @Test
@@ -102,11 +93,8 @@ void shouldNotAddFormConfigurationForExistingUpdateRequest() {
         // when
         formConfigurationMappingCustomizer.customizeConfigurationMapping(formConfiguration);
 
-        // and when
-        Class<?> registryCreateClass = formConfiguration.get(formId);
-
         // then
-        assertThat(registryCreateClass).isEqualTo(Object.class);
+        assertThat(formConfiguration).containsEntry(formId, Object.class);
     }
 
     @Test
@@ -118,10 +106,7 @@ void shouldLoadClassFromRequestPackage() {
         // when
         formConfigurationMappingCustomizer.customizeConfigurationMapping(formConfiguration);
 
-        // and when
-        Class<?> registryCreateClass = formConfiguration.get(formId);
-
         // then
-        assertThat(registryCreateClass).isEqualTo(RegistryCustomizerTestEntityCreateRequest.class);
+        assertThat(formConfiguration).containsEntry(formId, RegistryCustomizerTestEntityCreateRequest.class);
     }
 }