StaticTest.java

/*
 * Copyright (c) 2013-2014 the original author or authors
 *
 * 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 io.werval.controllers;

import com.jayway.restassured.response.Response;
import io.werval.runtime.routes.RoutesParserProvider;
import io.werval.test.WervalHttpRule;
import java.io.File;
import java.math.BigDecimal;
import org.junit.ClassRule;
import org.junit.Test;

import static com.jayway.restassured.RestAssured.expect;
import static org.hamcrest.Matchers.equalTo;
import static org.junit.Assert.assertThat;

/**
 * Assert that the Static controller behave correctly.
 *
 * Please note that this test rely on the fact that the current working directory is set to the module base dir.
 */
public class StaticTest
{
    @ClassRule
    public static final WervalHttpRule WERVAL = new WervalHttpRule( new RoutesParserProvider(
        "GET /single Static.file( String file = 'src/test/resources/logback.xml' )\n"
        + "GET /tree/*path Static.tree( String root = 'src/test/resources', String path )"
    ) );

    private static final File ROOT = new File( "src/test/resources" );

    @Test
    public void givenSingleStaticFileRouteWhenRequestingExpectCorrectResult()
    {
        int expectedLength = fileLength( "logback.xml" );
        Response response = expect().
            statusCode( 200 ).
            header( "Content-Length", String.valueOf( expectedLength ) ).
            when().
            get( "/single" );
        assertThat( response.body().asByteArray().length, equalTo( expectedLength ) );
    }

    @Test
    public void givenTreeStaticFileRouteWhenRequestingDirectoryExpectIndexFile()
        throws Exception
    {
        int expectedLength = fileLength( "staticfiles/index.html" );
        Response response = expect().
            statusCode( 200 ).
            header( "Content-Length", String.valueOf( expectedLength ) ).
            when().
            get( "/tree/staticfiles/" );
        assertThat( response.body().asByteArray().length, equalTo( expectedLength ) );
    }

    @Test
    public void givenTreeStaticFilesRouteWhenRequestingNotPresentExpectNotFound()
    {
        expect().
            statusCode( 404 ).
            when().
            get( "/tree/not.found" );
    }

    @Test
    public void givenTreeStaticFilesRouteWhenRequestingExpectCorrectResult()
        throws Exception
    {
        int expectedLength = fileLength( "logback.xml" );
        Response response = expect().
            statusCode( 200 ).
            header( "Content-Length", String.valueOf( expectedLength ) ).
            when().
            get( "/tree/logback.xml" );
        assertThat( response.body().asByteArray().length, equalTo( expectedLength ) );
    }

    private int fileLength( String relativePath )
    {
        return new BigDecimal( new File( ROOT, relativePath ).length() ).intValueExact();
    }
}