selenide integration (via #169)

Artem Eroshenko · web-flow · commit 0e325d6492a4 · 2017-12-04T19:15:17.000+03:00
diff --git a/allure-selenide/build.gradle b/allure-selenide/build.gradle
@@ -0,0 +1,10 @@
+description = 'Allure Selenide'
+
+apply from: "${gradleScriptDir}/maven-publish.gradle"
+apply from: "${gradleScriptDir}/bintray.gradle"
+apply plugin: 'maven'
+
+dependencies {
+    compile project(':allure-java-commons')
+    compile 'com.codeborne:selenide:4.7'
+}
diff --git a/allure-selenide/src/main/java/io/qameta/allure/selenide/AllureSelenide.java b/allure-selenide/src/main/java/io/qameta/allure/selenide/AllureSelenide.java
@@ -0,0 +1,66 @@
+package io.qameta.allure.selenide;
+
+import com.codeborne.selenide.WebDriverRunner;
+import com.codeborne.selenide.logevents.LogEvent;
+import com.codeborne.selenide.logevents.LogEventListener;
+import io.qameta.allure.Allure;
+import io.qameta.allure.AllureLifecycle;
+import io.qameta.allure.model.Status;
+import io.qameta.allure.model.StatusDetails;
+import io.qameta.allure.model.StepResult;
+import io.qameta.allure.util.ResultsUtils;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.TakesScreenshot;
+
+import java.nio.charset.StandardCharsets;
+import java.util.UUID;
+
+/**
+ * @author Artem Eroshenko.
+ */
+public class AllureSelenide implements LogEventListener {
+
+    private final AllureLifecycle lifecycle;
+
+    public AllureSelenide() {
+        this(Allure.getLifecycle());
+    }
+
+    public AllureSelenide(final AllureLifecycle lifecycle) {
+        this.lifecycle = lifecycle;
+    }
+
+    @Override
+    public void onEvent(final LogEvent event) {
+        lifecycle.getCurrentTestCase().ifPresent(uuid -> {
+            final String stepUUID = UUID.randomUUID().toString();
+            lifecycle.startStep(stepUUID, new StepResult()
+                    .withName(event.toString())
+                    .withStatus(Status.PASSED));
+
+            lifecycle.updateStep(stepResult -> stepResult.setStart(stepResult.getStart() - event.getDuration()));
+
+            if (LogEvent.EventStatus.FAIL.equals(event.getStatus())) {
+                lifecycle.addAttachment("Screenshot", "image/png", "png", getScreenshotBytes());
+                lifecycle.addAttachment("Page source", "text/html", "html", getPageSourceBytes());
+                lifecycle.updateStep(stepResult -> {
+                    final StatusDetails details = ResultsUtils.getStatusDetails(event.getError())
+                            .orElse(new StatusDetails());
+                    stepResult.setStatus(Status.FAILED);
+                    stepResult.setStatusDetails(details);
+                });
+            }
+            lifecycle.stopStep(stepUUID);
+        });
+    }
+
+
+    private static byte[] getScreenshotBytes() {
+        return ((TakesScreenshot) WebDriverRunner.getWebDriver()).getScreenshotAs(OutputType.BYTES);
+    }
+
+    private static byte[] getPageSourceBytes() {
+        return WebDriverRunner.getWebDriver().getPageSource().getBytes(StandardCharsets.UTF_8);
+    }
+
+}
diff --git a/settings.gradle b/settings.gradle
@@ -21,6 +21,7 @@ include 'allure-junit4-aspect'
 include 'allure-spock'
 include 'allure-assertj'
 include 'allure-jbehave'
+include 'allure-selenide'
 
 def examples = [
         'testng',
@@ -35,4 +36,3 @@ examples.forEach({
     include "examples/$it"
     project(":examples/$it").name = it
 })
-
