fix: update SQL parameter extraction to use URL parsing in query SQL tests

Author: Samson Gebre

tests/unit/data/test_select_star_expansion.py

@@ -5,6 +5,7 @@
 
 import pytest
 from unittest.mock import MagicMock
+from urllib.parse import parse_qs, urlparse
 
 from PowerPlatform.Dataverse.data._odata import _ODataClient
 
@@ -210,7 +211,8 @@ def test_query_sql_expands_select_star(self):
         c._list_columns.assert_called_once()
         # Verify the SQL sent to server has explicit columns, not *
         call_args = c._request.call_args
-        sent_sql = call_args[1]["params"]["sql"] if "params" in call_args[1] else call_args[0][2]["sql"]
+        sent_url = call_args[0][1]
+        sent_sql = parse_qs(urlparse(sent_url).query)["sql"][0]
         assert "*" not in sent_sql or "COUNT(*)" in sent_sql
         assert "accountid" in sent_sql
         assert "name" in sent_sql

tests/unit/data/test_sql_guardrails.py

@@ -7,6 +7,7 @@
 
 import pytest
 from unittest.mock import MagicMock
+from urllib.parse import parse_qs, urlparse
 
 from PowerPlatform.Dataverse.core.errors import ValidationError
 from PowerPlatform.Dataverse.data._odata import _ODataClient
@@ -363,8 +364,8 @@ def test_no_top_injection_in_server_request(self):
         c._query_sql("SELECT name FROM account")
 
         call_args = c._request.call_args
-        sent_params = call_args[1].get("params", {})
-        sent_sql = sent_params.get("sql", "")
+        sent_url = call_args[0][1]
+        sent_sql = parse_qs(urlparse(sent_url).query)["sql"][0]
         # SDK should NOT inject TOP 5000
         assert "TOP 5000" not in sent_sql
         assert sent_sql == "SELECT name FROM account"
@@ -380,6 +381,6 @@ def test_explicit_top_preserved_in_server_request(self):
         c._query_sql("SELECT TOP 50 name FROM account")
 
         call_args = c._request.call_args
-        sent_params = call_args[1].get("params", {})
-        sent_sql = sent_params.get("sql", "")
+        sent_url = call_args[0][1]
+        sent_sql = parse_qs(urlparse(sent_url).query)["sql"][0]
         assert "TOP 50" in sent_sql