Ben van Basten 1 rok pred
rodič
commit
623376c34b
1 zmenil súbory, kde vykonal 9 pridanie a 2 odobranie
  1. 9 2
      clean_python/sql/testing.py

+ 9 - 2
clean_python/sql/testing.py

@@ -1,6 +1,9 @@
 from contextlib import asynccontextmanager
 from contextlib import asynccontextmanager
+from typing import Any
 from typing import AsyncIterator
 from typing import AsyncIterator
+from typing import Dict
 from typing import List
 from typing import List
+from typing import Optional
 from unittest import mock
 from unittest import mock
 
 
 from sqlalchemy.dialects import postgresql
 from sqlalchemy.dialects import postgresql
@@ -17,7 +20,9 @@ class FakeSQLDatabase(SQLProvider):
         self.queries: List[List[Executable]] = []
         self.queries: List[List[Executable]] = []
         self.result = mock.Mock(return_value=[])
         self.result = mock.Mock(return_value=[])
 
 
-    async def execute(self, query: Executable) -> List[Json]:
+    async def execute(
+        self, query: Executable, _: Optional[Dict[str, Any]]
+    ) -> List[Json]:
         self.queries.append([query])
         self.queries.append([query])
         return self.result()
         return self.result()
 
 
@@ -33,7 +38,9 @@ class FakeSQLTransaction(SQLProvider):
         self.queries: List[Executable] = []
         self.queries: List[Executable] = []
         self.result = result
         self.result = result
 
 
-    async def execute(self, query: Executable) -> List[Json]:
+    async def execute(
+        self, query: Executable, _: Optional[Dict[str, Any]]
+    ) -> List[Json]:
         self.queries.append(query)
         self.queries.append(query)
         return self.result()
         return self.result()