darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-04-02 00:20:02 -04:00
Code Issues Packages Projects Releases Wiki Activity

Added mock to client request for get request of count

Browse Source
This commit is contained in:
Andrew Shumway
2023-10-27 10:00:45 -06:00
committed by Carlo Costino
parent 83767b8ba8
commit 62a09c5223
2 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/__init__.py
View File

@@ -290,8 +290,12 @@ def init_app(application):
if service_id:
global_limit = current_app.config["GLOBAL_SERVICE_MESSAGE_LIMIT"]
global_messages_count = service_api_client.get_global_notification_count(service_id)
remaining_global_messages = global_limit - global_messages_count.get("count")
global_messages_count = (
service_api_client.get_global_notification_count(service_id)
)
remaining_global_messages = global_limit - global_messages_count.get(
"count"
)
return {"daily_global_messages_remaining": remaining_global_messages}
@application.before_request

11
tests/conftest.py
View File

@@ -2423,8 +2423,15 @@ def _os_environ():
os.environ[k] = v
@pytest.fixture()
def client_request(logged_in_client, mocker, service_one): # noqa (C901 too complex)
@pytest.fixture() # noqa (C901 too complex)
def client_request(_logged_in_client, mocker, service_one): # noqa (C901 too complex)
def _get(mocker):
return {"count": 0}
mocker.patch(
"app.service_api_client.get_global_notification_count", side_effect=_get
)
class ClientRequest:
@staticmethod
@contextmanager