def test_restrict(self):
mgr = ntp_mgr()
old_restrict_list = mgr.get_restrict_list()
try:
new_restrict_list = [
{
"restrict_addr": "192.168.222.1",
"mask": "255.255.255.0"
}
]
mgr.set_restrict_list(new_restrict_list)
restrict_list = mgr.get_restrict_list()
self.assertEquals(len(restrict_list), 1)
restrict_conf = restrict_list[0]
self.assertEquals(restrict_conf["restrict_addr"], "192.168.222.1")
self.assertEquals(restrict_conf["mask"], "255.255.255.0")
new_restrict_list = [
{
"restrict_addr": "192.168.223.1",
"mask": "255.255.0.0"
}
]
mgr.set_restrict_list(new_restrict_list)
restrict_list = mgr.get_restrict_list()
self.assertEquals(len(restrict_list), 1)
restrict_conf = restrict_list[0]
self.assertEquals(restrict_conf["restrict_addr"], "192.168.223.1")
self.assertEquals(restrict_conf["mask"], "255.255.0.0")
finally:
mgr.set_restrict_list(old_restrict_list)
def test_server(self):
mgr = ntp_mgr()
old_server_list = mgr.get_server_conf_list()
try:
new_server_list = [
{
"server_addr": "0.asia.pool.ntp.org",
"prefer": True
}
]
mgr.set_server_conf_list(new_server_list)
server_list = mgr.get_server_conf_list()
self.assertEquals(len(server_list), 1)
server_conf = server_list[0]
self.assertEquals(server_conf["server_addr"], "0.asia.pool.ntp.org")
self.assertTrue(server_conf["prefer"])
new_server_list = [
{
"server_addr": "1.asia.pool.ntp.org",
"prefer": False
}
]
mgr.set_server_conf_list(new_server_list)
server_list = mgr.get_server_conf_list()
self.assertEquals(len(server_list), 1)
server_conf = server_list[0]
self.assertEquals(server_conf["server_addr"], "1.asia.pool.ntp.org")
self.assertFalse(server_conf["prefer"])
finally:
mgr.set_server_conf_list(old_server_list)