/*
 * Copyright (c) 2016-2017, Mihai Emil Andronache
 * All rights reserved.
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *  1)Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 *  2)Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *  3)Neither the name of charles-rest nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package com.amihaiemil.charles.aws;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.io.ByteArrayInputStream;
import java.net.HttpURLConnection;
import org.apache.commons.io.IOUtils;
import org.junit.Test;
import org.mockito.Mockito;
import com.amazonaws.AmazonServiceException;
import com.amazonaws.http.HttpResponse;

/**
 * Test cases for {@link SimpleAwsResponseHandler}
 * @author Mihai Andronache ([email protected])
 * @version $Id: 616903f73bc16b08368a804f9906cb2337c68b51 $
 * @since 1.0.0
 */
public class SimpleAwsResponseHandlerTestCase {

    /**
     * SimpleAwsResponseHandler returns the plain response as is, since the status is
     * in range [200, 300)
     * @throws Exception If something goes wrong.
     */
    @Test
    public void responseStatusInRange() throws Exception {
        HttpResponse response = Mockito.mock(HttpResponse.class);
        Mockito.when(response.getStatusCode()).thenReturn(HttpURLConnection.HTTP_OK);
        String content = "This content should not be affected!";
        Mockito.when(response.getContent()).thenReturn(
            new ByteArrayInputStream(content.getBytes())
        );
        SimpleAwsResponseHandler handler = new SimpleAwsResponseHandler(false);
        assertFalse(handler.needsConnectionLeftOpen());
        HttpResponse handled = handler.handle(response);
        assertTrue(handled.getStatusCode() == HttpURLConnection.HTTP_OK);
        String expected = new String(IOUtils.toByteArray(handled.getContent()));
        assertTrue(expected.equals(content));
    }

    /**
     * {@link SimpleAwsResponseHandler} can throw AmazonServiceException
     * in case the response status is not in the expected range.
     */
    @Test
    public void throwsAwsExceptionOnBadStatus() {
        HttpResponse response = Mockito.mock(HttpResponse.class);
        Mockito.when(response.getStatusCode()).thenReturn(HttpURLConnection.HTTP_BAD_METHOD);
        String content = "bad request message";
        Mockito.when(response.getContent())
            .thenReturn(
                new ByteArrayInputStream(content.getBytes())
            );
        try {
            new SimpleAwsResponseHandler(true).handle(response);
            fail("AmazonServiceException should have been thrown!");
        } catch (AmazonServiceException awsEx) {
            assertTrue(awsEx.getErrorMessage().equals(content));
            assertTrue(awsEx.getStatusCode() == HttpURLConnection.HTTP_BAD_METHOD);
        }
    }

}