com.cognifide.aet.executor.xmlparser.xml.XmlTestSuiteParser.java Source code

Introduction

Here is the source code for com.cognifide.aet.executor.xmlparser.xml.XmlTestSuiteParser.java

Source

/**
 * Automated Exploratory Tests
 *
 * Copyright (C) 2013 Cognifide Limited
 *
 * 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.cognifide.aet.executor.xmlparser.xml;

import com.google.common.base.Charsets;
import com.google.common.io.Files;

import com.cognifide.aet.communication.api.exceptions.AETException;
import com.cognifide.aet.executor.xmlparser.api.ParseException;
import com.cognifide.aet.executor.xmlparser.api.TestSuiteParser;
import com.cognifide.aet.executor.xmlparser.xml.models.Collect;
import com.cognifide.aet.executor.xmlparser.xml.models.Compare;
import com.cognifide.aet.executor.xmlparser.xml.models.TestSuite;
import com.cognifide.aet.executor.xmlparser.xml.utils.EscapeUtils;
import com.cognifide.aet.executor.model.TestSuiteRun;

import org.simpleframework.xml.Serializer;
import org.simpleframework.xml.convert.Registry;
import org.simpleframework.xml.convert.RegistryStrategy;
import org.simpleframework.xml.core.Persister;

import java.io.File;
import java.io.IOException;

public class XmlTestSuiteParser implements TestSuiteParser {

    @Override
    public TestSuiteRun parse(File testSuiteFile) throws ParseException {
        try {
            String testSuiteString = Files.toString(testSuiteFile, Charsets.UTF_8);
            return parse(testSuiteString);
        } catch (IOException e) {
            throw new ParseException(e.getMessage(), e);
        }
    }

    @Override
    public TestSuiteRun parse(String testSuiteString) throws ParseException {
        try {
            Serializer serializer = getSerializer();
            TestSuite testSuite = serializer.read(TestSuite.class, EscapeUtils.escapeUrls(testSuiteString));
            return testSuite.adaptToTestSuiteRun();
        } catch (Exception e) {
            throw new ParseException("Something is wrong with your suite definition!", e);
        }
    }

    private Serializer getSerializer() throws AETException {
        Registry registry = new Registry();
        Serializer serializer = new Persister(new RegistryStrategy(registry));
        try {
            registry.bind(Collect.class, new CollectConverter());
            registry.bind(Compare.class, new CompareConverter());
        } catch (Exception e) {
            throw new AETException("Error while serializing test suite.", e);
        }
        return serializer;
    }
}