from falcon import testing import json import pytest from unittest.mock import patch from dspace_statistics_api.app import api @pytest.fixture def client(): return testing.TestClient(api) def test_get_collection(client): """Test requesting a single collection.""" response = client.simulate_get("/collection/8ea4b611-1f59-4d4e-b78d-a9921a72cfe7") response_doc = json.loads(response.text) assert isinstance(response_doc["downloads"], int) assert isinstance(response_doc["id"], str) assert isinstance(response_doc["views"], int) assert response.status_code == 200 def test_get_missing_collection(client): """Test requesting a single non-existing collection.""" response = client.simulate_get("/collection/508abe0a-689f-402e-885d-2f6b02e7a39c") assert response.status_code == 404 def test_get_collections(client): """Test requesting 100 collections.""" response = client.simulate_get("/collections", query_string="limit=100") response_doc = json.loads(response.text) assert isinstance(response_doc["currentPage"], int) assert isinstance(response_doc["totalPages"], int) assert isinstance(response_doc["statistics"], list) assert response.status_code == 200 def test_get_collections_invalid_limit(client): """Test requesting 100 collections with an invalid limit parameter.""" response = client.simulate_get("/collections", query_string="limit=101") assert response.status_code == 400 def test_get_collections_invalid_page(client): """Test requesting 100 collections with an invalid page parameter.""" response = client.simulate_get("/collections", query_string="page=-1") assert response.status_code == 400 @pytest.mark.xfail def test_post_collections_valid_dateFrom(client): """Test POSTing a request to /collections with a valid dateFrom parameter in the request body.""" request_body = { "dateFrom": "2020-01-01T00:00:00Z", "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } response = client.simulate_post("/collections", json=request_body) assert response.status_code == 200 assert response.json["limit"] == 100 assert response.json["currentPage"] == 0 assert isinstance(response.json["totalPages"], int) assert len(response.json["statistics"]) == 2 assert isinstance(response.json["statistics"][0]["views"], int) assert isinstance(response.json["statistics"][0]["downloads"], int) assert isinstance(response.json["statistics"][1]["views"], int) assert isinstance(response.json["statistics"][1]["downloads"], int) def test_post_collections_valid_dateFrom_mocked(client): """Mock test POSTing a request to /collections with a valid dateFrom parameter in the request body.""" request_body = { "dateFrom": "2020-01-01T00:00:00Z", "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } get_views_return_value = { "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7": 21, "260548c8-fda4-4dc8-a979-03495753cdd5": 0, } get_downloads_return_value = { "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7": 575, "260548c8-fda4-4dc8-a979-03495753cdd5": 899, } with patch( "dspace_statistics_api.app.get_views", return_value=get_views_return_value ): with patch( "dspace_statistics_api.app.get_downloads", return_value=get_downloads_return_value, ): response = client.simulate_post("/collections", json=request_body) assert response.status_code == 200 assert response.json["limit"] == 100 assert response.json["currentPage"] == 0 assert isinstance(response.json["totalPages"], int) assert len(response.json["statistics"]) == 2 assert isinstance(response.json["statistics"][0]["views"], int) assert isinstance(response.json["statistics"][0]["downloads"], int) assert isinstance(response.json["statistics"][1]["views"], int) assert isinstance(response.json["statistics"][1]["downloads"], int) def test_post_collections_invalid_dateFrom(client): """Test POSTing a request to /collections with an invalid dateFrom parameter in the request body.""" request_body = { "dateFrom": "2020-01-01T00:00:00", "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } response = client.simulate_post("/collections", json=request_body) assert response.status_code == 400 @pytest.mark.xfail def test_post_collections_valid_dateTo(client): """Test POSTing a request to /collections with a valid dateTo parameter in the request body.""" request_body = { "dateTo": "2020-01-01T00:00:00Z", "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } response = client.simulate_post("/collections", json=request_body) assert response.status_code == 200 assert response.json["limit"] == 100 assert response.json["currentPage"] == 0 assert isinstance(response.json["totalPages"], int) assert len(response.json["statistics"]) == 2 assert isinstance(response.json["statistics"][0]["views"], int) assert isinstance(response.json["statistics"][0]["downloads"], int) assert isinstance(response.json["statistics"][1]["views"], int) assert isinstance(response.json["statistics"][1]["downloads"], int) def test_post_collections_valid_dateTo_mocked(client): """Mock test POSTing a request to /collections with a valid dateTo parameter in the request body.""" request_body = { "dateTo": "2020-01-01T00:00:00Z", "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } get_views_return_value = { "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7": 21, "260548c8-fda4-4dc8-a979-03495753cdd5": 0, } get_downloads_return_value = { "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7": 575, "260548c8-fda4-4dc8-a979-03495753cdd5": 899, } with patch( "dspace_statistics_api.app.get_views", return_value=get_views_return_value ): with patch( "dspace_statistics_api.app.get_downloads", return_value=get_downloads_return_value, ): response = client.simulate_post("/collections", json=request_body) assert response.status_code == 200 assert response.json["limit"] == 100 assert response.json["currentPage"] == 0 assert isinstance(response.json["totalPages"], int) assert len(response.json["statistics"]) == 2 assert isinstance(response.json["statistics"][0]["views"], int) assert isinstance(response.json["statistics"][0]["downloads"], int) assert isinstance(response.json["statistics"][1]["views"], int) assert isinstance(response.json["statistics"][1]["downloads"], int) def test_post_collections_invalid_dateTo(client): """Test POSTing a request to /collections with an invalid dateTo parameter in the request body.""" request_body = { "dateFrom": "2020-01-01T00:00:00", "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } response = client.simulate_post("/collections", json=request_body) assert response.status_code == 400 @pytest.mark.xfail def test_post_collections_valid_limit(client): """Test POSTing a request to /collections with a valid limit parameter in the request body.""" request_body = { "limit": 1, "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } response = client.simulate_post("/collections", json=request_body) assert response.status_code == 200 assert response.json["limit"] == 1 assert response.json["currentPage"] == 0 assert isinstance(response.json["totalPages"], int) assert len(response.json["statistics"]) == 1 assert isinstance(response.json["statistics"][0]["views"], int) assert isinstance(response.json["statistics"][0]["downloads"], int) def test_post_collections_valid_limit_mocked(client): """Mock test POSTing a request to /collections with a valid limit parameter in the request body.""" request_body = { "limit": 1, "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } get_views_return_value = {"8ea4b611-1f59-4d4e-b78d-a9921a72cfe7": 21} get_downloads_return_value = {"8ea4b611-1f59-4d4e-b78d-a9921a72cfe7": 575} with patch( "dspace_statistics_api.app.get_views", return_value=get_views_return_value ): with patch( "dspace_statistics_api.app.get_downloads", return_value=get_downloads_return_value, ): response = client.simulate_post("/collections", json=request_body) assert response.status_code == 200 assert response.json["limit"] == 1 assert response.json["currentPage"] == 0 assert isinstance(response.json["totalPages"], int) assert len(response.json["statistics"]) == 1 assert isinstance(response.json["statistics"][0]["views"], int) assert isinstance(response.json["statistics"][0]["downloads"], int) def test_post_collections_invalid_limit(client): """Test POSTing a request to /collections with an invalid limit parameter in the request body.""" request_body = { "limit": -1, "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } response = client.simulate_post("/collections", json=request_body) assert response.status_code == 400 @pytest.mark.xfail def test_post_collections_valid_page(client): """Test POSTing a request to /collections with a valid page parameter in the request body.""" request_body = { "page": 0, "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } response = client.simulate_post("/collections", json=request_body) assert response.status_code == 200 assert response.json["limit"] == 100 assert response.json["currentPage"] == 0 assert response.json["totalPages"] == 0 assert len(response.json["statistics"]) == 2 assert isinstance(response.json["statistics"][0]["views"], int) assert isinstance(response.json["statistics"][0]["downloads"], int) assert isinstance(response.json["statistics"][1]["views"], int) assert isinstance(response.json["statistics"][1]["downloads"], int) def test_post_collections_valid_page_mocked(client): """Mock test POSTing a request to /collections with a valid page parameter in the request body.""" request_body = { "page": 0, "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } get_views_return_value = { "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7": 21, "260548c8-fda4-4dc8-a979-03495753cdd5": 0, } get_downloads_return_value = { "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7": 575, "260548c8-fda4-4dc8-a979-03495753cdd5": 899, } with patch( "dspace_statistics_api.app.get_views", return_value=get_views_return_value ): with patch( "dspace_statistics_api.app.get_downloads", return_value=get_downloads_return_value, ): response = client.simulate_post("/collections", json=request_body) assert response.status_code == 200 assert response.json["limit"] == 100 assert response.json["currentPage"] == 0 assert isinstance(response.json["totalPages"], int) assert len(response.json["statistics"]) == 2 assert isinstance(response.json["statistics"][0]["views"], int) assert isinstance(response.json["statistics"][0]["downloads"], int) assert isinstance(response.json["statistics"][1]["views"], int) assert isinstance(response.json["statistics"][1]["downloads"], int) def test_post_collections_invalid_page(client): """Test POSTing a request to /collections with an invalid page parameter in the request body.""" request_body = { "page": -1, "collections": [ "8ea4b611-1f59-4d4e-b78d-a9921a72cfe7", "260548c8-fda4-4dc8-a979-03495753cdd5", ], } response = client.simulate_post("/collections", json=request_body) assert response.status_code == 400 # vim: set sw=4 ts=4 expandtab: