org.wiremock.grpc.internal.GrpcRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wiremock-grpc-extension Show documentation
Mock gRPC services with WireMock
The newest version!
/*
 * Copyright (C) 2023-2024 Thomas Akehurst
 *
 * 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.wiremock.grpc.internal;

import static java.util.Collections.*;

import com.github.tomakehurst.wiremock.common.Encoding;
import com.github.tomakehurst.wiremock.common.Strings;
import com.github.tomakehurst.wiremock.http.*;
import java.util.Collection;
import java.util.Map;
import java.util.Optional;
import java.util.Set;

public class GrpcRequest implements Request {

  private final String scheme;
  private final String host;
  private final int port;
  private final String serviceName;
  private final String methodName;

  private final String body;

  public GrpcRequest(
      String scheme, String host, int port, String serviceName, String methodName, String body) {
    this.scheme = scheme;
    this.host = host;
    this.port = port;
    this.serviceName = serviceName;
    this.methodName = methodName;
    this.body = body;
  }

  @Override
  public String getUrl() {
    return "/" + serviceName + "/" + methodName;
  }

  @Override
  public String getAbsoluteUrl() {
    return scheme + "://" + host + ":" + port + getUrl();
  }

  @Override
  public RequestMethod getMethod() {
    return RequestMethod.POST;
  }

  @Override
  public String getScheme() {
    return scheme;
  }

  @Override
  public String getHost() {
    return host;
  }

  @Override
  public int getPort() {
    return port;
  }

  @Override
  public String getClientIp() {
    return null;
  }

  @Override
  public String getHeader(String key) {
    return header(key).firstValue();
  }

  @Override
  public HttpHeader header(String key) {
    return getHeaders().getHeader(key);
  }

  @Override
  public ContentTypeHeader contentTypeHeader() {
    return getHeaders().getContentTypeHeader();
  }

  @Override
  public HttpHeaders getHeaders() {
    return HeaderCopyingServerInterceptor.HTTP_HEADERS_CONTEXT_KEY.get();
  }

  @Override
  public boolean containsHeader(String key) {
    return getHeader(key) != null;
  }

  @Override
  public Set getAllHeaderKeys() {
    return getHeaders().keys();
  }

  @Override
  public QueryParameter queryParameter(String key) {
    return QueryParameter.absent(key);
  }

  @Override
  public FormParameter formParameter(String key) {
    return FormParameter.absent(key);
  }

  @Override
  public Map formParameters() {
    return emptyMap();
  }

  @Override
  public Map getCookies() {
    return emptyMap();
  }

  @Override
  public byte[] getBody() {
    return Strings.bytesFromString(body);
  }

  @Override
  public String getBodyAsString() {
    return body;
  }

  @Override
  public String getBodyAsBase64() {
    return Encoding.encodeBase64(getBody());
  }

  @Override
  public boolean isMultipart() {
    return false;
  }

  @Override
  public Collection getParts() {
    return emptyList();
  }

  @Override
  public Part getPart(String name) {
    return null;
  }

  @Override
  public boolean isBrowserProxyRequest() {
    return false;
  }

  @Override
  public Optional getOriginalRequest() {
    return Optional.empty();
  }

  @Override
  public String getProtocol() {
    return "HTTP/2";
  }
}