From 797ee5dbef5ebcdc9234bd68f8123c7762bada8e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Petr=20Dlouh=C3=BD?= Date: Fri, 7 Jun 2024 15:25:07 +0200 Subject: [PATCH] fix tests in Django 5.1 --- pinax/referrals/tests/test_models.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/pinax/referrals/tests/test_models.py b/pinax/referrals/tests/test_models.py index e230b11..f0c6d46 100644 --- 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,7 +41,7 @@ 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") @@ -49,6 +49,6 @@ def test_code_generator_callback(self): 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")