org.opennaas.extensions.network.test.DummyParserTest.java Source code

Introduction

Here is the source code for org.opennaas.extensions.network.test.DummyParserTest.java

Source

package org.opennaas.extensions.network.test;

/*
 * #%L
 * OpenNaaS :: Network :: Model
 * %%
 * Copyright (C) 2007 - 2014 Fundaci Privada i2CAT, Internet i Innovaci a Catalunya
 * %%
 * 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.
 * #L%
 */

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Test;
import org.opennaas.core.resources.ResourceException;
import org.opennaas.core.resources.descriptor.network.NetworkTopology;
import org.opennaas.extensions.network.mock.MockNetworkDescriptor;
import org.xml.sax.SAXException;

public class DummyParserTest {
    Log log = LogFactory.getLog(DummyParserTest.class);

    @Test
    public void testNDLToJava() throws Exception {
        String path = new File(".").getCanonicalPath();
        System.out.println(path);
        String filePath = "src/test/resources/network_example1.xml";
        try {
            NetworkTopology exampleDescriptor = getNetworkDescriptor(filePath);
            log.info(exampleDescriptor.toString());
        } catch (IOException e) {
            log.error(e.getMessage());
            throw e;
        }
    }

    @Test
    public void testJavaToNDL() throws IOException {
        try {
            String filePath = "target/test1.xml";
            NetworkTopology mockRDF = MockNetworkDescriptor.newSimpleNDLNetworkDescriptor();
            addNetworkDescriptor(mockRDF, filePath);
        } catch (IOException e) {
            log.error(e.getMessage());
            throw e;
        }

    }

    @Test
    public void testNDLToJavaWithDiffLayers() throws Exception {
        String filePath = "src/test/resources/network_diffs_layer.xml";
        try {
            NetworkTopology exampleDescriptor = getNetworkDescriptor(filePath);
            log.info(exampleDescriptor.toString());
        } catch (IOException e) {
            log.error(e.getMessage());
            throw e;
        }
    }

    /**
     * Helper methods to test these functionality...
     * 
     * @param filename
     * @return
     * @throws JAXBException
     * @throws IOException
     * @throws ResourceException
     * @throws SAXException
     */
    private NetworkTopology getNetworkDescriptor(String filename)
            throws JAXBException, IOException, ResourceException, SAXException {
        FileInputStream inputStream = new FileInputStream(filename);
        NetworkTopology rd = loadNetworkDescriptor(inputStream);
        return rd;
    }

    private NetworkTopology loadNetworkDescriptor(InputStream stream) throws JAXBException, SAXException {

        NetworkTopology descriptor = null;
        try {
            JAXBContext context = JAXBContext.newInstance(NetworkTopology.class);

            Unmarshaller unmarshaller = context.createUnmarshaller();

            /* check wellformat xml with xsd */
            // TODO I CAN NOT UNDERSTAND WHY WE CAN GET THE LOADER FROM A COMMAND
            // SchemaFactory sf = SchemaFactory.newInstance(
            // javax.xml.XMLConstants.W3C_XML_SCHEMA_NS_URI);
            // ClassLoader loader = Thread.currentThread().getContextClassLoader();
            // Schema schema = sf.newSchema(new StreamSource(loader.getResourceAsStream(NAME_SCHEMA)));
            // unmarshaller.setSchema(schema);

            descriptor = (NetworkTopology) unmarshaller.unmarshal(stream);
        } finally {
            try {
                stream.close();
            } catch (IOException e) {
                // Ignore
            }
        }
        return descriptor;

    }

    private void addNetworkDescriptor(NetworkTopology networkDescriptor, String filename) throws IOException {

        OutputStream stream = new FileOutputStream(filename);
        saveNetworkDescriptor(networkDescriptor, stream);

    }

    public OutputStream saveNetworkDescriptor(NetworkTopology networkDescriptor, OutputStream stream) {

        try {
            JAXBContext context = JAXBContext.newInstance(NetworkTopology.class);
            Marshaller marshaller = context.createMarshaller();
            marshaller.marshal(networkDescriptor, stream);
        } catch (JAXBException e) {
            log.error(e.getMessage(), e.getCause());
        }
        return stream;

    }

}