com.github.aoreshin.allure.rest.assured.ApiExtractingSteps Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of allure-rest-assured Show documentation

Project to ease pain of test automation

There is a newer version: 83

package com.github.aoreshin.allure.rest.assured;

import static com.github.aoreshin.allure.rest.assured.ApiRequestSteps.apiRequest;
import static java.util.stream.Collectors.toMap;

import com.github.aoreshin.junit5.allure.steps.StepWrapperSteps;
import io.qameta.allure.Step;
import io.restassured.response.Response;
import java.util.Map;

/** Steps for extracting data from Rest Assured's Response */
public class ApiExtractingSteps extends StepWrapperSteps {
  private final Response response;

  ApiExtractingSteps(Response response) {
    this.response = response;
  }

  @Step("Сохранение {jsonPath} как {key}")
  public  ApiExtractingSteps saveBodyJsonPath(
      String jsonPath, String key, Map map, Class type) {
    T value = response.getBody().jsonPath().get(jsonPath);
    map.put(key, value);
    return this;
  }

  @Step("Сохранение полей с соответствующими ключами {pathsAndKeys}")
  public  ApiExtractingSteps saveBodyJsonPath(
      Map pathsAndKeys, Map map, Class type) {
    Map values =
        pathsAndKeys.entrySet().stream()
            .collect(
                toMap(
                    Map.Entry::getValue,
                    entry -> {
                      String expression = entry.getKey();
                      return response.getBody().jsonPath().get(expression);
                    }));
    map.putAll(values);
    return this;
  }

  @Step("Сохранение заголовка {header} как {key}")
  public ApiExtractingSteps saveHeader(String header, String key, Map map) {
    String value = response.getHeader(header);
    map.put(key, value);
    return this;
  }

  @Step("Сохранение заголовков как соответствующие им ключи {pathsAndKeys}")
  public ApiExtractingSteps saveHeader(
      Map headersAndKeys, Map map) {
    Map values =
        headersAndKeys.entrySet().stream()
            .collect(toMap(Map.Entry::getValue, entry -> response.getHeader(entry.getKey())));
    map.putAll(values);
    return this;
  }

  @Step("Сохранение cookie {cookieName} как {key}")
  public ApiExtractingSteps saveCookie(String cookieName, String key, Map map) {
    String cookie = response.cookie(cookieName);
    map.put(key, cookie);
    return this;
  }

  @Step("Сохранение всех cookies как {key}")
  public ApiExtractingSteps saveAllCookies(String key, Map map) {
    Map cookies = response.cookies();
    map.put(key, cookies);
    return this;
  }

  public ApiRequestSteps next() {
    return apiRequest();
  }
}