test: store in shipping-lists

Author: marcelogpinheiro

src/test/java/com/commercetools/sync/shoppinglists/utils/ShoppingListUpdateActionUtilsTest.java

@@ -17,7 +17,12 @@
 import com.commercetools.api.models.shopping_list.ShoppingListSetDeleteDaysAfterLastModificationAction;
 import com.commercetools.api.models.shopping_list.ShoppingListSetDescriptionAction;
 import com.commercetools.api.models.shopping_list.ShoppingListSetSlugAction;
+import com.commercetools.api.models.shopping_list.ShoppingListSetStoreAction;
 import com.commercetools.api.models.shopping_list.ShoppingListUpdateAction;
+import com.commercetools.api.models.store.StoreKeyReference;
+import com.commercetools.api.models.store.StoreKeyReferenceBuilder;
+import com.commercetools.api.models.store.StoreResourceIdentifier;
+import com.commercetools.api.models.store.StoreResourceIdentifierBuilder;
 import java.util.Locale;
 import java.util.Optional;
 import java.util.UUID;
@@ -351,4 +356,94 @@ void buildSetAnonymousId_WithSameValues_ShouldNotBuildUpdateActions() {
     assertThat(setDeleteDaysUpdateAction)
         .containsInstanceOf(ShoppingListSetDeleteDaysAfterLastModificationAction.class);
   }
+
+  @Test
+  void buildSetStoreUpdateAction_WithDifferentStoreKeys_ShouldBuildUpdateAction() {
+    final StoreKeyReference oldStoreReference =
+        StoreKeyReferenceBuilder.of().key("old-store-key").build();
+    final ShoppingList oldShoppingList = mock(ShoppingList.class);
+    when(oldShoppingList.getStore()).thenReturn(oldStoreReference);
+
+    final StoreResourceIdentifier newStoreIdentifier =
+        StoreResourceIdentifierBuilder.of().key("new-store-key").build();
+    final ShoppingListDraft newShoppingList = mock(ShoppingListDraft.class);
+    when(newShoppingList.getStore()).thenReturn(newStoreIdentifier);
+
+    final ShoppingListUpdateAction setStoreUpdateAction =
+        ShoppingListUpdateActionUtils.buildSetStoreUpdateAction(oldShoppingList, newShoppingList)
+            .orElse(null);
+
+    assertThat(setStoreUpdateAction).isNotNull();
+    assertThat(setStoreUpdateAction.getAction()).isEqualTo("setStore");
+    assertThat(((ShoppingListSetStoreAction) setStoreUpdateAction).getStore())
+        .isEqualTo(newStoreIdentifier);
+  }
+
+  @Test
+  void buildSetStoreUpdateAction_WithSameStoreKeys_ShouldNotBuildUpdateAction() {
+    final StoreKeyReference oldStoreReference =
+        StoreKeyReferenceBuilder.of().key("same-store-key").build();
+    final ShoppingList oldShoppingList = mock(ShoppingList.class);
+    when(oldShoppingList.getStore()).thenReturn(oldStoreReference);
+
+    final StoreResourceIdentifier newStoreIdentifier =
+        StoreResourceIdentifierBuilder.of().key("same-store-key").build();
+    final ShoppingListDraft newShoppingList = mock(ShoppingListDraft.class);
+    when(newShoppingList.getStore()).thenReturn(newStoreIdentifier);
+
+    final Optional<ShoppingListUpdateAction> setStoreUpdateAction =
+        ShoppingListUpdateActionUtils.buildSetStoreUpdateAction(oldShoppingList, newShoppingList);
+
+    assertThat(setStoreUpdateAction).isNotPresent();
+  }
+
+  @Test
+  void buildSetStoreUpdateAction_WithNullOldStore_ShouldBuildUpdateAction() {
+    final ShoppingList oldShoppingList = mock(ShoppingList.class);
+    when(oldShoppingList.getStore()).thenReturn(null);
+
+    final StoreResourceIdentifier newStoreIdentifier =
+        StoreResourceIdentifierBuilder.of().key("new-store-key").build();
+    final ShoppingListDraft newShoppingList = mock(ShoppingListDraft.class);
+    when(newShoppingList.getStore()).thenReturn(newStoreIdentifier);
+
+    final ShoppingListUpdateAction setStoreUpdateAction =
+        ShoppingListUpdateActionUtils.buildSetStoreUpdateAction(oldShoppingList, newShoppingList)
+            .orElse(null);
+
+    assertThat(setStoreUpdateAction).isNotNull();
+    assertThat(setStoreUpdateAction.getAction()).isEqualTo("setStore");
+  }
+
+  @Test
+  void buildSetStoreUpdateAction_WithNullNewStore_ShouldBuildUpdateAction() {
+    final StoreKeyReference oldStoreReference =
+        StoreKeyReferenceBuilder.of().key("old-store-key").build();
+    final ShoppingList oldShoppingList = mock(ShoppingList.class);
+    when(oldShoppingList.getStore()).thenReturn(oldStoreReference);
+
+    final ShoppingListDraft newShoppingList = mock(ShoppingListDraft.class);
+    when(newShoppingList.getStore()).thenReturn(null);
+
+    final ShoppingListUpdateAction setStoreUpdateAction =
+        ShoppingListUpdateActionUtils.buildSetStoreUpdateAction(oldShoppingList, newShoppingList)
+            .orElse(null);
+
+    assertThat(setStoreUpdateAction).isNotNull();
+    assertThat(setStoreUpdateAction.getAction()).isEqualTo("setStore");
+  }
+
+  @Test
+  void buildSetStoreUpdateAction_WithBothNull_ShouldNotBuildUpdateAction() {
+    final ShoppingList oldShoppingList = mock(ShoppingList.class);
+    when(oldShoppingList.getStore()).thenReturn(null);
+
+    final ShoppingListDraft newShoppingList = mock(ShoppingListDraft.class);
+    when(newShoppingList.getStore()).thenReturn(null);
+
+    final Optional<ShoppingListUpdateAction> setStoreUpdateAction =
+        ShoppingListUpdateActionUtils.buildSetStoreUpdateAction(oldShoppingList, newShoppingList);
+
+    assertThat(setStoreUpdateAction).isNotPresent();
+  }
 }

src/test/java/com/commercetools/sync/shoppinglists/utils/ShoppingListsReferenceResolutionUtilsTest.java

@@ -16,6 +16,8 @@
 import com.commercetools.api.models.shopping_list.ShoppingListLineItemDraftBuilder;
 import com.commercetools.api.models.shopping_list.TextLineItem;
 import com.commercetools.api.models.shopping_list.TextLineItemDraftBuilder;
+import com.commercetools.api.models.store.StoreKeyReference;
+import com.commercetools.api.models.store.StoreKeyReferenceBuilder;
 import com.commercetools.api.models.type.CustomFields;
 import com.commercetools.api.models.type.CustomFieldsDraftBuilder;
 import com.commercetools.api.models.type.TypeReference;
@@ -225,4 +227,49 @@ void mapToShoppingListDrafts_WithOtherFields_ShouldReturnDraftsCorrectly() {
                 .anonymousId("anonymousId")
                 .build());
   }
+
+  @Test
+  void mapToShoppingListDrafts_WithStoreReference_ShouldReturnDraftsWithStoreKey() {
+    final String storeKey = "store-key";
+
+    final ShoppingList mockShoppingList = mock(ShoppingList.class);
+    when(mockShoppingList.getName()).thenReturn(ofEnglish("name"));
+    when(mockShoppingList.getKey()).thenReturn("key");
+
+    final StoreKeyReference storeKeyReference = StoreKeyReferenceBuilder.of().key(storeKey).build();
+    when(mockShoppingList.getStore()).thenReturn(storeKeyReference);
+
+    when(mockShoppingList.getCustomer()).thenReturn(null);
+    when(mockShoppingList.getCustom()).thenReturn(null);
+    when(mockShoppingList.getLineItems()).thenReturn(null);
+    when(mockShoppingList.getTextLineItems()).thenReturn(null);
+
+    final List<ShoppingListDraft> shoppingListDrafts =
+        ShoppingListReferenceResolutionUtils.mapToShoppingListDrafts(
+            singletonList(mockShoppingList), referenceIdToKeyCache);
+
+    assertThat(shoppingListDrafts).hasSize(1);
+    assertThat(shoppingListDrafts.get(0).getStore()).isNotNull();
+    assertThat(shoppingListDrafts.get(0).getStore().getKey()).isEqualTo(storeKey);
+  }
+
+  @Test
+  void mapToShoppingListDrafts_WithNullStore_ShouldReturnDraftsWithNullStore() {
+    final ShoppingList mockShoppingList = mock(ShoppingList.class);
+    when(mockShoppingList.getName()).thenReturn(ofEnglish("name"));
+    when(mockShoppingList.getKey()).thenReturn("key");
+    when(mockShoppingList.getStore()).thenReturn(null);
+
+    when(mockShoppingList.getCustomer()).thenReturn(null);
+    when(mockShoppingList.getCustom()).thenReturn(null);
+    when(mockShoppingList.getLineItems()).thenReturn(null);
+    when(mockShoppingList.getTextLineItems()).thenReturn(null);
+
+    final List<ShoppingListDraft> shoppingListDrafts =
+        ShoppingListReferenceResolutionUtils.mapToShoppingListDrafts(
+            singletonList(mockShoppingList), referenceIdToKeyCache);
+
+    assertThat(shoppingListDrafts).hasSize(1);
+    assertThat(shoppingListDrafts.get(0).getStore()).isNull();
+  }
 }