org.catools.common.testng.model.CTestResults Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of common.testng Show documentation

The common TestNG extensions

The newest version!

package org.catools.common.testng.model;

import com.fasterxml.jackson.annotation.JsonIgnore;
import org.catools.common.collections.CHashMap;
import org.catools.common.collections.CList;
import org.catools.common.collections.CSet;
import org.testng.ITestNGMethod;

import java.util.stream.Stream;

public class CTestResults extends CList {
  public CTestResults() {
  }

  public CTestResults(CTestResult... c) {
    super(c);
  }

  public CTestResults(Stream stream) {
    super(stream);
  }

  public CTestResults(Iterable iterable) {
    super(iterable);
  }

  @JsonIgnore
  public CTestResult getTestResultOrNull(ITestNGMethod testNGMethod) {
    return getFirstOrNull(
        test ->
            test.getClassName().equals(testNGMethod.getTestClass().getName())
                && test.getMethodName().equals(testNGMethod.getMethodName()));
  }

  @JsonIgnore
  public boolean isPassed(ITestNGMethod method) {
    CTestResult testResultByMethod =
        getFirstOrNull(
            test ->
                test.getClassName().equals(method.getTestClass().getName())
                    && test.getMethodName().equals(method.getMethodName()));
    return testResultByMethod != null && testResultByMethod.getStatus().isPassed();
  }

  @JsonIgnore
  public CTestResult getTestResultByIdOrNull(String id) {
    return getFirstOrNull(test -> test.getTestIds() != null && test.getTestIds().contains(id));
  }

  public CSet getAllIssueIds() {
    CSet ids = new CSet<>();
    for (CTestResult result : this) {
      ids.addAll(result.getTestIds());
    }
    return ids;
  }

  public CHashMap getStatusMap() {
    CHashMap keys = new CHashMap<>();
    for (CTestResult result : this) {
      keys.putIfAbsent(result.getStatus(), new CTestResults());
      keys.get(result.getStatus()).add(result);
    }
    return keys;
  }
}