From 3b4059135be47f7752a08e4277a85f9e57ceaa9d Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Tue, 25 Aug 2026 11:09:52 -0700
Subject: Ruff typing from this morning. WIP

---
 tests/managers/gr/test_business.py | 12 ++++++------
 tests/managers/gr/test_team.py     |  4 ++--
 2 files changed, 8 insertions(+), 8 deletions(-)

(limited to 'tests/managers/gr')

diff --git a/tests/managers/gr/test_business.py b/tests/managers/gr/test_business.py
index 74a5450..3490403 100644
--- a/tests/managers/gr/test_business.py
+++ b/tests/managers/gr/test_business.py
@@ -8,7 +8,7 @@ class TestBusinessBankAccountManager:
     def test_init(self, business_bank_account_manager, gr_db):
         assert business_bank_account_manager.pg_config == gr_db
 
-    def test_create(self, business, business_bank_account_manager):
+    def test_create(self, business: Business, business_bank_account_manager):
         from generalresearch.models.gr.business import (
             BusinessBankAccount,
             TransferMethod,
@@ -33,7 +33,7 @@ class TestBusinessBankAccountManager:
 
 class TestBusinessAddressManager:
 
-    def test_create(self, business, business_address_manager):
+    def test_create(self, business: Business, business_address_manager):
         from generalresearch.models.gr.business import BusinessAddress
 
         res = business_address_manager.create(uuid=uuid4().hex, business_id=business.id)
@@ -81,7 +81,7 @@ class TestBusinessManager:
         res = business_manager.get_by_user_id(user_id=gr_user.id)
         assert len(res) == 0
 
-        # Create a Business, but don't add it to anything
+        # Create a business: Business, but don't add it to anything
         b1 = business_manager.create_dummy()
         res = business_manager.get_by_user_id(user_id=gr_user.id)
         assert len(res) == 0
@@ -113,11 +113,11 @@ class TestBusinessManager:
     def test_get_uuids_by_user_id(self):
         pass
 
-    def test_get_by_uuid(self, business, business_manager):
+    def test_get_by_uuid(self, business: Business, business_manager):
         instance = business_manager.get_by_uuid(business_uuid=business.uuid)
         assert business.id == instance.id
 
-    def test_get_by_id(self, business, business_manager):
+    def test_get_by_id(self, business: Business, business_manager):
         instance = business_manager.get_by_id(business_id=business.id)
         assert business.uuid == instance.uuid
 
@@ -131,7 +131,7 @@ class TestBusinessManager:
     #     business = BusinessManager.create(
     #         uuid=b_uuid,
     #         name=f"test-{b_uuid[:6]}")
-    #     assert isinstance(business, Business)
+    #     assert isinstance(business: Business, Business)
     #
     #     # Try to make it again
     #     with pytest.raises(expected_exception=psycopg.errors.UniqueViolation):
diff --git a/tests/managers/gr/test_team.py b/tests/managers/gr/test_team.py
index 0918ab8..5e5c565 100644
--- a/tests/managers/gr/test_team.py
+++ b/tests/managers/gr/test_team.py
@@ -89,10 +89,10 @@ class TestTeamManager:
         gr_user_token,
         gr_user,
         membership,
-        product_factory,
+        product_factory: Callable[..., Product],
         membership_factory,
         team,
-        thl_web_rr,
+        thl_web_rr: PostgresConfig,
         gr_redis_config,
         gr_db,
     ):
-- 
cgit v1.2.3