1 files changed, 130 insertions, 0 deletions
diff --git a/test/mocks/netconf-pnp-simulator/engine/tests/test_turing_machine.py b/test/mocks/netconf-pnp-simulator/engine/tests/test_turing_machine.py
new file mode 100644
index 000000000..8ac38b0f5
--- /dev/null
+++ b/test/mocks/netconf-pnp-simulator/engine/tests/test_turing_machine.py
@@ -0,0 +1,130 @@
+import nctest
+
+_NAMESPACES = {
+    "nc": "urn:ietf:params:xml:ns:netconf:base:1.0",
+    "tm": "http://example.net/turing-machine"
+}
+
+
+def check_labels_only_in_data(data):
+    children = data.xpath("/nc:rpc-reply/nc:data/*", namespaces=_NAMESPACES)
+    assert children
+    for child in children:
+        assert child.tag.endswith("turing-machine")
+    children = data.xpath("/nc:rpc-reply/nc:data/tm:turing-machine/*", namespaces=_NAMESPACES)
+    assert children
+    for child in children:
+        assert child.tag.endswith("transition-function")
+    children = data.xpath("/nc:rpc-reply/nc:data/tm:turing-machine/tm:transition-function/*", namespaces=_NAMESPACES)
+    assert children
+    for child in children:
+        assert child.tag.endswith("delta")
+    children = data.xpath("/nc:rpc-reply/nc:data/tm:turing-machine/tm:transition-function/tm:delta/*",
+                          namespaces=_NAMESPACES)
+    assert children
+    for child in children:
+        assert child.tag.endswith("label")
+
+
+def check_deltas_in_data(data):
+    deltas = data.xpath("/nc:rpc-reply/nc:data/tm:turing-machine/tm:transition-function/*", namespaces=_NAMESPACES)
+    assert deltas
+    for d in deltas:
+        assert d.tag.endswith("delta")
+
+
+class TestTuringMachine(nctest.NCTestCase):
+    """ Tests basic NETCONF operations on the turing-machine YANG module. """
+
+    def test_get(self):
+        reply = self.nc.get()
+        nctest.check_reply_data(reply)
+        check_deltas_in_data(reply.data)
+
+    def test_get_config_startup(self):
+        reply = self.nc.get_config(source="startup")
+        nctest.check_reply_data(reply)
+        check_deltas_in_data(reply.data)
+
+    def test_get_config_running(self):
+        reply = self.nc.get_config(source="running")
+        nctest.check_reply_data(reply)
+        check_deltas_in_data(reply.data)
+
+    def test_get_subtree_filter(self):
+        filter_xml = """<nc:filter xmlns:nc="urn:ietf:params:xml:ns:netconf:base:1.0">
+            <turing-machine xmlns="http://example.net/turing-machine">
+                <transition-function>
+                    <delta>
+                        <label />
+                    </delta>
+                </transition-function>
+            </turing-machine>
+            </nc:filter>"""
+        reply = self.nc.get_config(source="running", filter=filter_xml)
+        nctest.check_reply_data(reply)
+        check_deltas_in_data(reply.data)
+        check_labels_only_in_data(reply.data)
+
+    def test_get_xpath_filter(self):
+        # https://github.com/ncclient/ncclient/issues/166
+        filter_xml = """<nc:filter type="xpath" xmlns:nc="urn:ietf:params:xml:ns:netconf:base:1.0"
+            xmlns:tm="http://example.net/turing-machine"
+            select="/tm:turing-machine/transition-function/delta/label" />
+            """
+        reply = self.nc.get(filter=filter_xml)
+        nctest.check_reply_data(reply)
+        check_deltas_in_data(reply.data)
+        check_labels_only_in_data(reply.data)
+
+    def test_edit_config(self):
+        config_xml = """<nc:config xmlns:nc="urn:ietf:params:xml:ns:netconf:base:1.0">
+            <turing-machine xmlns="http://example.net/turing-machine">
+                <transition-function>
+                    <delta nc:operation="{}">
+                        <label>test-transition-rule</label>
+                        <input>
+                            <symbol>{}</symbol>
+                            <state>{}</state>
+                        </input>
+                    </delta>
+                </transition-function>
+            </turing-machine></nc:config>"""
+        # merge
+        reply = self.nc.edit_config(target='running', config=config_xml.format("merge", 9, 99))
+        nctest.check_reply_ok(reply)
+        # get
+        reply = self.nc.get_config(source="running")
+        nctest.check_reply_data(reply)
+        deltas = reply.data.xpath(
+            "/nc:rpc-reply/nc:data/tm:turing-machine/tm:transition-function/tm:delta[tm:label='test-transition-rule']",
+            namespaces=_NAMESPACES)
+        assert len(deltas) == 1
+        # create already existing - expect error
+        reply = self.nc.edit_config(target='running', config=config_xml.format("create", 9, 99))
+        nctest.check_reply_err(reply)
+        # replace
+        reply = self.nc.edit_config(target='running', config=config_xml.format("replace", 9, 88))
+        nctest.check_reply_ok(reply)
+        # get
+        reply = self.nc.get_config(source="running")
+        nctest.check_reply_data(reply)
+        states = reply.data.xpath(
+            "/nc:rpc-reply/nc:data/tm:turing-machine/tm:transition-function/tm:delta[tm:label='test-transition-rule']/"
+            "tm:input/tm:state",
+            namespaces=_NAMESPACES)
+        assert len(states) == 1
+        assert states[0].text == "88"
+        # delete
+        reply = self.nc.edit_config(target='running', config=config_xml.format("delete", 9, 88))
+        nctest.check_reply_ok(reply)
+        # delete non-existing - expect error
+        reply = self.nc.edit_config(target='running', config=config_xml.format("delete", 9, 88))
+        nctest.check_reply_err(reply)
+        # get - should be empty
+        reply = self.nc.get_config(source="running")
+        nctest.check_reply_data(reply)
+        deltas = reply.data.xpath(
+            "/nc:rpc-reply/nc:data/tm:turing-machine/tm:transition-function/tm:delta[tm:label='test-transition-rule']",
+            namespaces=_NAMESPACES)
+        assert not deltas