org.springframework.batch.item.xml.Jaxb2NamespaceMarshallingTests.java Source code

Introduction

Here is the source code for org.springframework.batch.item.xml.Jaxb2NamespaceMarshallingTests.java

Source

/*
 * Copyright 2010-2014 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 org.springframework.batch.item.xml;

import static org.junit.Assert.assertTrue;

import java.io.File;
import java.io.FileReader;
import java.io.StringWriter;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;

import javax.xml.transform.stream.StreamResult;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.custommonkey.xmlunit.XMLAssert;
import org.custommonkey.xmlunit.XMLUnit;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.springframework.batch.item.ExecutionContext;
import org.springframework.batch.item.xml.StaxEventItemWriter;
import org.springframework.batch.item.xml.domain.QualifiedTrade;
import org.springframework.batch.support.transaction.ResourcelessTransactionManager;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.Resource;
import org.springframework.oxm.Marshaller;
import org.springframework.oxm.jaxb.Jaxb2Marshaller;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.TransactionCallback;
import org.springframework.transaction.support.TransactionTemplate;
import org.springframework.util.ClassUtils;
import org.springframework.util.StopWatch;

public class Jaxb2NamespaceMarshallingTests {

    private Log logger = LogFactory.getLog(getClass());

    private static final int MAX_WRITE = 100;

    private StaxEventItemWriter<QualifiedTrade> writer = new StaxEventItemWriter<QualifiedTrade>();

    private Resource resource;

    private File outputFile;

    private Resource expected = new ClassPathResource("expected-qualified-output.xml", getClass());

    @SuppressWarnings("serial")
    private List<QualifiedTrade> objects = new ArrayList<QualifiedTrade>() {
        {
            add(new QualifiedTrade("isin1", 1, new BigDecimal(1.0), "customer1"));
            add(new QualifiedTrade("isin2", 2, new BigDecimal(2.0), "customer2"));
            add(new QualifiedTrade("isin3", 3, new BigDecimal(3.0), "customer3"));
        }
    };

    /**
     * Write list of domain objects and check the output file.
     */
    @SuppressWarnings("resource")
    @Test
    public void testWrite() throws Exception {
        StopWatch stopWatch = new StopWatch(getClass().getSimpleName());
        stopWatch.start();
        for (int i = 0; i < MAX_WRITE; i++) {
            new TransactionTemplate(new ResourcelessTransactionManager()).execute(new TransactionCallback<Void>() {
                @Override
                public Void doInTransaction(TransactionStatus status) {
                    try {
                        writer.write(objects);
                    } catch (RuntimeException e) {
                        throw e;
                    } catch (Exception e) {
                        throw new IllegalStateException("Exception encountered on write", e);
                    }
                    return null;
                }
            });
        }
        writer.close();
        stopWatch.stop();
        logger.info("Timing for XML writer: " + stopWatch);
        XMLUnit.setIgnoreWhitespace(true);
        // String content = FileUtils.readFileToString(resource.getFile());
        // System.err.println(content);
        XMLAssert.assertXMLEqual(new FileReader(expected.getFile()), new FileReader(resource.getFile()));

    }

    @Before
    public void setUp() throws Exception {

        File directory = new File("target/data");
        directory.mkdirs();
        outputFile = File.createTempFile(ClassUtils.getShortName(this.getClass()), ".xml", directory);
        resource = new FileSystemResource(outputFile);

        writer.setResource(resource);

        writer.setMarshaller(getMarshaller());
        writer.setRootTagName("{urn:org.springframework.batch.io.oxm.domain}trades");

        writer.afterPropertiesSet();

        writer.open(new ExecutionContext());

    }

    @After
    public void tearDown() throws Exception {
        outputFile.delete();
    }

    protected Marshaller getMarshaller() throws Exception {

        Jaxb2Marshaller marshaller = new Jaxb2Marshaller();
        marshaller.setClassesToBeBound(new Class<?>[] { QualifiedTrade.class });
        marshaller.afterPropertiesSet();

        StringWriter string = new StringWriter();
        marshaller.marshal(new QualifiedTrade("FOO", 100, BigDecimal.valueOf(10.), "bar"),
                new StreamResult(string));
        String content = string.toString();
        assertTrue("Wrong content: " + content, content.contains("<customer>bar</customer>"));
        return marshaller;
    }

}