com.codeborne.selenide.junit.SoftAsserts Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of selenide-junit4 Show documentation

Selenide = concise API for Selenium WebDriver

The newest version!

package com.codeborne.selenide.junit;

import com.codeborne.selenide.logevents.ErrorsCollector;
import com.codeborne.selenide.logevents.SelenideLogger;
import com.codeborne.selenide.logevents.SoftAssertsErrorsCollector;
import org.jspecify.annotations.Nullable;
import org.junit.rules.ExternalResource;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;

import static com.codeborne.selenide.logevents.ErrorsCollector.LISTENER_SOFT_ASSERT;
import static java.util.Objects.requireNonNull;

/**
 * Rule for assert selenide verification softly.
 * 

 * Using:
 * 1. Add rule for test class: {@code @Rule public SoftAsserts softAsserts = new SoftAsserts();}
 * 

 * 2. Configure selenide to assert softly: {@code Configuration.assertionMode = SOFT;}
 */
public class SoftAsserts extends ExternalResource {
  @Nullable
  private Description currentTest;
  private final boolean fullStacktraces;

  public SoftAsserts() {
    this(true);
  }

  public SoftAsserts(boolean fullStacktraces) {
    this.fullStacktraces = fullStacktraces;
  }

  @Override
  public Statement apply(Statement base, Description description) {
    currentTest = description;
    return super.apply(base, description);
  }

  @Override
  protected void before() {
    SelenideLogger.addListener(LISTENER_SOFT_ASSERT, new SoftAssertsErrorsCollector());
  }

  @Override
  protected void after() {
    ErrorsCollector errorsCollector = requireNonNull(SelenideLogger.removeListener(LISTENER_SOFT_ASSERT));

    // if both test and "after" method threw an error, JUnit4 collects all of them
    // and throws org.junit.internal.runners.model.MultipleFailureException
    String testName = currentTest == null ? "?" : currentTest.getDisplayName();
    errorsCollector.cleanAndThrowAssertionError(testName, null, fullStacktraces);
  }
}