io.vertx.ext.unit.collect.impl.EventBusCollectorImpl Maven / Gradle / Ivy

Go to download
package io.vertx.ext.unit.collect.impl;

import io.vertx.core.Handler;
import io.vertx.core.Vertx;
import io.vertx.core.eventbus.Message;
import io.vertx.core.eventbus.MessageConsumer;
import io.vertx.core.json.JsonObject;
import io.vertx.core.streams.ReadStream;
import io.vertx.ext.unit.collect.EventBusCollector;
import io.vertx.ext.unit.impl.FailureImpl;
import io.vertx.ext.unit.impl.TestResultImpl;
import io.vertx.ext.unit.report.Failure;
import io.vertx.ext.unit.report.TestCaseReport;
import io.vertx.ext.unit.report.TestResult;
import io.vertx.ext.unit.report.TestSuiteReport;

/**
 * @author Julien Viet
 */
public class EventBusCollectorImpl implements EventBusCollector, Handler> {

  private final Vertx vertx;
  private final Handler reporter;
  private Handler testCaseRunnerHandler;
  private Handler exceptionHandler;
  private Handler endHandler;
  private TestSuiteReport runner;
  private Handler testCaseHandler;

  public EventBusCollectorImpl(Vertx vertx, Handler reporter) {
    this.reporter = reporter;
    this.vertx = vertx;
  }

  @Override
  public void handle(Message event) {
    JsonObject body = event.body();
    String type = body.getString("type", "");
    String name = body.getString("name");
    switch (type) {
      case EVENT_TEST_SUITE_BEGIN: {
        runner = new TestSuiteReport() {
          @Override
          public String name() {
            return name;
          }
          @Override
          public TestSuiteReport exceptionHandler(Handler handler) {
            exceptionHandler = handler;
            return this;
          }
          @Override
          public TestSuiteReport handler(Handler handler) {
            testCaseRunnerHandler = handler;
            return this;
          }
          @Override
          public ReadStream fetch(long amount) {
            return this;
          }
          @Override
          public TestSuiteReport pause() {
            return this;
          }
          @Override
          public TestSuiteReport resume() {
            return this;
          }
          @Override
          public TestSuiteReport endHandler(Handler handler) {
            endHandler = handler;
            return this;
          }
        };
        reporter.handle(runner);
        break;
      }
      case EVENT_TEST_CASE_BEGIN: {
        if (testCaseRunnerHandler != null) {
          testCaseRunnerHandler.handle(new TestCaseReport() {
            @Override
            public String name() {
              return name;
            }
            @Override
            public TestCaseReport endHandler(Handler handler) {
              testCaseHandler = handler;
              return this;
            }
          });
        }
        break;
      }
      case EVENT_TEST_SUITE_ERROR: {
        JsonObject failureJson = body.getJsonObject("failure");
        if (failureJson != null && exceptionHandler != null) {
          FailureImpl failure = new FailureImpl(failureJson);
          Throwable cause = failure.cause();
          if (cause == null) {
            // Best effort
            cause = new Exception(failureJson.getString("message"));
          }
          exceptionHandler.handle(cause);
        }
        break;
      }
      case EVENT_TEST_CASE_END: {
        if (testCaseHandler != null) {
          JsonObject failureJson = body.getJsonObject("failure");
          Failure failure = null;
          if (failureJson != null) {
            failure = new FailureImpl(failureJson);
          }
          TestResult result = new TestResultImpl(name, body.getLong("beginTime", 0L), body.getLong("durationTime", 0L), failure);
          testCaseHandler.handle(result);
          testCaseHandler = null;
        }
        break;
      }
      case EVENT_TEST_SUITE_END: {
        if (endHandler != null) {
          endHandler.handle(null);
        }
        break;
      }
    }
  }

  @Override
  public MessageConsumer register(String address) {
    return vertx.eventBus().consumer(address, this);
  }

  @Override
  public Handler> asMessageHandler() {
    return this;
  }
}