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)