modelmapper-115

import static org.junit.Assert.*;

import java.util.Arrays;

import org.junit.Test;
import org.modelmapper.ModelMapper;
import org.modelmapper.PropertyMap;

public class ModelmapperTest {

    ModelMapper mapper = new ModelMapper();

    static class A {
        private String title;
        private byte[] someBytes;
        public String getTitle() {
            return title;
        }
        public void setTitle(String title) {
            this.title = title;
        }
        public byte[] getSomeBytes() {
            return someBytes;
        }
        public void setSomeBytes(byte[] someBytes) {
            this.someBytes = someBytes;
        }
    }

    static class B {
        private String title;
        private byte[] someBytes;

        public String getTitle() {
            return title;
        }
        public void setTitle(String title) {
            this.title = title;
        }
        public byte[] getSomeBytes() {
            return someBytes;
        }
        public void setSomeBytes(byte[] someBytes) {
            this.someBytes = someBytes;
        }
    }

    @Test
    public void testOk() {
        A input = new A();
        input.title = "ABCDEF";
        input.someBytes = input.title.getBytes();
        
        B result = mapper.map(input, B.class);
        
        assertEquals(input.title, result.title);
        assertTrue(Arrays.equals(input.someBytes, result.someBytes));
    }

    @Test
    public void testFailing() {
        // fails right here:
        mapper.addMappings(new PropertyMap<A, B>() {
            @Override
            protected void configure() {
                skip().setSomeBytes(null);
            }
        });

        A input = new A();
        input.title = "ABCDEF";
        input.someBytes = input.title.getBytes();
        
        B result = mapper.map(input, B.class);
        
        assertEquals(input.title, result.title);
        assertNull(result.someBytes);
    }
}