From 57017b80c73a148a8f8e2103622f29625a81cc5b Mon Sep 17 00:00:00 2001
From: Michal Charemza <michal@charemza.name>
Date: Thu, 16 Jan 2020 16:11:33 +0000
Subject: [PATCH] (feat) Test non existant domain is proxied

---
 test.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/test.py b/test.py
index 76fd237..47cfbf9 100644
--- a/test.py
+++ b/test.py
@@ -8,6 +8,7 @@ from aiodnsresolver import (
     Resolver,
     IPv4AddressExpiresAt,
     DnsResponseCode,
+    DnsRecordDoesNotExist,
 )
 from dnsrewriteproxy import (
     DnsProxy,
@@ -62,16 +63,15 @@ class TestProxy(unittest.TestCase):
         self.assertTrue(isinstance(response[0], IPv4AddressExpiresAt))
 
     @async_test
-    async def test_e2e_default_resolver_match_all(self):
-        resolve, clear_cache = Resolver()
+    async def test_e2e_default_resolver_match_all_non_existing_domain(self):
+        resolve, clear_cache = get_resolver(53)
         self.add_async_cleanup(clear_cache)
         start = DnsProxy(rules=((r'(^.*$)', r'\1'),))
         stop = await start()
         self.add_async_cleanup(stop)
 
-        response = await resolve('www.google.com', TYPES.A)
-
-        self.assertTrue(isinstance(response[0], IPv4AddressExpiresAt))
+        with self.assertRaises(DnsRecordDoesNotExist):
+            await resolve('doesnotexist.charemza.name', TYPES.A)
 
 
 def get_socket(port):