CRF: add missing handlers

edgedb · Oct 2, 2024 · cb448c4 · cb448c4
1 parent af8eb76
commit cb448c4
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1 deletion.
diff --git a/edgedb/abstract.py b/edgedb/abstract.py
@@ -204,7 +204,7 @@ def query(self, query: str, *args, **kwargs) -> list:
  query_options=_query_opts,
  retry_options=self._get_retry_options(),
  state=self._get_state(),
- warning_handlerr=self._get_warning_handler(),
+ warning_handler=self._get_warning_handler(),
  ))
 
  def query_single(

diff --git a/edgedb/transaction.py b/edgedb/transaction.py
@@ -201,6 +201,7 @@ async def _privileged_execute(self, query: str) -> None:
  query=abstract.QueryWithArgs(query, (), {}),
  cache=self._get_query_cache(),
  state=self._get_state(),
+ warning_handler=self._get_warning_handler(),
  ))
 
 

diff --git a/tests/test_async_query.py b/tests/test_async_query.py
@@ -1035,6 +1035,7 @@ async def test_json_elements(self):
  ),
  retry_options=None,
  state=None,
+ warning_handler=lambda _ex, _: None,
  )
  )
  self.assertEqual(

diff --git a/tests/test_sync_query.py b/tests/test_sync_query.py
@@ -792,6 +792,7 @@ def test_json_elements(self):
  ),
  retry_options=None,
  state=None,
+ warning_handler=lambda _ex, _: None,
  )
  )
  )