try empty list

Author: blink1073

test/asynchronous/test_srv_polling.py

@@ -65,8 +65,8 @@ def enable(self):
         async def mock_get_hosts_and_min_ttl(resolver, *args):
             assert self.old_dns_resolver_response is not None
             nodes, ttl = await self.old_dns_resolver_response(resolver)
-            # if self.nodelist_callback is not None:
-            #     nodes = self.nodelist_callback()
+            if self.nodelist_callback is not None:
+                nodes = self.nodelist_callback()
             if self.ttl_time is not None:
                 ttl = self.ttl_time
             return nodes, ttl
@@ -220,7 +220,7 @@ async def test_dns_failures(self):
 
             def response_callback(*args):
                 # raise exc("DNS Failure!")
-                pass
+                return []
 
             print(exc)
             # await asyncio.sleep(0.5)

test/test_srv_polling.py

@@ -65,8 +65,8 @@ def enable(self):
         def mock_get_hosts_and_min_ttl(resolver, *args):
             assert self.old_dns_resolver_response is not None
             nodes, ttl = self.old_dns_resolver_response(resolver)
-            # if self.nodelist_callback is not None:
-            #     nodes = self.nodelist_callback()
+            if self.nodelist_callback is not None:
+                nodes = self.nodelist_callback()
             if self.ttl_time is not None:
                 ttl = self.ttl_time
             return nodes, ttl
@@ -220,7 +220,7 @@ def test_dns_failures(self):
 
             def response_callback(*args):
                 # raise exc("DNS Failure!")
-                pass
+                return []
 
             print(exc)
             # time.sleep(0.5)