Extensive use of type checking. Movement of pytest conf towards handling managers (for db agnostic unittest). Starting to organize pytests.

1 files changed, 6 insertions, 5 deletions

diff --git a/jb/managers/bonus.py b/jb/managers/bonus.py
index 0cb8b02..89b81f0 100644
--- a/jb/managers/bonus.py
+++ b/jb/managers/bonus.py
@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Any
 
 from psycopg import sql
 
@@ -37,12 +37,12 @@ class BonusManager(PostgresManager):
                 c.execute(query, data)
                 res = c.fetchone()
                 conn.commit()
-        bonus.id = res["id"]
-        bonus.assignment_id = res["assignment_id"]
+        bonus.id = res["id"]  # type: ignore
+        bonus.assignment_id = res["assignment_id"]  # type: ignore
         return None
 
     def filter(self, amt_assignment_id: str) -> List[Bonus]:
-        res = self.pg_config.execute_sql_query(
+        res: List[Any] = self.pg_config.execute_sql_query(
             """
             SELECT mb.*, ma.amt_assignment_id
             FROM mtwerk_bonus mb
@@ -51,4 +51,5 @@ class BonusManager(PostgresManager):
         """,
             params={"amt_assignment_id": amt_assignment_id},
         )
-        return [Bonus.from_postgres(x) for x in res]
+
+        return [Bonus.from_postgres(data=x) for x in res]