/*
 * SoapUI, Copyright (C) 2004-2016 SmartBear Software 
 *
 * Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent 
 * versions of the EUPL (the "Licence"); 
 * You may not use this work except in compliance with the Licence. 
 * You may obtain a copy of the Licence at: 
 * 
 * http://ec.europa.eu/idabc/eupl 
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the Licence is 
 * distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 
 * express or implied. See the Licence for the specific language governing permissions and limitations 
 * under the Licence. 
 */

package com.eviware.soapui.impl.rest.mock;

import com.eviware.soapui.impl.wsdl.mock.WsdlMockRunContext;
import com.eviware.soapui.model.mock.MockResult;
import com.eviware.soapui.utils.ModelItemFactory;
import org.apache.commons.httpclient.HttpStatus;
import org.junit.Before;
import org.junit.Test;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import static com.eviware.soapui.utils.MockedServlet.mockHttpServletRequest;
import static com.eviware.soapui.utils.MockedServlet.mockHttpServletResponse;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.anyString;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;

public class RestMockResponseWriterTest {

    private RestMockResponse mockResponse;
    private MockResult result;
    private String originalResponseContent = "<content>awful lot of content</content>";;
    private HttpServletRequest servletRequest;
    private HttpServletResponse servletResponse;

    @Before
    public void setUp() throws Exception {
        servletRequest = mockHttpServletRequest();
        servletResponse = mockHttpServletResponse();
        mockResponse = ModelItemFactory.makeRestResponse();
        result = createMockResult();
    }

    @Test
    public void writesResponse() throws Exception {
        String writtenResponseContent = mockResponse.writeResponse(result, originalResponseContent);
        assertThat(writtenResponseContent, is( originalResponseContent ));
    }

    @Test
    public void shouldWriteContentLength() throws Exception {
        mockResponse.writeResponse(result, originalResponseContent);
        verify(servletResponse).addHeader("Content-Length", "" + originalResponseContent.getBytes().length);
    }

    @Test
    public void shouldNotWriteContentLengthIfTransferEncodingIsPresent() throws Exception {
        result.addHeader("Transfer-Encoding", "chunked");
        mockResponse.writeResponse(result, originalResponseContent);
        verify(servletResponse, never()).addHeader(eq("Content-Length"), anyString());
    }

    @Test
    public void shouldSetStatusCode() throws Exception {
        mockResponse.setResponseHttpStatus(HttpStatus.SC_CONFLICT);
        mockResponse.writeResponse(result, originalResponseContent);
        verify(servletResponse).setStatus(HttpStatus.SC_CONFLICT);
    }

    public MockResult createMockResult() throws Exception {
        WsdlMockRunContext runContext = mock(WsdlMockRunContext.class);
        return new RestMockResult(createMockRequest(runContext));
    }

    public RestMockRequest createMockRequest(WsdlMockRunContext runContext) throws Exception {
        return new RestMockRequest(servletRequest, servletResponse, runContext);
    }

}