From ad4065e682c3317e844928d78346672ac021fadb Mon Sep 17 00:00:00 2001 From: Sabrina Demagny Date: Wed, 30 Oct 2024 01:11:52 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=A7(demo)=20modify=20data=20set=20to?= =?UTF-8?q?=20fix=20e2e=20tests?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit This part needs more refactoring. Demo data set needs to be no totally random. --- .../demo/management/commands/create_demo.py | 39 +++++++++++++++++++ .../demo/tests/test_commands_create_demo.py | 4 +- 2 files changed, 42 insertions(+), 1 deletion(-) diff --git a/src/backend/demo/management/commands/create_demo.py b/src/backend/demo/management/commands/create_demo.py index 97f989f..a4d2b4f 100755 --- a/src/backend/demo/management/commands/create_demo.py +++ b/src/backend/demo/management/commands/create_demo.py @@ -128,6 +128,45 @@ def create_demo(stdout): language=random.choice(settings.LANGUAGES)[0], ) ) + # this is a quick fix to fix e2e tests + # tests needs some no random data + queue.push( + models.User( + sub=uuid4(), + email="monique.test@example.com", + name="Monique Test", + password="!", + is_superuser=False, + is_active=True, + is_staff=False, + language=random.choice(settings.LANGUAGES)[0], + ) + ) + queue.push( + models.User( + sub=uuid4(), + email="jeanne.test@example.com", + name="Jean Test", + password="!", + is_superuser=False, + is_active=True, + is_staff=False, + language=random.choice(settings.LANGUAGES)[0], + ) + ) + queue.push( + models.User( + sub=uuid4(), + email="jean.somethingelse@example.com", + name="Jean Something", + password="!", + is_superuser=False, + is_active=True, + is_staff=False, + language=random.choice(settings.LANGUAGES)[0], + ) + ) + queue.flush() with Timeit(stdout, "Creating teams"): diff --git a/src/backend/demo/tests/test_commands_create_demo.py b/src/backend/demo/tests/test_commands_create_demo.py index f486275..b6ec35f 100644 --- a/src/backend/demo/tests/test_commands_create_demo.py +++ b/src/backend/demo/tests/test_commands_create_demo.py @@ -28,7 +28,9 @@ def test_commands_create_demo(): """The create_demo management command should create objects as expected.""" call_command("create_demo") - assert models.User.objects.count() == TEST_NB_OBJECTS["users"] + assert ( + models.User.objects.count() == TEST_NB_OBJECTS["users"] + 3 + ) # Monique Test, Jeanne Test and Jean Something (quick fix for e2e) assert models.Team.objects.count() == TEST_NB_OBJECTS["teams"] assert models.TeamAccess.objects.count() >= TEST_NB_OBJECTS["teams"] assert mailbox_models.MailDomain.objects.count() == TEST_NB_OBJECTS["domains"]