diff --git a/test/unit/common/middleware/test_cname_lookup.py b/test/unit/common/middleware/test_cname_lookup.py index ff40232c17..7983ce2007 100644 --- a/test/unit/common/middleware/test_cname_lookup.py +++ b/test/unit/common/middleware/test_cname_lookup.py @@ -298,7 +298,9 @@ class TestCNAMELookup(unittest.TestCase): resp = do_test('c.badtest.com') self.assertEqual(resp, bad_domain) - def test_host_is_storage_domain(self): + @mock.patch('dns.resolver.Resolver.query', + side_effect=dns.exception.DNSException) + def test_host_is_storage_domain(self, mock_lookup): conf = {'storage_domain': 'storage.example.com', 'lookup_depth': 2} app = cname_lookup.CNAMELookupMiddleware(FakeApp(), conf) @@ -311,9 +313,12 @@ class TestCNAMELookup(unittest.TestCase): bad_domain = ['CNAME lookup failed to resolve to a valid domain'] resp = do_test('c.badtest.com') self.assertEqual(resp, bad_domain) + self.assertEqual(1, len(mock_lookup.mock_calls)) + mock_lookup.reset_mock() resp = do_test('storage.example.com') self.assertEqual(resp, ['FAKE APP']) + self.assertEqual(0, len(mock_lookup.mock_calls)) def test_resolution_to_storage_domain_exactly(self): conf = {'storage_domain': 'example.com',