net.javacrumbs.airline.server.VanillaTest.java Source code

Introduction

Here is the source code for net.javacrumbs.airline.server.VanillaTest.java

Source

/*
 * Copyright 2005-2010 the original author or authors.
 *
 * 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 net.javacrumbs.airline.server;

import static org.junit.Assert.assertEquals;

import java.io.InputStream;
import java.util.List;

import javax.xml.bind.JAXB;
import javax.xml.datatype.DatatypeConfigurationException;
import javax.xml.datatype.DatatypeFactory;
import javax.xml.transform.TransformerException;
import javax.xml.transform.dom.DOMResult;
import javax.xml.transform.stream.StreamSource;

import net.javacrumbs.airline.model.Flight;
import net.javacrumbs.airline.model.GetFlightsRequest;
import net.javacrumbs.airline.model.GetFlightsResponse;
import net.javacrumbs.airline.model.ServiceClass;

import org.custommonkey.xmlunit.XMLAssert;
import org.custommonkey.xmlunit.XMLUnit;
import org.junit.Test;
import org.springframework.xml.transform.TransformerHelper;
import org.w3c.dom.Document;

/**
 * Test that does not use any help from the framework.
 * @author Lukas Krecan
 *
 */
public class VanillaTest {

    private AirlineEndpoint endpoint = new AirlineEndpoint();

    @Test
    public void testGetFlights() throws AirlineException, DatatypeConfigurationException {
        GetFlightsRequest request = new GetFlightsRequest();
        request.setFrom("PRG");
        request.setTo("DUB");
        request.setServiceClass(ServiceClass.BUSINESS);
        request.setDepartureDate(DatatypeFactory.newInstance().newXMLGregorianCalendarDate(2011, 02, 23, 0));
        //more setter calls could be here

        GetFlightsResponse response = endpoint.getFlights(request);

        List<Flight> flights = response.getFlight();
        assertEquals(1, flights.size());
        assertEquals(ServiceClass.BUSINESS, flights.get(0).getServiceClass());
        //more assertions here

    }

    @Test
    public void testGetFlightsXml() throws AirlineException, DatatypeConfigurationException, TransformerException {
        GetFlightsRequest request = JAXB.unmarshal(getStream("request1.xml"), GetFlightsRequest.class);

        GetFlightsResponse response = endpoint.getFlights(request);

        DOMResult domResponse = new DOMResult();
        JAXB.marshal(response, domResponse);

        XMLUnit.setIgnoreWhitespace(true);
        XMLAssert.assertXMLEqual(getDocument("response1.xml"), (Document) domResponse.getNode());

    }

    private InputStream getStream(String name) {
        return Thread.currentThread().getContextClassLoader().getResourceAsStream(name);
    }

    private Document getDocument(String name) throws TransformerException {

        DOMResult result = new DOMResult();
        new TransformerHelper().transform(new StreamSource(getStream(name)), result);
        return (Document) result.getNode();

    }

}