From 1746411f8dd0b880b3d069afc0b008a176fd811a Mon Sep 17 00:00:00 2001
From: stuppie
Date: Mon, 12 Oct 2026 16:59:03 -0700
Subject: BusinessCreate model. Remove payouts from Product (query for me).
 Pagination and ordering of business and product payouts

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

(limited to 'tests/models')

diff --git a/tests/models/gr/test_business.py b/tests/models/gr/test_business.py
index 8de2a47..7e0932c 100644
--- a/tests/models/gr/test_business.py
+++ b/tests/models/gr/test_business.py
@@ -218,7 +218,9 @@ class TestBusiness:
         assert isinstance(gr_business.teams, list)
         assert len(gr_business.teams) == 0
 
-        gr_team_manager.add_business(team=gr_team, business=gr_business)
+        gr_team_manager.add_business(
+            team_id=gr_team.id, business_id=gr_business.id
+        )
         assert len(gr_business.teams) == 0
         gr_business.prefetch_teams(pg_config=gr_db)
         assert len(gr_business.teams) == 1
@@ -1246,7 +1248,9 @@ class TestBusinessMethods:
 
         p1 = product_factory(team=gr_team, business=gr_business)
         u1 = user_factory(product=p1)
-        gr_team_manager.add_business(team=gr_team, business=gr_business)
+        gr_team_manager.add_business(
+            team_id=gr_team.id, business_id=gr_business.id
+        )
 
         # Business needs tx & incite to build balance
         delete_ledger_db()
diff --git a/tests/models/gr/test_team.py b/tests/models/gr/test_team.py
index 7c93209..8cc00c8 100644
--- a/tests/models/gr/test_team.py
+++ b/tests/models/gr/test_team.py
@@ -117,7 +117,9 @@ class TestTeam:
         assert isinstance(gr_team.businesses, list)
         assert len(gr_team.businesses) == 0
 
-        gr_team_manager.add_business(team=gr_team, business=gr_business)
+        gr_team_manager.add_business(
+            team_id=gr_team.id, business_id=gr_business.id
+        )
         assert len(gr_team.businesses) == 0
         gr_team.prefetch_businesses(gr_business_manager=gr_business_manager)
         assert len(gr_team.businesses) == 1
-- 
cgit v1.2.3