From 2e8a3996687a6bf095e3f0a5879c8fee4965840f Mon Sep 17 00:00:00 2001 From: Samuel Paccoud - DINUM Date: Sat, 4 Jan 2025 23:52:11 +0100 Subject: [PATCH] =?UTF-8?q?=E2=9C=85(backend)=20override=20defaults.NB=5FO?= =?UTF-8?q?BJECTS=20for=20the=20demo=20tests?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Otherwise when you increase the number of objects to test how the application scales, the demo test will take too long... --- src/backend/demo/tests/test_commands_create_demo.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/backend/demo/tests/test_commands_create_demo.py b/src/backend/demo/tests/test_commands_create_demo.py index 66f227bc..3cd52e17 100644 --- a/src/backend/demo/tests/test_commands_create_demo.py +++ b/src/backend/demo/tests/test_commands_create_demo.py @@ -1,4 +1,5 @@ """Test the `create_demo` management command""" +from unittest import mock from django.core.management import call_command from django.test import override_settings @@ -10,15 +11,20 @@ from core import models pytestmark = pytest.mark.django_db +@mock.patch("demo.defaults.NB_OBJECTS", { + "users": 10, + "docs": 10, + "max_users_per_document": 5, + } ) @override_settings(DEBUG=True) def test_commands_create_demo(): """The create_demo management command should create objects as expected.""" call_command("create_demo") assert models.Template.objects.count() == 1 - assert models.User.objects.count() >= 50 - assert models.Document.objects.count() >= 50 - assert models.DocumentAccess.objects.count() > 50 + assert models.User.objects.count() >= 10 + assert models.Document.objects.count() >= 10 + assert models.DocumentAccess.objects.count() > 10 # assert dev users have doc accesses user = models.User.objects.get(email="impress@impress.world")