add missing urllib submodules imports

druid-io · Jan 28, 2022 · 2d79d34 · 2d79d34
1 parent 8e43199
commit 2d79d34
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/pydruid/client.py b/pydruid/client.py
@@ -15,7 +15,8 @@
 #
 import json
 import re
-import urllib
+import urllib.error
+import urllib.request
 from base64 import b64encode
 
 from pydruid.query import QueryBuilder

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -1,6 +1,6 @@
 # -*- coding: UTF-8 -*-
 import textwrap
-import urllib
+import urllib.error
 from io import StringIO
 from unittest.mock import Mock, patch
 
@@ -89,8 +89,8 @@ def test_druid_returns_html_error(self, mock_urlopen):
             str(e.value)
             == textwrap.dedent(
                 """
-            HTTP Error 500: Internal Server Error 
-             Druid Error: javax.servlet.ServletException: java.lang.OutOfMemoryError: GC overhead limit exceeded 
+            HTTP Error 500: Internal Server Error
+             Druid Error: javax.servlet.ServletException: java.lang.OutOfMemoryError: GC overhead limit exceeded
              Query is: {
                 "aggregations": [
                     {