def test_format_map(self):
domain = Domain()
domain.domain_id = 252
domain.domain = "example.com"
group = Group()
group.group_id = 252
group.name = "Test Group 1"
map = DomainGroupMap()
map.map_id = 1
map.group_id = group
map.domain_id = domain
map.permissions = 7
formatted = map.format_map(map)
self.assertTrue(formatted["can_read"])
self.assertTrue(formatted["can_write"])
self.assertTrue(formatted["can_delete"])
self.assertEquals(formatted["map_id"], map.map_id)
self.assertEquals(formatted["group"], map.group_id.to_clean_dict())
self.assertEquals(formatted["domain"], map.domain_id.to_clean_dict())
def test_soa_record(self):
domain = Domain()
domain.domain = 'example.com'
model = Record()
# joined domain on domain_id
model.domain_id = domain
model.type = 'S'
model.host = 'hostmaster.example.com:ns1.example.com'
model.val = '16384:2048:1048576:2560:'
model.ttl = '86400'
expected = ("Zexample.com"
":ns1.example.com"
":hostmaster.example.com"
":"
":16384"
":2048"
":1048576"
":2560"
":86400\n")
self.assertEquals(self.export.data_line_from_model(model), expected)
