Changeset View
Changeset View
Standalone View
Standalone View
swh/counters/tests/test_server.py
| Show First 20 Lines • Show All 164 Lines • ▼ Show 20 Lines | for collection in data.keys(): | ||||
| assert data[collection] == int(m.group(1)) | assert data[collection] == int(m.group(1)) | ||||
| def test_server_counters_history(history_test_client, mocker): | def test_server_counters_history(history_test_client, mocker): | ||||
| """Test the counters history file download""" | """Test the counters history file download""" | ||||
| expected_result = {"content": [[1, 1], [2, 2]]} | expected_result = {"content": [[1, 1], [2, 2]]} | ||||
| mock = mocker.patch("swh.counters.history.History.get_history") | mock = mocker.patch("swh.counters.history.History.get_history") | ||||
| mock.return_value = json.dumps(expected_result) | mock.return_value = expected_result | ||||
| r = history_test_client.get("/counters_history/test.json") | r = history_test_client.get("/counters_history/test.json") | ||||
| assert 200 == r.status_code | assert 200 == r.status_code | ||||
| response = r.get_data().decode("utf-8") | response = r.get_data().decode("utf-8") | ||||
| response_json = json.loads(response) | response_json = json.loads(response) | ||||
| assert response_json == expected_result | assert response_json == expected_result | ||||
| assert "application/json" == r.headers["Content-Type"] | |||||
| def test_server_counters_history_file_not_found(history_test_client, mocker): | def test_server_counters_history_file_not_found(history_test_client, mocker): | ||||
| """ensure a 404 is returned when the file doesn't exists""" | """ensure a 404 is returned when the file doesn't exists""" | ||||
| mock = mocker.patch("swh.counters.history.History.get_history") | mock = mocker.patch("swh.counters.history.History.get_history") | ||||
| mock.side_effect = FileNotFoundError | mock.side_effect = FileNotFoundError | ||||
| r = history_test_client.get("/counters_history/fake.json") | r = history_test_client.get("/counters_history/fake.json") | ||||
| assert 404 == r.status_code | assert 404 == r.status_code | ||||