opendaylight/netconf/netconf-util/src/test/java/org/opendaylight/controller/netconf/util/test/XmlUnitUtil.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.controller.netconf.util.test;
  10
  11 import static org.custommonkey.xmlunit.XMLAssert.assertNodeTestPasses;
  12 import static org.junit.Assert.assertEquals;
  13 import static org.junit.Assert.assertTrue;
  14
  15 import org.custommonkey.xmlunit.AbstractNodeTester;
  16 import org.custommonkey.xmlunit.NodeTest;
  17 import org.custommonkey.xmlunit.NodeTestException;
  18 import org.custommonkey.xmlunit.NodeTester;
  19 import org.w3c.dom.Document;
  20 import org.w3c.dom.Element;
  21 import org.w3c.dom.Node;
  22 import org.w3c.dom.Text;
  23
  24 public class XmlUnitUtil {
  25
  26     private XmlUnitUtil() {}
  27
  28     public static void assertContainsElementWithText(final Document doc, final String textToFind) throws NodeTestException {
  29         NodeTest nt = new NodeTest(doc);
  30         NodeTester tester = new AbstractNodeTester() {
  31
  32             boolean textFound = false;
  33
  34             @Override
  35             public void testText(Text text) throws NodeTestException {
  36                 if(!textFound) {
  37                     if (text.getData().equalsIgnoreCase(textToFind)) {
  38                         textFound = true;
  39                     }
  40                 }
  41             }
  42
  43             @Override
  44             public void noMoreNodes(NodeTest forTest) throws NodeTestException {
  45                 assertTrue(textFound);
  46             }
  47         };
  48         assertNodeTestPasses(nt, tester, new short[]{Node.TEXT_NODE}, true);
  49     }
  50
  51     public static void assertContainsElement(final Document doc, final Element testElement) throws NodeTestException {
  52         NodeTest nt = new NodeTest(doc);
  53         NodeTester tester = new AbstractNodeTester() {
  54
  55             private boolean elementFound = false;
  56
  57             @Override
  58             public void testElement(Element element) throws NodeTestException {
  59                 if (!elementFound) {
  60                     if(element.isEqualNode(testElement)) {
  61                         elementFound = true;
  62                     }
  63                 }
  64             }
  65
  66             @Override
  67             public void noMoreNodes(NodeTest forTest) throws NodeTestException {
  68                 assertTrue(elementFound);
  69             }
  70         };
  71         assertNodeTestPasses(nt, tester, new short[]{Node.ELEMENT_NODE}, true);
  72     }
  73
  74     public static void assertContainsElementWithName(final Document doc, final String elementName) throws NodeTestException {
  75         NodeTest nt = new NodeTest(doc);
  76         NodeTester tester = new AbstractNodeTester() {
  77
  78             private boolean elementFound = false;
  79
  80             @Override
  81             public void testElement(Element element) throws NodeTestException {
  82                 if (!elementFound) {
  83                     if (element.getNodeName() != null && element.getNodeName().equals(elementName)) {
  84                         elementFound = true;
  85                     }
  86                 }
  87             }
  88
  89             @Override
  90             public void noMoreNodes(NodeTest forTest) throws NodeTestException {
  91                 assertTrue(elementFound);
  92             }
  93         };
  94         assertNodeTestPasses(nt, tester, new short[]{Node.ELEMENT_NODE}, true);
  95     }
  96
  97     public static void assertElementsCount(final Document doc, final String elementName, final int expectedCount) {
  98         NodeTest nt = new NodeTest(doc);
  99         NodeTester tester = new AbstractNodeTester() {
 100
 101             private int elementFound = 0;
 102
 103             @Override
 104             public void testElement(Element element) throws NodeTestException {
 105                 if (element.getNodeName() != null && element.getNodeName().equals(elementName)) {
 106                     elementFound++;
 107                 }
 108             }
 109
 110             @Override
 111             public void noMoreNodes(NodeTest forTest) throws NodeTestException {
 112                 assertEquals(expectedCount, elementFound);
 113             }
 114         };
 115         assertNodeTestPasses(nt, tester, new short[]{Node.ELEMENT_NODE}, true);
 116     }
 117 }