com.linecorp.armeria.server.grpc.GrpcServiceTest.java Source code

Introduction

Here is the source code for com.linecorp.armeria.server.grpc.GrpcServiceTest.java

Source

/*
 * Copyright 2016 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.grpc;

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

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.HttpHeaderNames;
import com.linecorp.armeria.common.HttpHeaders;
import com.linecorp.armeria.common.HttpMethod;
import com.linecorp.armeria.common.HttpRequest;
import com.linecorp.armeria.common.HttpResponse;
import com.linecorp.armeria.common.HttpStatus;
import com.linecorp.armeria.common.MediaType;
import com.linecorp.armeria.common.logging.DefaultRequestLog;
import com.linecorp.armeria.grpc.testing.TestServiceGrpc.TestServiceImplBase;
import com.linecorp.armeria.server.PathMapping;
import com.linecorp.armeria.server.ServiceRequestContext;

import io.netty.util.AsciiString;

// Tests error cases, success cases are checked in ArmeriaGrpcServiceInteropTest
public class GrpcServiceTest {

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

    @Mock
    private ServiceRequestContext ctx;

    private GrpcService grpcService;

    @Before
    public void setUp() {
        grpcService = (GrpcService) new GrpcServiceBuilder().addService(mock(TestServiceImplBase.class)).build();
        when(ctx.logBuilder()).thenReturn(new DefaultRequestLog(ctx));
    }

    @Test
    public void missingContentType() throws Exception {
        HttpResponse response = grpcService.doPost(ctx,
                HttpRequest.of(HttpHeaders.of(HttpMethod.POST, "/grpc.testing.TestService.UnaryCall")));
        assertThat(response.aggregate().get()).isEqualTo(AggregatedHttpMessage.of(
                HttpHeaders.of(HttpStatus.UNSUPPORTED_MEDIA_TYPE).contentType(MediaType.PLAIN_TEXT_UTF_8),
                HttpData.ofUtf8("Missing or invalid Content-Type header.")));
    }

    @Test
    public void badContentType() throws Exception {
        HttpResponse response = grpcService.doPost(ctx, HttpRequest.of(HttpHeaders
                .of(HttpMethod.POST, "/grpc.testing.TestService.UnaryCall").contentType(MediaType.JSON_UTF_8)));
        assertThat(response.aggregate().get()).isEqualTo(AggregatedHttpMessage.of(
                HttpHeaders.of(HttpStatus.UNSUPPORTED_MEDIA_TYPE).contentType(MediaType.PLAIN_TEXT_UTF_8),
                HttpData.ofUtf8("Missing or invalid Content-Type header.")));
    }

    @Test
    public void pathMissingSlash() throws Exception {
        when(ctx.mappedPath()).thenReturn("grpc.testing.TestService.UnaryCall");
        HttpResponse response = grpcService.doPost(ctx,
                HttpRequest.of(HttpHeaders.of(HttpMethod.POST, "grpc.testing.TestService.UnaryCall")
                        .set(HttpHeaderNames.CONTENT_TYPE, "application/grpc+proto")));
        assertThat(response.aggregate().get()).isEqualTo(AggregatedHttpMessage.of(
                HttpHeaders.of(HttpStatus.BAD_REQUEST).contentType(MediaType.PLAIN_TEXT_UTF_8),
                HttpData.ofUtf8("Invalid path.")));
    }

    @Test
    public void missingMethod() throws Exception {
        when(ctx.mappedPath()).thenReturn("/grpc.testing.TestService/FooCall");
        HttpResponse response = grpcService.doPost(ctx,
                HttpRequest.of(HttpHeaders.of(HttpMethod.POST, "/grpc.testing.TestService/FooCall")
                        .set(HttpHeaderNames.CONTENT_TYPE, "application/grpc+proto")));
        assertThat(
                response.aggregate().get())
                        .isEqualTo(
                                AggregatedHttpMessage.of(
                                        HttpHeaders.of(HttpStatus.OK)
                                                .set(HttpHeaderNames.CONTENT_TYPE, "application/grpc+proto")
                                                .set(AsciiString.of("grpc-status"), "12")
                                                .set(AsciiString.of("grpc-message"),
                                                        "Method not found: grpc.testing.TestService/FooCall")
                                                .set(HttpHeaderNames.CONTENT_LENGTH, "0"),
                                        HttpData.of(new byte[] {})));
    }

    @Test
    public void pathMappings() throws Exception {
        assertThat(grpcService.pathMappings()).containsExactlyInAnyOrder(
                PathMapping.ofExact("/armeria.grpc.testing.TestService/EmptyCall"),
                PathMapping.ofExact("/armeria.grpc.testing.TestService/UnaryCall"),
                PathMapping.ofExact("/armeria.grpc.testing.TestService/UnaryCall2"),
                PathMapping.ofExact("/armeria.grpc.testing.TestService/StreamingOutputCall"),
                PathMapping.ofExact("/armeria.grpc.testing.TestService/StreamingInputCall"),
                PathMapping.ofExact("/armeria.grpc.testing.TestService/FullDuplexCall"),
                PathMapping.ofExact("/armeria.grpc.testing.TestService/HalfDuplexCall"),
                PathMapping.ofExact("/armeria.grpc.testing.TestService/UnimplementedCall"));
    }
}