def createTemplater(self, chassis=[], ports=[], names=[]):
dm = DataModel()
props = {"ports": ports, "names": names}
dm.new("dummy-session", chassis, props)
return Templater(dm)
class TestLinker:
def createModel(self):
self.dm = DataModel()
self.dm.new("dummy-session", [], [])
self.root = project1 = self.dm.insert("project1",
{"object_type": "project"})
self.port1 = port1 = self.dm.insert("port1", {
"object_type": "port",
"name": "port 1"
}, project1)
self.port2 = port2 = self.dm.insert("port2", {
"object_type": "port",
"name": "port 2"
}, project1)
self.dev1 = self.dm.insert("emulateddevice1", {
"object_type": "emulateddevice",
"name": "Device 1"
}, port1)
self.ip1 = self.dm.insert("ipv4if1", {
"object_type": "ipv4if",
"name": "ipv4if 1"
}, self.dev1)
self.dev2 = self.dm.insert("emulateddevice2", {
"object_type": "emulateddevice",
"name": "dev 2",
"count": 3
}, port1)
self.ip2 = self.dm.insert("ipv4if2", {
"object_type": "ipv4if",
"name": "ipv4if 2"
}, self.dev2)
self.dev3 = self.dm.insert("emulateddevice2", {
"object_type": "emulateddevice",
"name": "dev 3"
}, port2)
self.ip3 = self.dm.insert("ipv4if2", {
"object_type": "ipv4if",
"name": "ipv4if 3"
}, self.dev3)
return self.dm
def test1a(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name=port 1]")
assert nodes.count() == 1 and nodes.get(0) == self.port1
def test1b(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name=port 1]")
assert nodes.count() == 1 and nodes.get(0) == self.port1
def test1c(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name='port 1']")
assert nodes.count() == 1 and nodes.get(0) == self.port1
def test1d(self):
linker = Linker(self.createModel())
nodes = linker._resolve('/port[name="port 1"]')
assert nodes.count() == 1 and nodes.get(0) == self.port1
def test1e(self):
linker = Linker(self.createModel())
nodes = linker._resolve('/port[name= "port 1" ]')
assert nodes.count() == 1 and nodes.get(0) == self.port1
def test2(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name=port 1]/emulateddevice")
assert nodes.count() == 2 and nodes.get(0) == self.dev1
def test3a(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name=port 1]/emulateddevice")
assert nodes.count() == 2 and nodes.get(0) == self.dev1 and nodes.get(
1) == self.dev2
def test3b(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name=port 1]/emulateddevice[*]")
assert nodes.count() == 2 and nodes.get(0) == self.dev1 and nodes.get(
1) == self.dev2
def test4(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name=port 1]/emulateddevice[*]/ipv4if")
assert nodes.count() == 2 and nodes.get(0) == self.ip1 and nodes.get(
1) == self.ip2
def test5(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name!= port 1]/emulateddevice/ipv4if")
assert nodes.count() == 1 and nodes.get(0) == self.ip3
def test7a(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name ~= Port]/emulateddevice/ipv4if")
assert nodes.count() == 3
def test7b(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name *= ort]/emulateddevice/ipv4if")
assert nodes.count() == 3
def test7c(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name ^= port]/emulateddevice/ipv4if")
assert nodes.count() == 3
def test7d(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[@name ^= ort]/emulateddevice/ipv4if")
assert nodes == None
def test8a(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[0]/emulateddevice/ipv4if")
assert nodes.count() == 2
def test8b(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[1]/emulateddevice/ipv4if")
assert nodes.count() == 1
def test8c(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[2]/emulateddevice/ipv4if")
assert nodes == None
def test9a(self):
linker = Linker(self.createModel())
nodes = linker._resolve(
"/port/emulateddevice[@name *= dev][count = 3]/ipv4if")
assert nodes.count() == 1
def test9c(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port/emulateddevice[@name *= dev][0]/ipv4if")
assert nodes.count() == 1
def test9d(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port/*[@name *= dev][0]/ipv4if")
assert nodes.count() == 1
def test10a(self):
# Syntax error
linker = Linker(self.createModel())
with pytest.raises(Exception):
assert linker._resolve(
"/port/emulateddevice[@name *= 'dev']'/ipv4if")
def test11(self):
# mulitple refs
linker = Linker(self.createModel())
nodes = linker.resolveObjects(
"ref:/port/emulateddevice[*]/ipv4if,ref:/port/emulateddevice[*]/ipv4if"
)
assert nodes.count() == 6
class TestSelector:
def createModel(self):
self.dm = DataModel()
self.dm.new("dummy-session", [], [])
self.root = project1 = self.dm.insert("project1",
{"object_type": "project"})
self.port1 = self.dm.insert("port1", {
"object_type": "port",
"name": "port1 //abc"
}, project1)
self.port2 = self.dm.insert("port2", {
"object_type": "port",
"name": "port11"
}, project1)
self.port3 = self.dm.insert("port3", {
"object_type": "port",
"name": "port3"
}, project1)
dev1 = self.dm.insert("emulateddevice1", {
"object_type": "emulateddevice",
"name": "Device 1"
}, project1)
self.dm.insert("ipv4if1", {
"object_type": "ipv4if",
"name": "ipv4if 1"
}, dev1)
dev2 = self.dm.insert("emulateddevice2",
{"object_type": "emulateddevice"}, project1)
self.dm.insert("ipv4if2", {
"object_type": "ipv4if",
"name": "ipv4if 2"
}, dev2)
def test1(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("emulateddevice"))
assert n == 2
def test2(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("xproject"))
assert n == 0
def test3(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("emulateddevice [ @name = device 1]"))
assert n == 1
def test4(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("emulateddevice [@name= device x]"))
assert n == 0
def test5(self):
self.createModel()
selector = NodeSelector(self.root)
selector.select(Selector("emulateddevice"))
n = selector.select(Selector("ipv4if"))
assert n == 2
def test6(self):
self.createModel()
selector = NodeSelector(self.root)
selector.select(Selector("emulateddevice [ @name = device 1]"))
n = selector.select(Selector("ipv4if"))
assert n == 1
def test7(self):
self.createModel()
selector = NodeSelector(self.root)
selector.select(Selector("emulateddevice [ name = device 1]"))
n = selector.select(Selector("ipv4if [ @name = ipv4if 1 ]"))
assert n == 1
def test8(self):
self.createModel()
selector = NodeSelector(self.root)
selector.select(Selector("emulateddevice"))
n = selector.select(Selector("ipv4if [ @name = ipv4if 1 ]"))
assert n == 1
def test9(self):
self.createModel()
selector = NodeSelector(self.root)
selector.select(Selector("emulateddevice [ name = device 2]"))
n = selector.select(Selector("ipv4if [ @name = ipv4if 1 ]"))
assert n == 0
def test10(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("port [ name = port1]"))
assert n == 1
def test10a(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("port [ name = port1 //10/1]"))
assert n == 1
def test11(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("port [ name = port3 //2/1]"))
assert n == 1
def test11a(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("port [ name = port33]"))
assert n == 0
def test11b(self):
self.createModel()
selector = NodeSelector(self.root)
n = selector.select(Selector("port [ name = port3 4]"))
assert n == 0
class TestSet:
def createModel(self):
self.dm = DataModel()
self.dm.new("dummy-session", [], [])
self.root = project1 = self.dm.insert("project1",
{"object_type": "project"})
self.port1 = self.dm.insert("port1", {
"object_type": "port",
"name": "port1 //abc"
}, project1)
self.port2 = self.dm.insert("port2", {
"object_type": "port",
"name": "port11"
}, project1)
self.port3 = self.dm.insert("port3", {
"object_type": "port",
"name": "port3"
}, project1)
dev1 = self.dm.insert("emulateddevice1", {
"object_type": "emulateddevice",
"name": "Device 1"
}, project1)
self.dm.insert("ipv4if1", {
"object_type": "ipv4if",
"name": "ipv4if 1"
}, dev1)
dev2 = self.dm.insert("emulateddevice2",
{"object_type": "emulateddevice"}, project1)
self.dm.insert("ipv4if2", {
"object_type": "ipv4if",
"name": "ipv4if 2"
}, dev2)
def test_extend(self):
self.createModel()
set1 = NodeSelector(self.root)
set2 = NodeSelector(self.root)
set3 = NodeSelector(self.root)
set1.select(Selector("port[name=port1]"))
set2.select(Selector("port[name=port11]"))
set3.select(Selector("port[name=nope]"))
set1.extend(set2)
assert set1.count() == 2
assert set2.count() == 1
set1.extend(set3)
assert set1.count() == 2
assert set3.count() == 0
set1.extend(None)
assert set1.count() == 2
def test_different(self):
self.createModel()
set1 = NodeSelector(self.root)
set2 = NodeSelector(self.root)
set3 = NodeSelector(self.root)
set1.select(Selector("port[name=port1]"))
set2.select(Selector("port[name=port1]"))
set3.select(Selector("port[name=port11]"))
assert set1.isDifferent(set2) == False
assert set1.isDifferent(set3) == True
assert set1.isDifferent(None) == True
assert NodeSelector().isDifferent(set1)
assert NodeSelector().isDifferent(NodeSelector()) == False
assert NodeSelector().isDifferent(None)
assert set1.isDifferent(NodeSelector())
def test_intersect(self):
self.createModel()
set1 = NodeSelector(self.root)
set2 = NodeSelector(self.root)
set3 = NodeSelector(self.root)
set1.select(Selector("port"))
set2.select(Selector("port[name=port1]"))
set3.select(Selector("port[name=nope]"))
assert set1.intersect(set2).count() == 1
assert set1.intersect(set3).count() == 0
assert set1.intersect(None).count() == 0
class TestLinker:
def createModel(self):
self.dm = DataModel()
self.dm.new("dummy-session", [])
self.root = project1 = self.dm.insert("project1",
{"object_type": "project"})
self.port1 = port1 = self.dm.insert("port1", {
"object_type": "port",
"name": "port 1"
}, project1)
self.port2 = port2 = self.dm.insert("port2", {
"object_type": "port",
"name": "port 2"
}, project1)
self.dev1 = self.dm.insert("emulateddevice1", {
"object_type": "emulateddevice",
"name": "Device 1"
}, port1)
self.ip1 = self.dm.insert("ipv4if1", {
"object_type": "ipv4if",
"name": "ipv4if 1"
}, self.dev1)
self.dev2 = self.dm.insert("emulateddevice2", {
"object_type": "emulateddevice",
"name": "dev 2",
"count": 3
}, port1)
self.ip2 = self.dm.insert("ipv4if2", {
"object_type": "ipv4if",
"name": "ipv4if 2"
}, self.dev2)
self.dev3 = self.dm.insert("emulateddevice2", {
"object_type": "emulateddevice",
"name": "dev 3"
}, port2)
self.ip3 = self.dm.insert("ipv4if2", {
"object_type": "ipv4if",
"name": "ipv4if 3"
}, self.dev3)
return self.dm
def test1(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name=port 1]")
assert nodes.count() == 1 and nodes.get(0) == self.port1
def test2(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name=port 1]/emulateddevice")
assert nodes.count() == 2 and nodes.get(0) == self.dev1
def test3a(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name=port 1]/emulateddevice")
assert nodes.count() == 2 and nodes.get(0) == self.dev1 and nodes.get(
1) == self.dev2
def test3b(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name=port 1]/emulateddevice[*]")
assert nodes.count() == 2 and nodes.get(0) == self.dev1 and nodes.get(
1) == self.dev2
def test4(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name=port 1]/emulateddevice[*]/ipv4if")
assert nodes.count() == 2 and nodes.get(0) == self.ip1 and nodes.get(
1) == self.ip2
def test5(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name!= port 1]/emulateddevice/ipv4if")
assert nodes.count() == 1 and nodes.get(0) == self.ip3
def test7a(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name ~= Port]/emulateddevice/ipv4if")
assert nodes.count() == 3
def test7b(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name *= ort]/emulateddevice/ipv4if")
assert nodes.count() == 3
def test7c(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name ^= port]/emulateddevice/ipv4if")
assert nodes.count() == 3
def test7d(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[name ^= ort]/emulateddevice/ipv4if")
assert nodes == None
def test8a(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[0]/emulateddevice/ipv4if")
assert nodes.count() == 2
def test8b(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[1]/emulateddevice/ipv4if")
assert nodes.count() == 1
def test8c(self):
linker = Linker(self.createModel())
nodes = linker._resolve("/port[2]/emulateddevice/ipv4if")
assert nodes == None
def test9a(self):
linker = Linker(self.createModel())
nodes = linker._resolve(
"/port/emulateddevice[name *= dev][count = 3]/ipv4if")
assert nodes.count() == 1