From 91fc8dacf203a874b1a4bc6d4cd739b18c61ffff Mon Sep 17 00:00:00 2001 From: Jolan Date: Thu, 27 Mar 2025 13:11:26 +0100 Subject: [PATCH] chore: minor update to let Mauritz work on it --- backend/tests/test_workout_file_upload.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/backend/tests/test_workout_file_upload.py b/backend/tests/test_workout_file_upload.py index 2d6dbe8..9cc6e8c 100644 --- a/backend/tests/test_workout_file_upload.py +++ b/backend/tests/test_workout_file_upload.py @@ -1,20 +1,23 @@ import os -from users.models import User +from rest_framework.test import APIClient from django.core.files.uploadedfile import SimpleUploadedFile from django.test import TestCase -from myapp.models import WorkoutFile # Replace with the actual model import +from workouts.models import WorkoutFile # Replace with the actual model import from django.urls import reverse +from django.contrib.auth import get_user_model + class TestWorkoutFileUpload(TestCase): def setUp(self): - self.url = reverse('workout-file-list') # Use 'workout-file-list' from urls.py - self.user = User.objects.create_user( - username='testuser', password='password' - ) - self.client.login(username='testuser', password='password') + self.client = APIClient() + self.user = get_user_model().objects.create_user(username=os.getenv("DJANGO_SUPERUSER_USERNAME"), password=os.getenv("DJANGO_SUPERUSER_PASSWORD")) + self.client.force_authenticate(self.user) + + self.url = reverse("workout-file-list") def test_upload_multiple_valid_pdf_files(self): + print(self.client.get(self.url).status_code) # Create mock PDF files under 2MB files = {} for i in range(8): # 8 files