This commit is contained in:
Kenneth Kehl
2024-07-11 09:38:32 -07:00
parent 3a2ba3a8c0
commit 058a999ed0
22 changed files with 235 additions and 70 deletions

View File

@@ -1931,14 +1931,14 @@ def test_fetch_daily_stats(
)
with app.test_client() as client:
with client.session_transaction() as sess:
sess['service_id'] = service_id
sess["service_id"] = service_id
socketio_client = SocketIOTestClient(app, socketio, flask_test_client=client)
connected = socketio_client.is_connected()
assert connected, "Client should be connected"
socketio_client.emit('fetch_daily_stats')
socketio_client.emit("fetch_daily_stats")
received = socketio_client.get_received()
mock_service_api.assert_called_once_with(
@@ -1967,8 +1967,13 @@ def test_fetch_daily_stats(
SERVICE_ONE_ID,
USER_ONE_ID,
{"start_date": "2024-01-01", "days": 7},
{"service_id": SERVICE_ONE_ID, "user_id": USER_ONE_ID, "start_date": "2024-01-01", "days": 7},
{"id": USER_ONE_ID, "name": "Test User"}
{
"service_id": SERVICE_ONE_ID,
"user_id": USER_ONE_ID,
"start_date": "2024-01-01",
"days": 7,
},
{"id": USER_ONE_ID, "name": "Test User"},
),
],
)
@@ -2001,15 +2006,15 @@ def test_fetch_daily_stats_by_user(
with app.test_client() as client:
with client.session_transaction() as sess:
sess['service_id'] = service_id
sess['user_id'] = user_id
sess["service_id"] = service_id
sess["user_id"] = user_id
socketio_client = SocketIOTestClient(app, socketio, flask_test_client=client)
connected = socketio_client.is_connected()
assert connected, "Client should be connected"
socketio_client.emit('fetch_daily_stats_by_user')
socketio_client.emit("fetch_daily_stats_by_user")
received = socketio_client.get_received()
mock_service_api.assert_called_once_with(