ru.yandex.qatools.allure.data.io.TestSuiteReader 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.io;

import ru.yandex.qatools.allure.commons.AllureFileUtils;
import ru.yandex.qatools.allure.model.TestSuiteResult;

import javax.inject.Inject;
import javax.xml.bind.JAXB;
import java.io.File;
import java.util.Iterator;

/**
 * eroshenkoam
 * 02/02/15
 */
public class TestSuiteReader implements Reader {

    private final Iterator testSuiteResultFiles;

    @Inject
    public TestSuiteReader(@ResultDirectories File... resultDirectories) {
        testSuiteResultFiles = AllureFileUtils.listTestSuiteFiles(resultDirectories).iterator();
    }

    @Override
    public Iterator iterator() {
        return new Iterator() {
            @Override
            public boolean hasNext() {
                return testSuiteResultFiles.hasNext();
            }

            @Override
            public TestSuiteResult next() {
                if (!hasNext()) {
                    return null;
                }
                return JAXB.unmarshal(testSuiteResultFiles.next(), TestSuiteResult.class);
            }

            @Override
            public void remove() {
                throw new UnsupportedOperationException();
            }
        };
    }
}