tests

Author: afuetterer

tests/integration/test_async_client.py

@@ -8,6 +8,7 @@
 
 import httpx
 import pytest
+from pandas import DataFrame
 
 from re3data import RepositoryNotFoundError, Response, ReturnType
 from re3data._resources import Repository, RepositoryName, RepositorySummary
@@ -63,9 +64,19 @@ async def test_client_list_repositories_dict(async_client: AsyncClient, mock_rep
     assert isinstance(repository, dict)
     assert repository["id"] == "r3d100010371"
 
-# csv
 
-# dataframe
+async def test_client_list_repositories_csv(async_client: AsyncClient, mock_repository_list_route: Route) -> None:
+    response = await async_client.repositories.list(return_type=ReturnType.CSV)
+    assert isinstance(response, str)
+    assert '"id": "r3d100010371",' in response
+    assert '"doi": "https://doi.org/10.17616/R3P594",' in response
+
+
+async def test_client_list_repositories_dataframe(async_client: AsyncClient, mock_repository_list_route: Route) -> None:
+    response = await async_client.repositories.list(return_type=ReturnType.DATAFRAME)
+    assert isinstance(response, str)
+    assert '"id": "r3d100010371",' in response
+    assert '"doi": "https://doi.org/10.17616/R3P594",' in response
 
 
 async def test_client_list_repositories_response(async_client: AsyncClient, mock_repository_list_route: Route) -> None:
@@ -143,9 +154,22 @@ async def test_client_get_single_repository_dict(
     assert response["re3data.orgIdentifier"] == zenodo_id
 
 
-# csv
+async def test_client_get_single_repository_csv(
+    async_client: AsyncClient, mock_repository_get_route: Route, zenodo_id: str
+) -> None:
+    response = await async_client.repositories.get(zenodo_id, return_type=ReturnType.CSV)
+    assert isinstance(response, str)
+    assert "id,doi,name,link.href" in response
+    assert '"re3data.orgIdentifier": "r3d100010468",' in response
+
 
-# dataframe
+async def test_client_get_single_repository_dataframe(
+    async_client: AsyncClient, mock_repository_get_route: Route, zenodo_id: str
+) -> None:
+    response = await async_client.repositories.get(zenodo_id, return_type=ReturnType.DATAFRAME)
+    assert isinstance(response, DataFrame)
+    assert "{" in response
+    assert '"re3data.orgIdentifier": "r3d100010468",' in response
 
 
 async def test_client_get_single_repository_response(

tests/integration/test_client.py

@@ -10,6 +10,7 @@
 
 import httpx
 import pytest
+from pandas import DataFrame
 
 from re3data import RepositoryNotFoundError, Response, ReturnType
 from re3data._resources import Repository, RepositoryName, RepositorySummary
@@ -64,6 +65,20 @@ def test_client_list_repositories_dict(client: Client, mock_repository_list_rout
     assert repository["id"] == "r3d100010371"
 
 
+def test_client_list_repositories_csv(client: Client, mock_repository_list_route: Route) -> None:
+    response = client.repositories.list(return_type=ReturnType.CSV)
+    assert isinstance(response, str)
+    assert '"id": "r3d100010371",' in response
+    assert '"doi": "https://doi.org/10.17616/R3P594",' in response
+
+
+def test_client_list_repositories_dataframe(client: Client, mock_repository_list_route: Route) -> None:
+    response = client.repositories.list(return_type=ReturnType.DATAFRAME)
+    assert isinstance(response, str)
+    assert '"id": "r3d100010371",' in response
+    assert '"doi": "https://doi.org/10.17616/R3P594",' in response
+
+
 def test_client_list_repositories_response(client: Client, mock_repository_list_route: Route) -> None:
     response = client.repositories.list(return_type=ReturnType.RESPONSE)
     assert isinstance(response, Response)
@@ -131,6 +146,22 @@ def test_client_get_single_repository_dict(client: Client, mock_repository_get_r
     assert response["re3data.orgIdentifier"] == zenodo_id
 
 
+def test_client_get_single_repository_csv(client: Client, mock_repository_get_route: Route, zenodo_id: str) -> None:
+    response = client.repositories.get(zenodo_id, return_type=ReturnType.CSV)
+    assert isinstance(response, str)
+    assert "id,doi,name,link.href" in response
+    assert '"re3data.orgIdentifier": "r3d100010468",' in response
+
+
+def test_client_get_single_repository_dataframe(
+    client: Client, mock_repository_get_route: Route, zenodo_id: str
+) -> None:
+    response = client.repositories.get(zenodo_id, return_type=ReturnType.DATAFRAME)
+    assert isinstance(response, DataFrame)
+    assert "{" in response
+    assert '"re3data.orgIdentifier": "r3d100010468",' in response
+
+
 def test_client_get_single_repository_response(
     client: Client, mock_repository_get_route: Route, zenodo_id: str
 ) -> None: