com.atypon.wayf.request.ResponseWriterTest.java Source code

Introduction

Here is the source code for com.atypon.wayf.request.ResponseWriterTest.java

Source

/*
 * Copyright 2017 Atypon Systems, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.atypon.wayf.request;

import com.atypon.wayf.data.ServiceException;
import com.atypon.wayf.facade.ErrorLoggerFacade;
import com.atypon.wayf.facade.impl.ErrorLoggerFacadeMockImpl;
import org.apache.http.HttpStatus;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

public class ResponseWriterTest {

    private ResponseWriterMock responseWriter;
    private ErrorLoggerFacadeMockImpl errorLoggerFacade;

    @Before
    public void setup() {
        ResponseWriterMock responseWriterMock = new ResponseWriterMock();
        errorLoggerFacade = new ErrorLoggerFacadeMockImpl();
        responseWriterMock.setErrorLoggerFacade(errorLoggerFacade);
        this.responseWriter = responseWriterMock;
    }

    @After
    public void teardown() {
        RequestContextAccessor.set(null);
    }

    @Test
    public void testHasMore() {
        RequestContextAccessor
                .set(new RequestContext().setRequestUrl("http://localhost:8080/list?param1=1&limit=30&offset=0")
                        .setOffset(0).setLimit(30).setHasAnotherDbPage(Boolean.TRUE));
        String link = responseWriter._getLinkHeaderValue();
        assertEquals("<http://localhost:8080/list?param1=1&limit=30&offset=30>; rel=\"next\"", link);
    }

    @Test
    public void testHasNoMore() {
        RequestContextAccessor.set(new RequestContext().setRequestUrl("/list").setOffset(0).setLimit(30)
                .setHasAnotherDbPage(Boolean.FALSE));
        String link = responseWriter._getLinkHeaderValue();
        assertEquals("", link);
    }

    @Test
    public void testLogsError() {
        responseWriter.buildFailure(new RoutingContextMock());

        int statusCode = errorLoggerFacade.getStatusCode();
        Throwable exception = errorLoggerFacade.getException();

        assertEquals(HttpStatus.SC_FAILED_DEPENDENCY, statusCode);
        assertEquals(ServiceException.class, exception.getClass());
    }

    private class ResponseWriterMock extends ResponseWriter {
        public ResponseWriterMock() {
        }

        public void setErrorLoggerFacade(ErrorLoggerFacade errorLoggerFacade) {
            super.errorLoggerFacade = errorLoggerFacade;
        }

        public String _getLinkHeaderValue() {
            return super.buildLinkHeaderValue();
        }
    }

}