org.nosceon.titanite.ControllersTest.java Source code

Introduction

Here is the source code for org.nosceon.titanite.ControllersTest.java

Source

/*
 * Copyright 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 org.nosceon.titanite;

import io.netty.handler.codec.http.HttpMethod;
import org.junit.Test;

import java.util.concurrent.CompletionStage;

import static com.jayway.restassured.RestAssured.given;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.nosceon.titanite.Response.ok;

/**
 * @author Johan Siebens
 */
public class ControllersTest extends AbstractE2ETest {

    public static class ControllerA extends Controller {

        public ControllerA() {
            get("/a", this::handle);
            post("/a", this::handle);
            put("/a", this::handle);
            delete("/a", this::handle);
            patch("/a", this::handle);
        }

        private CompletionStage<Response> handle(Request request) {
            return ok().body(request.method().name()).toFuture();
        }

    }

    public static class ControllerB extends Controller {

        {
            get("/b", (r) -> ok().body(r.method().name()).toFuture());
            post("/b", (r) -> ok().body(r.method().name()).toFuture());
            put("/b", (r) -> ok().body(r.method().name()).toFuture());
            delete("/b", (r) -> ok().body(r.method().name()).toFuture());
            patch("/b", (r) -> ok().body(r.method().name()).toFuture());
        }

    }

    @Override
    protected Shutdownable configureAndStartHttpServer(HttpServer server) {
        return server.register(new ControllerA()).register(ControllerB.class).start();
    }

    @Test
    public void test() {
        given().expect().statusCode(200).body(equalTo(HttpMethod.GET.name())).when().get(uri("/a"));
        given().expect().statusCode(200).body(equalTo(HttpMethod.POST.name())).when().post(uri("/a"));
        given().expect().statusCode(200).body(equalTo(HttpMethod.PUT.name())).when().put(uri("/a"));
        given().expect().statusCode(200).body(equalTo(HttpMethod.DELETE.name())).when().delete(uri("/a"));
        given().expect().statusCode(200).body(equalTo(HttpMethod.PATCH.name())).when().patch(uri("/a"));

        given().expect().statusCode(200).body(equalTo(HttpMethod.GET.name())).when().get(uri("/b"));
        given().expect().statusCode(200).body(equalTo(HttpMethod.POST.name())).when().post(uri("/b"));
        given().expect().statusCode(200).body(equalTo(HttpMethod.PUT.name())).when().put(uri("/b"));
        given().expect().statusCode(200).body(equalTo(HttpMethod.DELETE.name())).when().delete(uri("/b"));
        given().expect().statusCode(200).body(equalTo(HttpMethod.PATCH.name())).when().patch(uri("/b"));
    }

}