class TestDomainResource(unittest.TestCase): def setUp(self): self.zone = MagicMock() self.dnsserver = MagicMock() self.resource = DomainResource(self.zone, self.dnsserver) def test_GET(self): data = [ ("A", "www", "192.168.0.1"), ("A", "x", "192.168.0.2"), ] self.zone.a_records = MagicMock(return_value=data) rv = self.resource.render_GET(None) self.assertEqual(rv, "\n".join(["%s %s %s" % (x, y, z) for (x, y, z) in data]))
def setUp(self): self.zone = MagicMock() self.dnsserver = MagicMock() self.resource = DomainResource(self.zone, self.dnsserver)