ru.yandex.qatools.allure.data.plugins.DefectsPlugin.groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of allure-report-data Show documentation

There is a newer version: 1.5.4

package ru.yandex.qatools.allure.data.plugins

import groovy.transform.EqualsAndHashCode
import ru.yandex.qatools.allure.data.AllureDefect
import ru.yandex.qatools.allure.data.AllureDefects
import ru.yandex.qatools.allure.data.AllureTestCase
import ru.yandex.qatools.allure.data.DefectItem
import ru.yandex.qatools.allure.data.utils.PluginUtils
import ru.yandex.qatools.allure.model.Status

import static ru.yandex.qatools.allure.data.utils.TextUtils.generateUid
import static ru.yandex.qatools.allure.data.utils.TextUtils.getMessageMask
import static ru.yandex.qatools.allure.model.Status.BROKEN
import static ru.yandex.qatools.allure.model.Status.FAILED

/**
 * @author Dmitry Baev [email protected]
 *         Date: 06.02.15
 */
class DefectsPlugin implements ProcessPlugin {

    public static final String DEFECTS_JSON = "defects.json"

    AllureDefects defects = new AllureDefects(defectsList: [
            new AllureDefect(title: "Product defects", status: FAILED),
            new AllureDefect(title: "Test defects", status: BROKEN)
    ]);

    private Map defectItems = new HashMap<>();

    @Override
    void process(AllureTestCase testCase) {
        if (!(testCase.status in [FAILED, BROKEN])) {
            return;
        }
        Key key = new Key(uid: getMessageMask(testCase?.failure?.message), status: testCase.status);
        if (!defectItems.containsKey(key)) {
            DefectItem item = new DefectItem(uid: generateUid(), failure: testCase.failure);
            defectItems.put(key, item);
            defects.defectsList.find { it.status == testCase.status }?.defects?.add(item);
        }

        use(PluginUtils) {
            defectItems[key].testCases.add(testCase.toInfo());
        }
    }

    @Override
    List getPluginData() {
        return Arrays.asList(new PluginData(DEFECTS_JSON, defects));
    }

    @Override
    Class getType() {
        return AllureTestCase;
    }

    @EqualsAndHashCode
    class Key {
        Status status;
        String uid;
    }
}