1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
import unittest
from xml.etree import ElementTree
from ncclient.operations.util import *
try:
from unittest.mock import MagicMock # Python 3.4 and later
except ImportError:
from mock import MagicMock
xml = """<filter type="xpath">
<configuration>
<system>
<services/>
</system>
</configuration>
</filter>"""
class TestUtils(unittest.TestCase):
def test_one_of_1(self):
self.assertEqual(one_of(None, 10, None), None)
def test_one_of_2(self):
self.assertRaises(OperationError,
one_of, None, 10, 10)
def test_one_of_3(self):
self.assertRaises(OperationError,
one_of, None, None)
def test_datastore_url(self):
node = new_ele("target")
sub_ele(node, "candidate")
result = ElementTree.tostring(
datastore_or_url(
"target",
"candidate"))
self.assertEqual(result, ElementTree.tostring(node))
def test_datastore_url_2(self):
node = new_ele("web")
sub_ele(node, "url").text = "http://juniper.net"
result = ElementTree.tostring(
datastore_or_url(
"web",
"http://juniper.net",
capcheck=MagicMock()))
self.assertEqual(result, ElementTree.tostring(node))
def test_datastore_url_3(self):
node = new_ele("web")
result = ElementTree.tostring(
datastore_or_url(
"web",
"http://juniper.net"))
self.assertEqual(result, ElementTree.tostring(node))
def test_build_filter(self):
reply = build_filter(xml)
call = ElementTree.tostring(reply)
self.assertEqual(call, ElementTree.tostring(to_ele(xml)))
def test_build_filter_xpath(self):
criteria = "configuration/system"
filter = ("xpath", criteria)
reply = build_filter(filter)
call = ElementTree.tostring(reply)
node = new_ele("filter", type="xpath")
node.attrib["select"] = criteria
self.assertEqual(call, ElementTree.tostring(node))
def test_build_filter_xpath_ns(self):
select = "configuration/system"
ns = {"ns0": "http://www.xxx.org"}
criteria = (ns, select)
filter = ("xpath", criteria)
reply = build_filter(filter)
call = ElementTree.tostring(reply)
node = new_ele_nsmap("filter", ns, type="xpath")
node.attrib["select"] = select
self.assertEqual(call, ElementTree.tostring(node))
def test_build_filter_subtree(self):
criteria = """<configuration>
<system>
<services/>
</system>
</configuration>"""
filter = ("subtree", criteria)
reply = build_filter(filter, capcheck="cap")
call = ElementTree.tostring(reply)
node = new_ele("filter", type="subtree")
node.append(to_ele(criteria))
self.assertEqual(call, ElementTree.tostring(node))
def test_build_filter_other(self):
criteria = """<configuration>
<system>
<services/>
</system>
</configuration>"""
filter = ("text", criteria)
self.assertRaises(OperationError,
build_filter, filter, capcheck="cap")
|
