Ruff typing from this morning. WIP

5 files changed, 56 insertions, 56 deletions

diff --git a/tests/incite/mergers/foundations/test_enriched_session.py b/tests/incite/mergers/foundations/test_enriched_session.py
index ba11725..8254d81 100644
--- a/tests/incite/mergers/foundations/test_enriched_session.py
+++ b/tests/incite/mergers/foundations/test_enriched_session.py
@@ -26,20 +26,20 @@ class TestEnrichedSession:
     def test_base(
         self,
         client_no_amm,
-        product,
-        user_factory,
+        product: Product,
+        user_factory: Callable[..., User],
         wall_collection,
         session_collection,
         enriched_session_merge,
         thl_web_rr: PostgresConfig,
-        delete_df_collection,
+        delete_df_collection: Callable[..., None],
         incite_item_factory,
     ):
         from generalresearch.models.thl.user import User
 
         delete_df_collection(coll=session_collection)
 
-        u1: User = user_factory(product=product, created=session_collection.start)
+        u1: User = user_factory(product=product: Product, created=session_collection.start)
 
         for item in session_collection.items:
             incite_item_factory(item=item, user=u1)
@@ -52,7 +52,7 @@ class TestEnrichedSession:
             client=client_no_amm,
             wall_coll=wall_collection,
             session_coll=session_collection,
-            pg_config=thl_web_rr,
+            pg_config=thl_web_rr: PostgresConfig,
         )
 
         # --
@@ -92,11 +92,11 @@ class TestEnrichedSessionAdmin:
         session_collection,
         thl_web_rr: PostgresConfig,
         session_report_request,
-        user_factory,
+        user_factory: Callable[..., User],
         start,
         session_factory,
-        product_factory,
-        delete_df_collection,
+        product_factory: Callable[..., Product],
+        delete_df_collection: Callable[..., None],
     ):
         delete_df_collection(coll=wall_collection)
         delete_df_collection(coll=session_collection)
@@ -120,7 +120,7 @@ class TestEnrichedSessionAdmin:
             client=client_no_amm,
             session_coll=session_collection,
             wall_coll=wall_collection,
-            pg_config=thl_web_rr,
+            pg_config=thl_web_rr: PostgresConfig,
         )
 
         df = enriched_session_merge.to_admin_response(
diff --git a/tests/incite/mergers/foundations/test_enriched_task_adjust.py b/tests/incite/mergers/foundations/test_enriched_task_adjust.py
index 8c3a647..a33a55a 100644
--- a/tests/incite/mergers/foundations/test_enriched_task_adjust.py
+++ b/tests/incite/mergers/foundations/test_enriched_task_adjust.py
@@ -21,16 +21,16 @@ class TestEnrichedTaskAdjust:
     def test_base(
         self,
         client_no_amm,
-        user_factory,
-        product,
+        user_factory: Callable[..., User],
+        product: Product,
         task_adj_collection,
         wall_collection,
         session_collection,
         enriched_wall_merge,
         enriched_task_adjust_merge,
         incite_item_factory,
-        delete_df_collection,
-        thl_web_rr,
+        delete_df_collection: Callable[..., None],
+        thl_web_rr: PostgresConfig,
     ):
         from generalresearch.models.thl.user import User
 
@@ -48,14 +48,14 @@ class TestEnrichedTaskAdjust:
             client=client_no_amm,
             session_coll=session_collection,
             wall_coll=wall_collection,
-            pg_config=thl_web_rr,
+            pg_config=thl_web_rr: PostgresConfig,
         )
 
         enriched_task_adjust_merge.build(
             client=client_no_amm,
             task_adjust_coll=task_adj_collection,
             enriched_wall=enriched_wall_merge,
-            pg_config=thl_web_rr,
+            pg_config=thl_web_rr: PostgresConfig,
         )
 
         # --
diff --git a/tests/incite/mergers/foundations/test_enriched_wall.py b/tests/incite/mergers/foundations/test_enriched_wall.py
index 0e28bce..a0ca4dd 100644
--- a/tests/incite/mergers/foundations/test_enriched_wall.py
+++ b/tests/incite/mergers/foundations/test_enriched_wall.py
@@ -21,13 +21,13 @@ class TestEnrichedWall:
     def test_base(
         self,
         client_no_amm,
-        product,
-        user_factory,
+        product: Product,
+        user_factory: Callable[..., User],
         wall_collection,
-        thl_web_rr,
+        thl_web_rr: PostgresConfig,
         session_collection,
         enriched_wall_merge,
-        delete_df_collection,
+        delete_df_collection: Callable[..., None],
         incite_item_factory,
     ):
         from generalresearch.models.thl.user import User
@@ -35,7 +35,7 @@ class TestEnrichedWall:
         # -- Build & Setup
         delete_df_collection(coll=session_collection)
         delete_df_collection(coll=wall_collection)
-        u1: User = user_factory(product=product, created=session_collection.start)
+        u1: User = user_factory(product=product: Product, created=session_collection.start)
 
         for item in session_collection.items:
             incite_item_factory(item=item, user=u1)
@@ -48,7 +48,7 @@ class TestEnrichedWall:
             client=client_no_amm,
             wall_coll=wall_collection,
             session_coll=session_collection,
-            pg_config=thl_web_rr,
+            pg_config=thl_web_rr: PostgresConfig,
         )
 
         # --
@@ -64,18 +64,18 @@ class TestEnrichedWall:
     def test_base_item(
         self,
         client_no_amm,
-        product,
-        user_factory,
+        product: Product,
+        user_factory: Callable[..., User],
         wall_collection,
         session_collection,
         enriched_wall_merge,
-        delete_df_collection,
-        thl_web_rr,
+        delete_df_collection: Callable[..., None],
+        thl_web_rr: PostgresConfig,
         incite_item_factory,
     ):
         # -- Build & Setup
         delete_df_collection(coll=session_collection)
-        u = user_factory(product=product, created=session_collection.start)
+        u = user_factory(product=product: Product, created=session_collection.start)
 
         for item in session_collection.items:
             incite_item_factory(item=item, user=u)
@@ -87,7 +87,7 @@ class TestEnrichedWall:
             client=client_no_amm,
             wall_coll=wall_collection,
             session_coll=session_collection,
-            pg_config=thl_web_rr,
+            pg_config=thl_web_rr: PostgresConfig,
         )
 
         # --
@@ -106,7 +106,7 @@ class TestEnrichedWall:
                 client=client_no_amm,
                 wall_coll=wall_collection,
                 session_coll=session_collection,
-                pg_config=thl_web_rr,
+                pg_config=thl_web_rr: PostgresConfig,
             )
             modified_time2 = path.stat().st_mtime
 
@@ -172,12 +172,12 @@ class TestEnrichedWallToAdmin:
         client_no_amm,
         wall_collection,
         session_collection,
-        thl_web_rr,
+        thl_web_rr: PostgresConfig,
         user,
         session_factory,
-        delete_df_collection,
-        product_factory,
-        user_factory,
+        delete_df_collection: Callable[..., None],
+        product_factory: Callable[..., Product],
+        user_factory: Callable[..., User],
         start,
     ):
         delete_df_collection(coll=wall_collection)
@@ -203,7 +203,7 @@ class TestEnrichedWallToAdmin:
             client=client_no_amm,
             wall_coll=wall_collection,
             session_coll=session_collection,
-            pg_config=thl_web_rr,
+            pg_config=thl_web_rr: PostgresConfig,
         )
 
         df = enriched_wall_merge.to_admin_response(
diff --git a/tests/incite/mergers/test_pop_ledger.py b/tests/incite/mergers/test_pop_ledger.py
index dc01179..d054eb6 100644
--- a/tests/incite/mergers/test_pop_ledger.py
+++ b/tests/incite/mergers/test_pop_ledger.py
@@ -33,17 +33,17 @@ class TestMergePOPLedger:
         client_no_amm,
         ledger_collection,
         pop_ledger_merge,
-        product,
-        user_factory,
-        create_main_accounts,
+        product: Product,
+        user_factory: Callable[..., User],
+        create_main_accounts: Callable[..., None],
         thl_lm,
-        delete_df_collection,
+        delete_df_collection: Callable[..., None],
         incite_item_factory,
-        delete_ledger_db,
+        delete_ledger_db: Callable[..., None],
     ):
         from generalresearch.models.thl.ledger import LedgerAccount
 
-        u = user_factory(product=product, created=ledger_collection.start)
+        u = user_factory(product=product: Product, created=ledger_collection.start)
 
         # -- Build & Setup
         delete_ledger_db()
@@ -127,26 +127,26 @@ class TestMergePOPLedger:
         ledger_collection,
         pop_ledger_merge,
         mnt_filepath,
-        product,
-        user_factory,
-        create_main_accounts,
+        product: Product,
+        user_factory: Callable[..., User],
+        create_main_accounts: Callable[..., None],
         offset,
         duration,
         start,
         thl_lm,
         incite_item_factory,
-        delete_df_collection,
-        delete_ledger_db,
+        delete_df_collection: Callable[..., None],
+        delete_ledger_db: Callable[..., None],
         session_collection,
     ):
         from generalresearch.models.thl.finance import ProductBalances
         from generalresearch.models.thl.ledger import LedgerAccount
         from generalresearch.models.thl.product import Product
 
-        u = user_factory(product=product, created=session_collection.start)
+        u = user_factory(product=product: Product, created=session_collection.start)
 
         assert ledger_collection.finished is not None
-        assert isinstance(u.product, Product)
+        assert isinstance(u.product: Product, Product)
         delete_ledger_db()
         create_main_accounts(),
         delete_df_collection(coll=ledger_collection)
@@ -228,14 +228,14 @@ class TestMergePOPLedger:
         ledger_collection,
         pop_ledger_merge,
         mnt_filepath,
-        user_factory,
-        product,
-        create_main_accounts,
+        user_factory: Callable[..., User],
+        product: Product,
+        create_main_accounts: Callable[..., None],
         offset,
         duration,
         start,
         thl_lm,
-        delete_df_collection,
+        delete_df_collection: Callable[..., None],
         incite_item_factory,
     ):
         from generalresearch.models.thl.user import User
diff --git a/tests/incite/mergers/test_ym_survey_merge.py b/tests/incite/mergers/test_ym_survey_merge.py
index 850df8a..a0b8b87 100644
--- a/tests/incite/mergers/test_ym_survey_merge.py
+++ b/tests/incite/mergers/test_ym_survey_merge.py
@@ -28,20 +28,20 @@ class TestYMSurveyMerge:
     def test_base(
         self,
         client_no_amm,
-        user_factory,
-        product,
+        user_factory: Callable[..., User],
+        product: Product,
         ym_survey_wall_merge,
         wall_collection,
         session_collection,
         enriched_session_merge,
-        delete_df_collection,
+        delete_df_collection: Callable[..., None],
         incite_item_factory,
-        thl_web_rr,
+        thl_web_rr: PostgresConfig,
     ):
         from generalresearch.models.thl.user import User
 
         delete_df_collection(coll=session_collection)
-        user: User = user_factory(product=product, created=session_collection.start)
+        user: User = user_factory(product=product: Product, created=session_collection.start)
 
         # -- Build & Setup
         assert ym_survey_wall_merge.start is None
@@ -61,7 +61,7 @@ class TestYMSurveyMerge:
             client=client_no_amm,
             session_coll=session_collection,
             wall_coll=wall_collection,
-            pg_config=thl_web_rr,
+            pg_config=thl_web_rr: PostgresConfig,
         )
         assert enriched_session_merge.progress.has_archive.eq(True).all()