CallsSteps.java

/**
 * 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 com.github.cchacin.cucumber.steps;

import static com.xebialabs.restito.builder.stub.StubHttp.whenHttp;
import static com.xebialabs.restito.semantics.Action.resourceContent;
import static com.xebialabs.restito.semantics.Action.status;
import static com.xebialabs.restito.semantics.Condition.delete;
import static com.xebialabs.restito.semantics.Condition.get;
import static com.xebialabs.restito.semantics.Condition.parameter;
import static com.xebialabs.restito.semantics.Condition.post;
import static com.xebialabs.restito.semantics.Condition.put;

import java.util.ArrayList;
import java.util.List;

import lombok.Value;

import org.glassfish.grizzly.http.util.HttpStatus;

import com.xebialabs.restito.builder.stub.StubHttp;
import com.xebialabs.restito.semantics.Condition;
import com.xebialabs.restito.semantics.ConditionWithApplicables;
import com.xebialabs.restito.server.StubServer;

import cucumber.api.DataTable;
import cucumber.api.java.After;
import cucumber.api.java.Before;
import cucumber.api.java.en.Given;

public class CallsSteps {

  private StubServer server;
  private StubHttp stubHttp;

  @Before
  public void setUp() {
    server = new StubServer(9090).run();
    stubHttp = whenHttp(server);
  }

  @After
  public void tearDown() {
    server.stop();
  }

  @Given("^The call to external service should be:$")
  public void the_call_to_external_service_should_be(final DataTable data) throws Throwable {

    final List<Call> calls = data.asList(Call.class);

    for (final Call call : calls) {
      stubHttp.match(call.getHttpMethod(), call.buildQueryParams()).then(
          status(HttpStatus.getHttpStatus(call.getStatusCode())),
          resourceContent(Thread.currentThread().getContextClassLoader()
              .getResource("fixtures/" + call.getFilename())));
    }
  }

  @Value
  private static class Call {
    private String method;
    private String url;
    private int statusCode;
    private String filename;

    ConditionWithApplicables getHttpMethod() {
      switch (getMethod().toUpperCase()) {
        case "POST":
          return post(buildUrl());
        case "PUT":
          return put(buildUrl());
        case "DELETE":
          return delete(buildUrl());
        case "GET":
          return get(buildUrl());
        default:
          return get(buildUrl());
      }
    }

    String buildUrl() {
      return (getUrl().contains("?")) ? getUrl().split("\\?")[0] : getUrl();
    }

    Condition buildQueryParams() {
      final String queryParamsStr = (getUrl().contains("?")) ? getUrl().split("\\?")[1] : null;

      final String[] kvs = queryParamsStr != null ? queryParamsStr.split("\\&") : new String[0];
      final List<Condition> conditions = new ArrayList<>();
      for (final String kv : kvs) {
        final String[] sp = kv.split("\\=");
        conditions.add(parameter(sp[0], sp[1]));
      }
      return Condition.composite(conditions.toArray(new Condition[conditions.size()]));
    }
  }
}