fixes for failing tests

Author: SamWheating

pydruid/query.py

@@ -246,6 +246,8 @@ def parse_datasource(datasource, query_type):
                 "Datasource definition not valid. Must be string or "
                 "dict or list of strings"
             )
+        if isinstance(datasource, list):
+            return {"type": "union", "dataSources": datasource}
         return datasource
 
     @staticmethod

tests/db/test_cursor.py

@@ -355,7 +355,7 @@ def test_ssl_client_cert_authentication_with_patch_imported(
             ANY,
             stream=True,
             headers={"Content-Type": "application/json"},
-            json={"query": "SELECT * FROM table", "context": {}, "header": False},
+            json={"query": "SELECT * FROM table", "context": {}, "header": True},
             auth=ANY,
             verify=True,
             cert="path/to/cert",

tests/db/test_druid_dialect.py

@@ -4,7 +4,7 @@
 import warnings
 from unittest.mock import patch
 
-from sqlalchemy import exc, types
+from sqlalchemy import exc, types, text
 
 from pydruid.db.sqlalchemy import DruidDialect
 
@@ -91,7 +91,9 @@ def test_do_ping_success(self, connection_mock):
         result = self.dialect.do_ping(connection_mock)
 
         # asserts the ping executes with a raw string
-        connection_mock.execute.assert_called_once_with("SELECT 1")
+        connection_mock.execute.assert_called_once()
+        called_query = connection_mock.execute.call_args[0][0]
+        self.assertEqual(called_query.text, "SELECT 1")
         self.assertTrue(result)
 
     @patch("pydruid.db.api.Connection")