class TestRootResource(unittest.TestCase):
def setUp(self):
self.config = MagicMock()
self.dnsserver = MagicMock()
self.resource = RootResource(self.config, self.dnsserver)
def test_get(self):
self.dnsserver.zones = MagicMock(return_value=["foo", "bar"])
rv = self.resource.render_GET(None)
self.assertEqual(rv, "\n".join(["foo", "bar"]))
def test_getChild_exists(self):
self.config.get = MagicMock(return_value="")
zone = MagicMock()
def get_zone(x):
if x == "foo":
return zone
raise KeyError
self.dnsserver.get_zone.side_effect = get_zone
rv = self.resource.getChild("foo", None)
self.assert_(isinstance(rv, DomainResource))
self.assertEqual(rv.zone, zone)
rv = self.resource.getChild("bar", None)
self.assert_(isinstance(rv, MissingDomainResource))
self.assertEqual(rv.name, "bar")
def test_getChild_exists_with_lockdown(self):
self.config.get = MagicMock(return_value="foo bar")
zone = MagicMock()
def get_zone(x):
if x == "foo":
return zone
raise KeyError
self.dnsserver.get_zone.side_effect = get_zone
rv = self.resource.getChild("foo", None)
self.assert_(isinstance(rv, DomainResource))
self.assertEqual(rv.zone, zone)
rv = self.resource.getChild("bar", None)
self.assert_(isinstance(rv, MissingDomainResource))
self.assertEqual(rv.name, "bar")
rv = self.resource.getChild("baz", None)
self.assert_(isinstance(rv, ForbiddenDomainResource))
def setUp(self):
self.config = MagicMock()
self.dnsserver = MagicMock()
self.resource = RootResource(self.config, self.dnsserver)