OffsetDateTimeMapperTest.java

package io.dropwizard.jdbi.args;

import org.junit.Test;
import org.mockito.Mockito;

import java.sql.ResultSet;
import java.sql.Timestamp;
import java.time.Instant;
import java.time.OffsetDateTime;
import java.time.ZoneId;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.when;

public class OffsetDateTimeMapperTest {

    private final ResultSet resultSet = Mockito.mock(ResultSet.class);

    @Test
    public void mapColumnByName() throws Exception {
        final Instant now = OffsetDateTime.now().toInstant();

        when(resultSet.getTimestamp("name")).thenReturn(Timestamp.from(now));

        OffsetDateTime actual = new OffsetDateTimeMapper().mapColumn(resultSet, "name", null);

        assertThat(actual).isEqualTo(OffsetDateTime.ofInstant(now, ZoneId.systemDefault()));
    }

    @Test
    public void mapColumnByName_TimestampIsNull() throws Exception {
        when(resultSet.getTimestamp("name")).thenReturn(null);

        OffsetDateTime actual = new OffsetDateTimeMapper().mapColumn(resultSet, "name", null);

        assertThat(actual).isNull();
    }

    @Test
    public void mapColumnByIndex() throws Exception {
        final Instant now = OffsetDateTime.now().toInstant();

        when(resultSet.getTimestamp(1)).thenReturn(Timestamp.from(now));

        OffsetDateTime actual = new OffsetDateTimeMapper().mapColumn(resultSet, 1, null);

        assertThat(actual).isEqualTo(OffsetDateTime.ofInstant(now, ZoneId.systemDefault()));
    }

    @Test
    public void mapColumnByIndex_TimestampIsNull() throws Exception {
        when(resultSet.getTimestamp(1)).thenReturn(null);

        OffsetDateTime actual = new OffsetDateTimeMapper().mapColumn(resultSet, 1, null);

        assertThat(actual).isNull();
    }
}