fix tests in Django 5.1

pinax · Jun 7, 2024 · 797ee5d · 797ee5d
1 parent 772de04
commit 797ee5d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/pinax/referrals/tests/test_models.py b/pinax/referrals/tests/test_models.py
@@ -19,8 +19,8 @@ def test_referral_responses_for_request_no_user(self):
         # Create a referral with a blank user
         request = RequestFactory().get("/referral/")
         request.session = self.client.session
-        baker.make("ReferralResponse", session_key=request.session.session_key)
-        baker.make("ReferralResponse", user=baker.make("User"))
+        baker.make("ReferralResponse", session_key=request.session.session_key, target=None, referral__target=None)
+        baker.make("ReferralResponse", user=baker.make("User"), target=None, referral__target=None)
         queryset = Referral.referral_responses_for_request(request)
         self.assertEqual(queryset.count(), 1)
         self.assertEqual(queryset.get().session_key, request.session.session_key)
@@ -31,8 +31,8 @@ def test_referral_responses_for_request_user(self):
         request = RequestFactory().get("/referral/")
         request.session = self.client.session
         request.user = baker.make("User")
-        baker.make("ReferralResponse", user=request.user, session_key="foo_bar")
-        baker.make("ReferralResponse", session_key="session_key")
+        baker.make("ReferralResponse", user=request.user, session_key="foo_bar", target=None, referral__target=None)
+        baker.make("ReferralResponse", session_key="session_key", target=None, referral__target=None)
         queryset = Referral.referral_responses_for_request(request)
         self.assertEqual(queryset.count(), 1)
         self.assertEqual(queryset.get().user, request.user)
@@ -41,14 +41,14 @@ def test_referral_responses_for_request_user(self):
     @override_settings(PINAX_REFERRALS_CODE_GENERATOR_CALLBACK=generate_code_callback)
     def test_code_generator_callback(self):
         user = baker.make("User", username="joe")
-        referral = baker.make("Referral", user=user, code=None)
+        referral = baker.make("Referral", user=user, code=None, target=None)
         # the callback function is being called in the save() method, inside baker.make
         self.assertEqual(referral.code, "recommended-by-joe")
 
     @override_settings(PINAX_REFERRALS_CODE_GENERATOR_CALLBACK=legacy_generate_code_callback)
     def test_legacy_code_generator_callback(self):
         # the new callback signature accepts two parameters, but the old callback signature
         # with just the referral_class as parameter should still work
-        referral = baker.make("Referral", code=None)
+        referral = baker.make("Referral", code=None, target=None)
         # the callback function is being called in the save() method, inside baker.make
         self.assertEqual(referral.code, "some-generated-code")