com.google.jstestdriver.output.XmlPrinterImpl.java Source code

Introduction

Here is the source code for com.google.jstestdriver.output.XmlPrinterImpl.java

Source

/*
 * Copyright 2009 Google Inc.
 *
 * 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.google.jstestdriver.output;

import com.google.common.base.Supplier;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Multimap;
import com.google.common.collect.Multimaps;
import com.google.inject.Inject;
import com.google.inject.name.Named;
import com.google.jstestdriver.BrowserInfo;
import com.google.jstestdriver.TestResult;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Collection;
import java.util.Map;
import java.util.Set;

/**
 * @author jeremiele@google.com (Jeremie Lenfant-Engelmann)
 */
public class XmlPrinterImpl implements XmlPrinter {

    private final Logger logger = LoggerFactory.getLogger(XmlPrinterImpl.class);
    private final TestResultHolder resultHolder;
    private final String xmlOutputDir;
    private final String fileNameFormat = "TEST-%s.xml";
    private final FileNameFormatter formatter;

    @Inject
    public XmlPrinterImpl(TestResultHolder resultHolder, @Named("testOutput") String xmlOutputDir,
            FileNameFormatter formatter) {
        this.resultHolder = resultHolder;
        this.xmlOutputDir = xmlOutputDir;
        this.formatter = formatter;
    }

    public void writeXmlReportFiles() {
        Map<BrowserInfo, String> browserNames = createUniqueBrowserNames(resultHolder.getResults().keySet());
        for (BrowserInfo browser : resultHolder.getResults().keySet()) {
            Multimap<String, TestResult> testCaseRollup = newMultiMap();
            for (TestResult testResult : resultHolder.getResults().get(browser)) {
                testCaseRollup.put(testResult.getTestCaseName(), testResult);
            }
            for (String testCaseName : testCaseRollup.keySet()) {
                String suiteName = formatSuiteName(browserNames.get(browser), testCaseName);
                File xmlOutputFile = new File(xmlOutputDir, formatFileName(suiteName));
                try {
                    xmlOutputFile.createNewFile();
                    TestXmlSerializer serializer = new TestXmlSerializer(new FileWriter(xmlOutputFile));

                    serializer.writeTestCase(suiteName, testCaseRollup.get(testCaseName));
                } catch (IOException e) {
                    logger.error("Could not create file: {}", xmlOutputFile.getAbsolutePath(), e);
                }
            }
        }
    }

    public static Map<BrowserInfo, String> createUniqueBrowserNames(Set<BrowserInfo> browserInfos) {
        Map<BrowserInfo, String> result = Maps.newHashMap();
        for (BrowserInfo browserInfo : browserInfos) {
            if (result.containsValue(browserInfo.toString())) {
                result.put(browserInfo, browserInfo.toUniqueString());
            } else {
                result.put(browserInfo, browserInfo.toString());
            }
        }
        return result;
    }

    private String formatFileName(String suiteName) {
        return formatter.format(suiteName, fileNameFormat);
    }

    private String formatSuiteName(String browser, String testCaseName) {
        return String.format("%s.%s", browser.replaceAll("\\s", "_").replaceAll("\\.", ""), testCaseName);
    }

    private Multimap<String, TestResult> newMultiMap() {
        return Multimaps.newMultimap(Maps.<String, Collection<TestResult>>newHashMap(),
                new Supplier<Collection<TestResult>>() {
                    public Collection<TestResult> get() {
                        return Lists.newLinkedList();
                    }
                });
    }
}