Skip to content

Commit 4f22dd3

Browse files
committed
fix: handle filter value format based on Django version
1 parent 4de71f7 commit 4f22dd3

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

course_discovery/apps/course_metadata/tests/test_admin.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import ddt
44
import pytest
55
from bs4 import BeautifulSoup
6+
from django import VERSION as DJANGO_VERSION
67
from django.contrib.admin.sites import AdminSite
78
from django.contrib.contenttypes.models import ContentType
89
from 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

Comments
 (0)