Requests.status_count can be scoped to creator

2018-08-15 11:43:12 -04:00
parent 3a77e07c3f
commit 920ad1c554
2 changed files with 24 additions and 7 deletions
--- a/tests/domain/test_requests.py
+++ b/tests/domain/test_requests.py
@@ -77,3 +77,14 @@ def test_status_count(session):
    assert Requests.status_count(RequestStatus.PENDING_FINANCIAL_VERIFICATION) == 1
    assert Requests.status_count(RequestStatus.STARTED) == 1
    assert Requests.in_progress_count() == 2
+
+def test_status_count_scoped_to_creator(session):
+    # make sure table is empty
+    session.query(Request).delete()
+
+    user = UserFactory.create()
+    request1 = RequestFactory.create()
+    request2 = RequestFactory.create(creator=user)
+
+    assert Requests.status_count(RequestStatus.STARTED) == 2
+    assert Requests.status_count(RequestStatus.STARTED, creator=user) == 1