diff --git a/tests/factories.py b/tests/factories.py index f6ed669f..c49b585d 100644 --- a/tests/factories.py +++ b/tests/factories.py @@ -37,6 +37,7 @@ class UserFactory(factory.alchemy.SQLAlchemyModelFactory): role = Roles.get(atat_role_name) return cls.create(atat_role=role, **kwargs) + class RequestStatusEventFactory(factory.alchemy.SQLAlchemyModelFactory): class Meta: model = RequestStatusEvent diff --git a/tests/routes/test_requests_index.py b/tests/routes/test_requests_index.py index 03d214a4..e33c7cb8 100644 --- a/tests/routes/test_requests_index.py +++ b/tests/routes/test_requests_index.py @@ -9,7 +9,7 @@ def test_action_required_mission_owner(): Requests.submit(requests[0]) context = RequestsIndex(creator).execute() - assert context['requests'][0]['action_required'] == False + assert context["requests"][0]["action_required"] == False def test_action_required_ccpo(): @@ -20,4 +20,4 @@ def test_action_required_ccpo(): ccpo = UserFactory.from_atat_role("ccpo") context = RequestsIndex(ccpo).execute() - assert context['num_action_required'] == 1 + assert context["num_action_required"] == 1