diff options
Diffstat (limited to 'ctrack/organisations/management')
| -rw-r--r-- | ctrack/organisations/management/commands/populate_db.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/ctrack/organisations/management/commands/populate_db.py b/ctrack/organisations/management/commands/populate_db.py index 4477b63..1674ec0 100644 --- a/ctrack/organisations/management/commands/populate_db.py +++ b/ctrack/organisations/management/commands/populate_db.py @@ -11,6 +11,7 @@ from ctrack.organisations.tests.factories import OrganisationFactory from ctrack.organisations.tests.factories import PersonFactory from ctrack.organisations.tests.factories import RoleFactory from ctrack.organisations.tests.factories import UserFactory +from ctrack.register.tests.factories import EngagementEventFactory class Command(BaseCommand): @@ -71,3 +72,30 @@ class Command(BaseCommand): f"Created {number} Person object[s]! Go forth and multiply." ) ) + + # set up some EngagementEvents + + p1 = PersonFactory.create( + role=role, + updated_by=user, + predecessor=None, + organisation__submode=submodes[randint(0, len(submodes) - 1)], + organisation=org, + ) + p2 = PersonFactory.create( + role=role, + updated_by=user, + predecessor=None, + organisation__submode=submodes[randint(0, len(submodes) - 1)], + organisation=org, + ) + p3 = PersonFactory.create( + role=role, + updated_by=user, + predecessor=None, + organisation__submode=submodes[randint(0, len(submodes) - 1)], + organisation=org, + ) + + ee1 = EngagementEventFactory.create(user=user, participants=[p1, p2]) + ee2 = EngagementEventFactory.create(user=user, participants=[p3]) |