test: additional addresss and state

Author: marcelogpinheiro

src/test/java/com/commercetools/sync/customers/utils/AddressUpdateActionUtilsTest.java

@@ -529,6 +529,121 @@ void buildChangeAddressUpdateActions_WithNullAddresses_ShouldNotReturnChangeAddr
     assertThat(updateActions).isEmpty();
   }
 
+  @Test
+  void buildChangeAddressUpdateActions_WithDifferentState_ShouldReturnChangeAddressAction() {
+
+    when(oldCustomer.getAddresses())
+        .thenReturn(
+            singletonList(
+                AddressBuilder.of()
+                    .country(CountryCode.US.toString())
+                    .key("address-key-1")
+                    .id("address-id-1")
+                    .state("California")
+                    .build()));
+
+    final AddressDraft address1 =
+        AddressDraftBuilder.of()
+            .country(CountryCode.US.toString())
+            .key("address-key-1")
+            .state("Texas")
+            .id("address-id-new-1")
+            .build();
+
+    final CustomerDraft newCustomer =
+        CustomerDraftBuilder.of()
+            .email("email")
+            .password("pass")
+            .addresses(singletonList(address1))
+            .build();
+
+    final List<CustomerUpdateAction> updateActions =
+        CustomerUpdateActionUtils.buildChangeAddressUpdateActions(oldCustomer, newCustomer);
+
+    assertThat(updateActions)
+        .containsExactly(
+            CustomerChangeAddressActionBuilder.of()
+                .addressId("address-id-1")
+                .address(address1)
+                .build());
+  }
+
+  @Test
+  void
+      buildChangeAddressUpdateActions_WithDifferentAdditionalStreetInfo_ShouldReturnChangeAddressAction() {
+
+    when(oldCustomer.getAddresses())
+        .thenReturn(
+            singletonList(
+                AddressBuilder.of()
+                    .country(CountryCode.US.toString())
+                    .key("address-key-1")
+                    .id("address-id-1")
+                    .additionalStreetInfo("Suite 100")
+                    .build()));
+
+    final AddressDraft address1 =
+        AddressDraftBuilder.of()
+            .country(CountryCode.US.toString())
+            .key("address-key-1")
+            .additionalStreetInfo("Suite 200")
+            .id("address-id-new-1")
+            .build();
+
+    final CustomerDraft newCustomer =
+        CustomerDraftBuilder.of()
+            .email("email")
+            .password("pass")
+            .addresses(singletonList(address1))
+            .build();
+
+    final List<CustomerUpdateAction> updateActions =
+        CustomerUpdateActionUtils.buildChangeAddressUpdateActions(oldCustomer, newCustomer);
+
+    assertThat(updateActions)
+        .containsExactly(
+            CustomerChangeAddressActionBuilder.of()
+                .addressId("address-id-1")
+                .address(address1)
+                .build());
+  }
+
+  @Test
+  void
+      buildChangeAddressUpdateActions_WithSameStateAndAdditionalStreetInfo_ShouldNotReturnAction() {
+
+    when(oldCustomer.getAddresses())
+        .thenReturn(
+            singletonList(
+                AddressBuilder.of()
+                    .country(CountryCode.US.toString())
+                    .key("address-key-1")
+                    .id("address-id-1")
+                    .state("California")
+                    .additionalStreetInfo("Suite 100")
+                    .build()));
+
+    final CustomerDraft newCustomer =
+        CustomerDraftBuilder.of()
+            .email("email")
+            .password("pass")
+            .addresses(
+                singletonList(
+                    AddressBuilder.of()
+                        .country(CountryCode.US.toString())
+                        .key("address-key-1")
+                        .id("address-id-new-1")
+                        .state("California")
+                        .additionalStreetInfo("Suite 100")
+                        .build()))
+            .build();
+
+    final List<CustomerUpdateAction> updateActions =
+        CustomerUpdateActionUtils.buildChangeAddressUpdateActions(oldCustomer, newCustomer);
+
+    assertThat(updateActions).isEmpty();
+  }
+
   @Test
   void
       buildChangeAddressUpdateActions_WithAddressesWithoutKeys_ShouldNotReturnChangeAddressActions() {

src/test/java/com/commercetools/sync/customers/utils/CustomerReferenceResolutionUtilsTest.java

@@ -212,4 +212,69 @@ void mapToCustomerDrafts_WithAddresses_ShouldReturnResourceIdentifiersWithCorrec
     assertThat(customerDraft.getBillingAddresses()).isEqualTo(asList(0, 2));
     assertThat(customerDraft.getShippingAddresses()).isEmpty();
   }
+
+  @Test
+  void mapToCustomerDrafts_WithStateAndAdditionalStreetInfo_ShouldMapFieldsCorrectly() {
+    final Customer mockCustomer = mock(Customer.class);
+    when(mockCustomer.getEmail()).thenReturn("email");
+    when(mockCustomer.getPassword()).thenReturn("password");
+    when(mockCustomer.getAddresses())
+        .thenReturn(
+            asList(
+                AddressBuilder.of()
+                    .country(CountryCode.US.toString())
+                    .id("address-id1")
+                    .key("address-key1")
+                    .state("California")
+                    .additionalStreetInfo("Suite 100")
+                    .build(),
+                AddressBuilder.of()
+                    .country(CountryCode.DE.toString())
+                    .id("address-id2")
+                    .key("address-key2")
+                    .state("Bavaria")
+                    .additionalStreetInfo("Building A")
+                    .build()));
+
+    final List<CustomerDraft> referenceReplacedDrafts =
+        CustomerReferenceResolutionUtils.mapToCustomerDrafts(
+            singletonList(mockCustomer), referenceIdToKeyCache);
+
+    final CustomerDraft customerDraft = referenceReplacedDrafts.get(0);
+
+    assertThat(customerDraft.getAddresses()).hasSize(2);
+    assertThat(customerDraft.getAddresses().get(0).getState()).isEqualTo("California");
+    assertThat(customerDraft.getAddresses().get(0).getAdditionalStreetInfo())
+        .isEqualTo("Suite 100");
+    assertThat(customerDraft.getAddresses().get(1).getState()).isEqualTo("Bavaria");
+    assertThat(customerDraft.getAddresses().get(1).getAdditionalStreetInfo())
+        .isEqualTo("Building A");
+  }
+
+  @Test
+  void mapToCustomerDrafts_WithNullStateAndAdditionalStreetInfo_ShouldMapFieldsAsNull() {
+    final Customer mockCustomer = mock(Customer.class);
+    when(mockCustomer.getEmail()).thenReturn("email");
+    when(mockCustomer.getPassword()).thenReturn("password");
+    when(mockCustomer.getAddresses())
+        .thenReturn(
+            singletonList(
+                AddressBuilder.of()
+                    .country(CountryCode.US.toString())
+                    .id("address-id1")
+                    .key("address-key1")
+                    .state(null)
+                    .additionalStreetInfo(null)
+                    .build()));
+
+    final List<CustomerDraft> referenceReplacedDrafts =
+        CustomerReferenceResolutionUtils.mapToCustomerDrafts(
+            singletonList(mockCustomer), referenceIdToKeyCache);
+
+    final CustomerDraft customerDraft = referenceReplacedDrafts.get(0);
+
+    assertThat(customerDraft.getAddresses()).hasSize(1);
+    assertThat(customerDraft.getAddresses().get(0).getState()).isNull();
+    assertThat(customerDraft.getAddresses().get(0).getAdditionalStreetInfo()).isNull();
+  }
 }