ErrorResponseBuilderTest.java

package io.katharsis.errorhandling;

import org.junit.Test;

import static org.assertj.core.api.Assertions.assertThat;

public class ErrorResponseBuilderTest {

    private static final int STATUS = 500;

    @Test
    public void shouldSetStatus() throws Exception {
        ErrorResponse response = ErrorResponse.builder()
                .setStatus(STATUS)
                .build();

        assertThat(response.getHttpStatus()).isEqualTo(STATUS);
    }

    @Test
    public void shouldSetSingleErrorData() throws Exception {
        ErrorResponse response = ErrorResponse.builder()
                .setSingleErrorData(ErrorDataMother.fullyPopulatedErrorData())
                .build();

        assertThat((Iterable<?>) response.getResponse().getEntity())
                .hasSize(1)
                .containsExactly(ErrorDataMother.fullyPopulatedErrorData());
    }

    @Test
    public void shouldSetErrorDataCollection() throws Exception {
        ErrorResponse response = ErrorResponse.builder()
                .setErrorData(ErrorDataMother.oneSizeCollectionOfErrorData())
                .build();

        assertThat((Iterable<?>) response.getResponse().getEntity())
                .hasSize(1)
                .containsExactly(ErrorDataMother.fullyPopulatedErrorData());
    }
}