Suggestion : custom map function about remap HOT 5 CLOSED

import static org.junit.Assert.assertEquals;

import org.junit.Test;

import com.remondis.remap.AssertMapping;
import com.remondis.remap.Mapper;
import com.remondis.remap.Mapping;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

public class MapperTest {

  @NoArgsConstructor
  @AllArgsConstructor
  @Data
  @Builder
  public static class AccountEntity {
    private BillingAddress billingAddress;
  }

  @NoArgsConstructor
  @AllArgsConstructor
  @Data
  @Builder
  public static class BillingAddress {
    private String street;
    private String city;
    private String zip;
    private String state;
    private String country;
  }

  @NoArgsConstructor
  @AllArgsConstructor
  @Data
  @Builder
  public static class SfAccount {
    private String billingStreet;
    private String billingCity;
    private String billingPostalCode;
    private String billingState;
    private String billingCountry;

  }

  @Test
  public void multiMapping() {
    Mapper<AccountEntity, SfAccount> mapper = Mapping.from(AccountEntity.class)
        .to(SfAccount.class)
        .replace(AccountEntity::getBillingAddress, SfAccount::getBillingStreet)
        .withSkipWhenNull(BillingAddress::getStreet)
        .replace(AccountEntity::getBillingAddress, SfAccount::getBillingCity)
        .withSkipWhenNull(BillingAddress::getCity)
        .replace(AccountEntity::getBillingAddress, SfAccount::getBillingPostalCode)
        .withSkipWhenNull(BillingAddress::getZip)
        .replace(AccountEntity::getBillingAddress, SfAccount::getBillingState)
        .withSkipWhenNull(BillingAddress::getState)
        .replace(AccountEntity::getBillingAddress, SfAccount::getBillingCountry)
        .withSkipWhenNull(BillingAddress::getCountry)
        .mapper();
    BillingAddress billingAddress = BillingAddress.builder()
        .city("city")
        .country("country")
        .state("state")
        .street("street")
        .zip("zip")
        .build();
    AccountEntity account = AccountEntity.builder()
        .billingAddress(billingAddress)
        .build();
    SfAccount sfaccount = mapper.map(account);
    assertEquals(billingAddress.getStreet(), sfaccount.getBillingStreet());
    assertEquals(billingAddress.getCity(), sfaccount.getBillingCity());
    assertEquals(billingAddress.getZip(), sfaccount.getBillingPostalCode());
    assertEquals(billingAddress.getState(), sfaccount.getBillingState());
    assertEquals(billingAddress.getCountry(), sfaccount.getBillingCountry());

    // The following would be the assert for your tests:
    AssertMapping.of(mapper)
        .expectReplace(AccountEntity::getBillingAddress, SfAccount::getBillingStreet)
        .andSkipWhenNull()
        .expectReplace(AccountEntity::getBillingAddress, SfAccount::getBillingCity)
        .andSkipWhenNull()
        .expectReplace(AccountEntity::getBillingAddress, SfAccount::getBillingPostalCode)
        .andSkipWhenNull()
        .expectReplace(AccountEntity::getBillingAddress, SfAccount::getBillingState)
        .andSkipWhenNull()
        .expectReplace(AccountEntity::getBillingAddress, SfAccount::getBillingCountry)
        .andSkipWhenNull()
        .ensure();
  }

}

        .replace(AccountEntity::getBillingAddress, SfAccount::getBillingLatitude)
        .withSkipWhenNull(billingAddress -> billingAddress.getLocation()[0])
        .replace(AccountEntity::getBillingAddress, SfAccount::getBillingLongitude)
        .withSkipWhenNull(billingAddress -> billingAddress.getLocation()[1])