Increase coverage

commercetools · Nov 29, 2023 · 68a4953 · 68a4953
1 parent 07908ac
commit 68a4953
Showing 1 changed file with 21 additions and 5 deletions.
diff --git a/src/test/java/com/commercetools/sync/states/utils/StateReferenceResolutionUtilsTest.java b/src/test/java/com/commercetools/sync/states/utils/StateReferenceResolutionUtilsTest.java
@@ -126,11 +126,27 @@ void mapToStateDrafts_WithMissingRequiredFields_ShouldNotFailAndReturnEmptyDraft
     final State mockState = mock(State.class);
     when(mockState.getTransitions()).thenReturn(null);
 
-    // test
-    final List<StateDraft> referenceReplacedDrafts =
-        StateReferenceResolutionUtils.mapToStateDrafts(List.of(mockState), referenceIdToKeyCache);
-
-    assertThat(referenceReplacedDrafts.get(0)).isEqualTo(StateDraft.of());
+    // asserts
+    assertThat(
+            StateReferenceResolutionUtils.mapToStateDrafts(
+                    List.of(mockState), referenceIdToKeyCache)
+                .get(0))
+        .isEqualTo(StateDraft.of());
+
+    when(mockState.getKey()).thenReturn("Any key");
+    assertThat(
+            StateReferenceResolutionUtils.mapToStateDrafts(
+                    List.of(mockState), referenceIdToKeyCache)
+                .get(0))
+        .isEqualTo(StateDraft.of());
+
+    when(mockState.getKey()).thenReturn(null);
+    when(mockState.getType()).thenReturn(StateTypeEnum.LINE_ITEM_STATE);
+    assertThat(
+            StateReferenceResolutionUtils.mapToStateDrafts(
+                    List.of(mockState), referenceIdToKeyCache)
+                .get(0))
+        .isEqualTo(StateDraft.of());
   }
 
   @Nonnull