Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 11 additions & 6 deletions backend/contributions/tests/test_pagination.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@
from rest_framework.test import APIClient
from rest_framework import status
from contributions.models import SubmittedContribution, ContributionType
from stewards.models import Steward
from datetime import date
from stewards.models import Steward, StewardPermission
from django.utils import timezone

User = get_user_model()

Expand All @@ -20,7 +20,7 @@ def setUp(self):
address='0x1234567890123456789012345678901234567890'
)
# Create the Steward profile
Steward.objects.create(user=self.steward)
self.steward_profile = Steward.objects.create(user=self.steward)

# Create a regular user for submissions
self.user = User.objects.create_user(
Expand All @@ -37,13 +37,18 @@ def setUp(self):
min_points=10,
max_points=100
)
StewardPermission.objects.create(
steward=self.steward_profile,
contribution_type=self.contribution_type,
action='accept'
)

# Create 37 submissions (to match the real scenario)
for i in range(37):
SubmittedContribution.objects.create(
user=self.user,
contribution_type=self.contribution_type,
contribution_date=date.today(),
contribution_date=timezone.now(),
notes=f'Test submission {i+1}',
state='pending'
)
Expand Down Expand Up @@ -102,7 +107,7 @@ def test_pagination_with_state_filter(self):
SubmittedContribution.objects.create(
user=self.user,
contribution_type=self.contribution_type,
contribution_date=date.today(),
contribution_date=timezone.now(),
notes=f'Accepted submission {i+1}',
state='accepted'
)
Expand All @@ -122,4 +127,4 @@ def test_pagination_max_page_size(self):
data = response.json()
self.assertEqual(data['count'], 37)
# Should be limited to max_page_size (100)
self.assertEqual(len(data['results']), 37) # All 37 items fit within max of 100
self.assertEqual(len(data['results']), 37) # All 37 items fit within max of 100