NetworkUtilTest.java

/*
 * 2012-3 Red Hat Inc. and/or its affiliates and other contributors.
 *
 * 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.overlord.rtgov.epn.util;

import static org.junit.Assert.*;

import org.junit.Test;
import org.overlord.rtgov.common.service.CacheManager;
import org.overlord.rtgov.epn.Network;
import org.overlord.rtgov.epn.Node;
import org.overlord.rtgov.epn.Subscription;
import org.overlord.rtgov.epn.testdata.TestCacheManager;
import org.overlord.rtgov.epn.testdata.TestEventProcessor1;
import org.overlord.rtgov.epn.testdata.TestEventProcessor2;
import org.overlord.rtgov.epn.testdata.TestEventProcessor3;
import org.overlord.rtgov.epn.testdata.TestPredicate1;
import org.overlord.rtgov.epn.testdata.TestPredicate2;
import org.overlord.rtgov.epn.util.NetworkUtil;

public class NetworkUtilTest {

    @Test
    public void testSerializeEPN() {
        Network epn=new Network();
        
        epn.setName("Test");
        epn.setVersion("123456");
        //epn.setRootNodeName("N0");
        
        Subscription sub1=new Subscription();
        sub1.setNodeName("N0");
        sub1.setSubject("First");
        epn.getSubscriptions().add(sub1);
        
        Subscription sub2=new Subscription();
        sub2.setNodeName("N1");
        sub2.setSubject("Second");
        epn.getSubscriptions().add(sub2);
        
        // Node 0
        Node n0=new Node();
        n0.setName("N0");
        epn.getNodes().add(n0);
        
        n0.setEventProcessor(new TestEventProcessor1());
        n0.setPredicate(new TestPredicate1());   
        
        n0.getEventProcessor().getServices().put("testCache",
                            new TestCacheManager());
        
        // Node 1
        Node n1=new Node();
        n1.setName("N1");
        epn.getNodes().add(n1);
        n1.getSourceNodes().add("N0");
        
        TestEventProcessor2 ep2=new TestEventProcessor2();
        n1.setEventProcessor(ep2);
        TestPredicate2 tp2=new TestPredicate2();
        tp2.setSomeProperty("TestProperty");
        n1.setPredicate(tp2);
        
        // Node 2
        Node n2=new Node();
        n2.setName("N2");
        epn.getNodes().add(n2);
        n2.getSourceNodes().add("N0");
        
        TestEventProcessor3 ep3=new TestEventProcessor3();
        n2.setEventProcessor(ep3);
        
        try {
            byte[] b=NetworkUtil.serialize(epn);   
            
            System.out.println("Serialized network="+new String(b));
        } catch(Exception e) {
            fail("Failed to serialize: "+e);
        }
    }

    @Test
    public void testDeserializeEPN() {
        try {
            java.io.InputStream is=NetworkUtilTest.class.getResourceAsStream("/jsondata/TestNetwork1.json");
            
            byte[] b = new byte[is.available()];
            is.read(b);
            
            is.close();
            
            Network network=NetworkUtil.deserialize(b);
            
            if (network.getNodes().size() != 3) {
                fail("Number of nodes not 3: "+network.getNodes().size());
            }
            
            Node n1=network.getNodes().get(0);
            Node n2=network.getNodes().get(1);
            Node n3=network.getNodes().get(2);
            
            if (n1.getPredicate() == null) {
                fail("Predicate 1 should not be null");
            }
            
            if (!(n1.getPredicate() instanceof TestPredicate1)) {
                fail("Predicate 1 not correct class");
            }
            
            if (n1.getEventProcessor() == null) {
                fail("Event Processor 1 should not be null");
            }
            
            if (!(n1.getEventProcessor() instanceof TestEventProcessor1)) {
                fail("Event Processor 1 not correct class");
            }
            
            TestEventProcessor1 tep1=(TestEventProcessor1)n1.getEventProcessor();
            if (!(tep1.getServices().get("testCache") instanceof CacheManager)) {
                fail("Failed to find test cache");
            }
            
            if (n2.getPredicate() == null) {
                fail("Predicate 2 should not be null");
            }
            
            if (!(n2.getPredicate() instanceof TestPredicate2)) {
                fail("Predicate 2 not correct class");
            }
            
            if (n2.getEventProcessor() == null) {
                fail("Event Processor 2 should not be null");
            }
            
            if (!(n2.getEventProcessor() instanceof TestEventProcessor2)) {
                fail("Event Processor 2 not correct class");
            }
            
            if (n3.getPredicate() != null) {
                fail("Predicate 3 should be null");
            }
            
            if (n3.getEventProcessor() == null) {
                fail("Event Processor 3 should not be null");
            }
            
            if (!(n3.getEventProcessor() instanceof TestEventProcessor3)) {
                fail("Event Processor 3 not correct class");
            }
            
        } catch(Exception e) {
            e.printStackTrace();
            fail("Failed to deserialize: "+e);
        }
    }
    
}