io.qameta.allure.internal.AllureStorage Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of allure-java-commons Show documentation
Allure Java Commons
There is a newer version: 2.29.1
package io.qameta.allure.internal;

import io.qameta.allure.model.FixtureResult;
import io.qameta.allure.model.StepResult;
import io.qameta.allure.model.TestResult;
import io.qameta.allure.model.TestResultContainer;
import io.qameta.allure.model.WithSteps;

import java.util.LinkedList;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Internal Allure data storage.
 *
 * @since 2.0
 */
public class AllureStorage {

    private final Map storage = new ConcurrentHashMap<>();

    private final ThreadLocal> currentStepContext =
            InheritableThreadLocal.withInitial(LinkedList::new);

    @SuppressWarnings("PMD.NullAssignment")
    public Optional getCurrentStep() {
        final LinkedList uids = currentStepContext.get();
        return uids.isEmpty()
                ? Optional.empty()
                : Optional.of(uids.getFirst());
    }

    @SuppressWarnings("PMD.NullAssignment")
    public String getRootStep() {
        final LinkedList uids = currentStepContext.get();
        return uids.isEmpty()
                ? null
                : uids.getLast();
    }

    public void startStep(final String uuid) {
        currentStepContext.get().push(uuid);
    }

    public void stopStep() {
        currentStepContext.get().pop();
    }

    public void clearStepContext() {
        currentStepContext.remove();
    }

    public Optional getContainer(final String uuid) {
        return get(uuid, TestResultContainer.class);
    }

    public void addContainer(final TestResultContainer container) {
        put(container.getUuid(), container);
    }

    public Optional removeContainer(final String uuid) {
        return remove(uuid, TestResultContainer.class);
    }

    public void addTestResult(final TestResult testResult) {
        put(testResult.getUuid(), testResult);
    }

    public Optional getTestResult(final String uuid) {
        return get(uuid, TestResult.class);
    }

    public Optional removeTestResult(final String uuid) {
        return remove(uuid, TestResult.class);
    }

    public Optional getFixture(final String uuid) {
        return get(uuid, FixtureResult.class);
    }

    public void addFixture(final String uuid, final FixtureResult fixtureResult) {
        put(uuid, fixtureResult);
    }

    public Optional removeFixture(final String uuid) {
        return remove(uuid, FixtureResult.class);
    }

    public Optional getStep(final String uuid) {
        return get(uuid, StepResult.class);
    }

    public void addStep(final String parentUuid, final String uuid, final StepResult step) {
        put(uuid, step);
        get(parentUuid, WithSteps.class).ifPresent(parentStep -> parentStep.getSteps().add(step));
    }

    public Optional removeStep(final String uuid) {
        return remove(uuid, StepResult.class);
    }

    public  T put(final String uuid, final T item) {
        Objects.requireNonNull(uuid, "Can't put item to storage: uuid can't be null");
        storage.put(uuid, item);
        return item;
    }

    public  Optional get(final String uuid, final Class clazz) {
        Objects.requireNonNull(uuid, "Can't get item from storage: uuid can't be null");
        return Optional.ofNullable(storage.get(uuid))
                .map(item -> cast(item, clazz));
    }

    public  Optional remove(final String uuid, final Class clazz) {
        Objects.requireNonNull(uuid, "Can't remove item from storage: uuid can't be null");
        return Optional.ofNullable(storage.remove(uuid))
                .map(item -> cast(item, clazz));
    }

    public  T cast(final Object obj, final Class clazz) {
        if (clazz.isInstance(obj)) {
            return clazz.cast(obj);
        }
        throw new IllegalStateException("Can not cast " + obj + " to " + clazz);
    }
}