integration-tests/src/main/java/com/epam/dlab/automation/http/HttpRequest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF 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
 *
 *   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 com.epam.dlab.automation.http;

import com.epam.dlab.automation.helper.ConfigPropertyValue;
import com.jayway.restassured.http.ContentType;
import com.jayway.restassured.response.Response;

import java.io.File;
import java.util.Map;

import static com.jayway.restassured.RestAssured.given;

public class HttpRequest {

	private static final String AUTHORIZATION = "Authorization";
	private static final String BEARER = "Bearer ";

	private void addHeader(String headerType, String headerValue) {
		given().header(headerType, headerValue);
	}

	public void addAuthorizationBearer(String token) {
		this.addHeader(AUTHORIZATION, BEARER + token);
	}

	public Response webApiGet(String url) {
		return given().contentType(ContentType.JSON).when().get(url);
	}

	public Response webApiGet(String url, String token) {
		return given().header(AUTHORIZATION, BEARER + token).contentType(ContentType.JSON).when().get(url);
	}
	
	public Response webApiGet(String url, String token, Map<String,?> params) {
		return given().header(AUTHORIZATION, BEARER + token).contentType(ContentType.JSON).params(params).when().get
				(url);
	}

	public Response webApiPost(String url, String contentType, Object body) {
		return given().contentType(contentType).body(body).when().post(url);
	}

	public Response webApiPost(String url, String contentType) {
		return given().contentType(contentType).when().post(url);
	}

	public Response webApiPost(String url, String contentType, String token) {
		return given()
				.contentType(contentType)
				.header(AUTHORIZATION, BEARER + token)
				.multiPart(new File(ConfigPropertyValue.getAccessKeyPubFileName()))
				.formParam(ConfigPropertyValue.getAccessKeyPubFileName())
				.contentType(contentType)
				.when()
				.post(url);
	}

	public Response webApiPost(String url, String contentType, Object body, String token) {
		return given().contentType(contentType).header(AUTHORIZATION, BEARER + token).body(body).when().post(url);
	}

	public Response webApiPut(String url, String contentType, Object body, String token) {
		return given().contentType(contentType).header(AUTHORIZATION, BEARER + token).body(body).when().put(url);
	}

	public Response webApiPut(String url, String contentType, String token) {
		return given().contentType(contentType).header(AUTHORIZATION, BEARER + token).when().put(url);
	}

	public Response webApiDelete(String url, String contentType, String token) {
		return given().contentType(contentType).header(AUTHORIZATION, BEARER + token).when().delete(url);
	}
}