com.linecorp.armeria.server.healthcheck.ManagedHttpHealthCheckServiceTest.java Source code

Introduction

Here is the source code for com.linecorp.armeria.server.healthcheck.ManagedHttpHealthCheckServiceTest.java

Source

/*
 * Copyright 2017 LINE Corporation
 *
 * LINE Corporation licenses this file to you 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:
 *
 *   https://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.linecorp.armeria.server.healthcheck;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.when;

import org.apache.http.HttpHeaders;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

import com.linecorp.armeria.common.AggregatedHttpMessage;
import com.linecorp.armeria.common.HttpData;
import com.linecorp.armeria.common.HttpMethod;
import com.linecorp.armeria.common.HttpRequest;
import com.linecorp.armeria.common.HttpRequestWriter;
import com.linecorp.armeria.common.HttpStatus;
import com.linecorp.armeria.common.MediaType;
import com.linecorp.armeria.common.logging.DefaultRequestLog;
import com.linecorp.armeria.server.ServiceRequestContext;

import io.netty.util.AsciiString;

public class ManagedHttpHealthCheckServiceTest {

    @Rule
    public MockitoRule rule = MockitoJUnit.rule();

    @Mock
    private ServiceRequestContext context;

    private final ManagedHttpHealthCheckService service = new ManagedHttpHealthCheckService();

    private final HttpRequest hcReq = HttpRequest.of(HttpMethod.HEAD, "/");

    private final HttpRequest hcTurnOffReq = HttpRequest.of(HttpMethod.PUT, "/", MediaType.PLAIN_TEXT_UTF_8, "off");
    private final HttpRequest hcTurnOnReq = HttpRequest.of(HttpMethod.PUT, "/", MediaType.PLAIN_TEXT_UTF_8, "on");

    @Before
    public void setUp() {
        when(context.logBuilder()).thenReturn(new DefaultRequestLog(context));
    }

    @Test
    public void turnOff() throws Exception {
        service.serverHealth.setHealthy(true);

        AggregatedHttpMessage res = service.serve(context, hcTurnOffReq).aggregate().get();

        assertThat(res.status()).isEqualTo(HttpStatus.OK);
        assertThat(res.headers().get(AsciiString.of(HttpHeaders.CONTENT_TYPE)))
                .isEqualTo(MediaType.PLAIN_TEXT_UTF_8.toString());
        assertThat(res.content().toStringUtf8()).isEqualTo("Set unhealthy.");

        res = service.serve(context, hcReq).aggregate().get();

        assertThat(res.status()).isEqualTo(HttpStatus.SERVICE_UNAVAILABLE);
        assertThat(res.headers().get(AsciiString.of(HttpHeaders.CONTENT_TYPE)))
                .isEqualTo(MediaType.PLAIN_TEXT_UTF_8.toString());
    }

    @Test
    public void turnOn() throws Exception {
        AggregatedHttpMessage res = service.serve(context, hcTurnOnReq).aggregate().get();

        assertThat(res.status()).isEqualTo(HttpStatus.OK);
        assertThat(res.headers().get(AsciiString.of(HttpHeaders.CONTENT_TYPE)))
                .isEqualTo(MediaType.PLAIN_TEXT_UTF_8.toString());
        assertThat(res.content().toStringUtf8()).isEqualTo("Set healthy.");

        res = service.serve(context, hcReq).aggregate().get();

        assertThat(res.status()).isEqualTo(HttpStatus.OK);
        assertThat(res.headers().get(AsciiString.of(HttpHeaders.CONTENT_TYPE)))
                .isEqualTo(MediaType.PLAIN_TEXT_UTF_8.toString());
    }

    @Test
    public void notSupported() throws Exception {
        HttpRequestWriter noopRequest = HttpRequest.streaming(HttpMethod.PUT, "/");
        noopRequest.write(() -> HttpData.ofAscii("noop"));
        noopRequest.close();

        AggregatedHttpMessage res = service.serve(context, noopRequest).aggregate().get();

        assertThat(res.status()).isEqualTo(HttpStatus.BAD_REQUEST);
        assertThat(res.headers().get(AsciiString.of(HttpHeaders.CONTENT_TYPE)))
                .isEqualTo(MediaType.PLAIN_TEXT_UTF_8.toString());
        assertThat(res.content().toStringUtf8()).isEqualTo("Not supported.");

        service.serverHealth.setHealthy(true);

        noopRequest = HttpRequest.streaming(HttpMethod.PUT, "/");
        noopRequest.write(() -> HttpData.ofAscii("noop"));
        noopRequest.close();

        res = service.serve(context, noopRequest).aggregate().get();

        assertThat(res.status()).isEqualTo(HttpStatus.BAD_REQUEST);
        assertThat(res.headers().get(AsciiString.of(HttpHeaders.CONTENT_TYPE)))
                .isEqualTo(MediaType.PLAIN_TEXT_UTF_8.toString());
        assertThat(res.content().toStringUtf8()).isEqualTo("Not supported.");
    }
}