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"]