33import ddt
44import pytest
55from bs4 import BeautifulSoup
6+ from django import VERSION as DJANGO_VERSION
67from django .contrib .admin .sites import AdminSite
78from django .contrib .contenttypes .models import ContentType
89from django .http import HttpRequest
@@ -485,7 +486,8 @@ def test_queryset_method_returns_eligible_programs(self):
485486 """ Verify that one click purchase eligible programs pass the filter. """
486487 verified_seat_type = factories .SeatTypeFactory .verified ()
487488 program_type = factories .ProgramTypeFactory (applicable_seat_types = [verified_seat_type ])
488- program_filter = ProgramEligibilityFilter (None , {self .parameter_name : 1 }, None , None )
489+ value = [1 ] if DJANGO_VERSION >= (5 , 2 ) else 1
490+ program_filter = ProgramEligibilityFilter (None , {self .parameter_name : value }, None , None )
489491 course_run = factories .CourseRunFactory (end = None , enrollment_end = None ,)
490492 factories .SeatFactory (course_run = course_run , type = verified_seat_type , upgrade_deadline = None )
491493 one_click_purchase_eligible_program = factories .ProgramFactory (
@@ -498,7 +500,8 @@ def test_queryset_method_returns_eligible_programs(self):
498500
499501 def test_queryset_method_returns_ineligible_programs (self ):
500502 """ Verify programs ineligible for one-click purchase do not pass the filter. """
501- program_filter = ProgramEligibilityFilter (None , {self .parameter_name : 0 }, None , None )
503+ value = [0 ] if DJANGO_VERSION >= (5 , 2 ) else 0
504+ program_filter = ProgramEligibilityFilter (None , {self .parameter_name : value }, None , None )
502505 one_click_purchase_ineligible_program = factories .ProgramFactory (one_click_purchase_enabled = False )
503506 with self .assertNumQueries (4 ):
504507 assert list (program_filter .queryset ({}, Program .objects .all ())) == [one_click_purchase_ineligible_program ]
0 commit comments