com.cognifide.aet.xmlparser.xml.XmlTestSuiteParser Maven / Gradle / Ivy

Go to download
/**
 * 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.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.model.TestSuiteRun;
import com.cognifide.aet.xmlparser.api.ParseException;
import com.cognifide.aet.xmlparser.api.TestSuiteParser;
import com.cognifide.aet.xmlparser.xml.models.Collect;
import com.cognifide.aet.xmlparser.xml.models.Compare;
import com.cognifide.aet.xmlparser.xml.models.TestSuite;
import com.cognifide.aet.xmlparser.xml.utils.EscapeUtils;

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;
  }
}