def test_save(self):
os.mkdir("savedir")
self.a = RuntimeAuthority("foo", "savedir")
self.a.set_record("baz", "A", {'address': "192.168.1.2"}, True)
self.a.set_record("bar", "A", {'address': "192.168.1.1"}, True)
self.assertEqual(json.load(open("savedir/foo")), [{
"bar": {
"type": "A",
"address": "192.168.1.1",
}
}, {
"baz": {
"type": "A",
"address": "192.168.1.2",
}
}])
def test_load(self):
os.mkdir("savedir")
json.dump((
{
"bar": {
"type": "A",
"address": "192.168.1.1",
}
},
{
"baz": {
"type": "A",
"address": "192.168.1.2",
}
},
), open("savedir/foo", "w"))
self.a = RuntimeAuthority("foo", "savedir")
self.assertEqual(
self.a.records, {
"bar.foo": [Record_A(address="192.168.1.1")],
"baz.foo": [Record_A(address="192.168.1.2")],
})
os.unlink("savedir/foo")
os.rmdir("savedir")
def setUp(self):
self.a = RuntimeAuthority("foo", None)