opendaylight/md-sal/sal-netconf-connector/src/test/java/org/opendaylight/controller/sal/connect/netconf/NetconfToNotificationTest.java

   1 package org.opendaylight.controller.sal.connect.netconf;
   2
   3 import static org.junit.Assert.assertEquals;
   4 import static org.junit.Assert.assertNotNull;
   5 import static org.junit.Assert.assertTrue;
   6
   7 import com.google.common.collect.Iterables;
   8 import java.io.InputStream;
   9 import java.util.Collections;
  10 import java.util.List;
  11 import java.util.Set;
  12 import javax.xml.parsers.DocumentBuilderFactory;
  13 import org.junit.Before;
  14 import org.junit.Test;
  15 import org.opendaylight.controller.netconf.api.NetconfMessage;
  16 import org.opendaylight.controller.netconf.util.xml.XmlUtil;
  17 import org.opendaylight.controller.sal.connect.netconf.schema.mapping.NetconfMessageTransformer;
  18 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  19 import org.opendaylight.yangtools.yang.model.api.Module;
  20 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  21 import org.opendaylight.yangtools.yang.model.parser.api.YangContextParser;
  22 import org.opendaylight.yangtools.yang.parser.impl.YangParserImpl;
  23 import org.w3c.dom.Document;
  24
  25 /**
  26  * @author Lukas Sedlak <lsedlak@cisco.com>
  27  */
  28 public class NetconfToNotificationTest {
  29
  30     NetconfMessageTransformer messageTransformer;
  31
  32     NetconfMessage userNotification;
  33
  34     @SuppressWarnings("deprecation")
  35     @Before
  36     public void setup() throws Exception {
  37         final SchemaContext schemaContext = getNotificationSchemaContext(getClass());
  38
  39         messageTransformer = new NetconfMessageTransformer(schemaContext);
  40
  41         final DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
  42         factory.setNamespaceAware(true);
  43         InputStream notifyPayloadStream = getClass().getResourceAsStream("/notification-payload.xml");
  44         assertNotNull(notifyPayloadStream);
  45
  46         final Document doc = XmlUtil.readXmlToDocument(notifyPayloadStream);
  47         assertNotNull(doc);
  48         userNotification = new NetconfMessage(doc);
  49     }
  50
  51     static SchemaContext getNotificationSchemaContext(Class<?> loadClass) {
  52         final List<InputStream> modelsToParse = Collections.singletonList(loadClass.getResourceAsStream("/schemas/user-notification.yang"));
  53         final YangContextParser parser = new YangParserImpl();
  54         final Set<Module> modules = parser.parseYangModelsFromStreams(modelsToParse);
  55         assertTrue(!modules.isEmpty());
  56         final SchemaContext schemaContext = parser.resolveSchemaContext(modules);
  57         assertNotNull(schemaContext);
  58         return schemaContext;
  59     }
  60
  61     @Test
  62     public void test() throws Exception {
  63         final ContainerNode root = messageTransformer.toNotification(userNotification);
  64         assertNotNull(root);
  65         assertEquals(6, Iterables.size(root.getValue()));
  66         assertEquals("user-visited-page", root.getNodeType().getLocalName());
  67     }
  68 }