def test_we_handle_the_host_zone(self):
response = Mock()
response.get.return_value = StubAWSResponse()
zone_id = dns.get_zone_id(response, "non-existent-domain.com")
self.assertEquals('XXXXXXXX', zone_id)
def test_we_raise_an_exception_if_we_have_no_zone_id(self):
try:
dns.get_zone_id(None, "non-existent-domain.com")
except Exception, e:
self.assertEquals("The domain non-existent-domain.com is not registered with AWS", e.message)