Files
2026-01-04 12:28:28 +01:00

110 lines
3.4 KiB
Python
Executable File

#!/usr/bin/env python3
"""
Basic NTP client test
Verify NTP client with multiple servers, ensure one get selected.
"""
import infamy
import infamy.ntp_server as ntp_server
import infamy.ntp as ntp
import infamy.util as util
def config_target(dut, data1, data2, data3):
dut.put_config_dicts({
"ietf-interfaces": {
"interfaces": {
"interface": [{
"name": data1,
"enabled": True,
"ipv4": {
"address": [{
"ip": "192.168.1.2",
"prefix-length": 24
}]
}
}, {
"name": data2,
"enabled": True,
"ipv4": {
"address": [{
"ip": "192.168.2.2",
"prefix-length": 24
}]
}
}, {
"name": data3,
"enabled": True,
"ipv4": {
"address": [{
"ip": "192.168.3.2",
"prefix-length": 24
}]
}
}]
}
},
"ietf-system": {
"system": {
"ntp": {
"enabled": True,
"server": [{
"name": "Server1",
"udp": {
"address": "192.168.1.1"
},
"iburst": True
}, {
"name": "Server2",
"udp": {
"address": "192.168.2.1"
},
"iburst": True
}, {
"name": "Server3",
"udp": {
"address": "192.168.3.1"
},
"iburst": True
}]
}
}
}
})
with infamy.Test() as test:
with test.step("Set up topology and attach to target DUT"):
env = infamy.Env()
target = env.attach("target", "mgmt")
with test.step("Configure NTP client on 'target'"):
_, data1 = env.ltop.xlate("target", "data1")
_, data2 = env.ltop.xlate("target", "data2")
_, data3 = env.ltop.xlate("target", "data3")
config_target(target, data1, data2, data3)
_, hport1 = env.ltop.xlate("host", "data1")
_, hport2 = env.ltop.xlate("host", "data2")
_, hport3 = env.ltop.xlate("host", "data3")
with infamy.IsolatedMacVlan(hport1) as ns1, \
infamy.IsolatedMacVlan(hport2) as ns2, \
infamy.IsolatedMacVlan(hport3) as ns3:
ns1.addip("192.168.1.1")
ns2.addip("192.168.2.1")
ns3.addip("192.168.3.1")
with ntp_server.Server(ns1) as ntp1, \
ntp_server.Server(ns2) as ntp2, \
ntp_server.Server(ns3) as ntp3:
with test.step("Verify one source is in 'selected' state on 'target'"):
util.until(lambda: ntp.any_source_selected(target), attempts=200)
with test.step("Verify three sources exist in NTP client on 'target'"):
assert ntp.number_of_sources(target) == 3
test.succeed()