com.hp.application.automation.tools.octane.tests.TestResultIterator.java Source code

Introduction

Here is the source code for com.hp.application.automation.tools.octane.tests.TestResultIterator.java

Source

/*
 *     Copyright 2017 Hewlett-Packard Development Company, L.P.
 *     Licensed under the Apache License, Version 2.0 (the "License");
 *     you may not use this file except in compliance with the License.
 *     You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *     Unless required by applicable law or agreed to in writing, software
 *     distributed under the License is distributed on an "AS IS" BASIS,
 *     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *     See the License for the specific language governing permissions and
 *     limitations under the License.
 *
 */

package com.hp.application.automation.tools.octane.tests;

import com.hp.application.automation.tools.octane.tests.junit.JUnitTestResult;
import com.hp.application.automation.tools.octane.tests.junit.TestResultStatus;
import org.apache.commons.io.IOUtils;

import javax.xml.namespace.QName;
import javax.xml.stream.XMLEventReader;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.events.Attribute;
import javax.xml.stream.events.StartElement;
import javax.xml.stream.events.XMLEvent;
import java.io.FileNotFoundException;
import java.io.Reader;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.NoSuchElementException;

public class TestResultIterator implements Iterator<JUnitTestResult> {

    private Reader input;
    private XMLEventReader reader;
    private LinkedList<JUnitTestResult> items = new LinkedList<JUnitTestResult>();
    private boolean closed;
    private String serverId;
    private String jobId;
    private String buildId;
    private String subType;

    public TestResultIterator(Reader input) throws FileNotFoundException, XMLStreamException {
        this.input = input;
        reader = XMLInputFactory.newInstance().createXMLEventReader(input);
    }

    @Override
    public boolean hasNext() {
        try {
            while (items.isEmpty() && !closed) {
                if (reader.hasNext()) {
                    XMLEvent event = reader.nextEvent();
                    if (event instanceof StartElement) {
                        StartElement element = (StartElement) event;
                        String localName = element.getName().getLocalPart();
                        if ("test_run".equals(localName)) {
                            String moduleName = element.getAttributeByName(new QName("module")).getValue();
                            String packageName = element.getAttributeByName(new QName("package")).getValue();
                            String className = element.getAttributeByName(new QName("class")).getValue();
                            String testName = element.getAttributeByName(new QName("name")).getValue();
                            long duration = Long
                                    .valueOf(element.getAttributeByName(new QName("duration")).getValue());
                            TestResultStatus status = TestResultStatus
                                    .fromPrettyName(element.getAttributeByName(new QName("status")).getValue());
                            long started = Long
                                    .valueOf(element.getAttributeByName(new QName("started")).getValue());
                            items.add(new JUnitTestResult(moduleName, packageName, className, testName, status,
                                    duration, started, null, null));
                        } else if ("build".equals(localName)) {
                            serverId = element.getAttributeByName(new QName("server_id")).getValue();
                            jobId = element.getAttributeByName(new QName("job_id")).getValue();
                            buildId = element.getAttributeByName(new QName("build_id")).getValue();
                            Attribute subType = element.getAttributeByName(new QName("sub_type"));
                            if (subType != null) {
                                this.subType = subType.getValue();
                            }
                        }
                    }
                } else {
                    closed = true;
                    IOUtils.closeQuietly(input);
                    reader.close();
                }
            }
            return !items.isEmpty();
        } catch (XMLStreamException e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public JUnitTestResult next() {
        if (!hasNext()) {
            throw new NoSuchElementException();
        }
        return items.removeFirst();
    }

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

    public String getServerId() {
        hasNext();
        return serverId;
    }

    public String getJobId() {
        hasNext();
        return jobId;
    }

    public String getBuildId() {
        hasNext();
        return buildId;
    }

    public String getSubType() {
        hasNext();
        return subType;
    }
}