feat: update tests

NiceAesth · Nov 16, 2024 · 6198dde · 6198dde
1 parent 02c5db9
commit 6198dde
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 2 deletions.
diff --git a/aiosu/v2/client.py b/aiosu/v2/client.py
@@ -1856,7 +1856,7 @@ async def get_score_replay_lazer(
         :return: Replay file
         :rtype: io.BytesIO
         """
-        url = f"{self.base_url}/api/v2/scores/{mode}/{score_id}/download"
+        url = f"{self.base_url}/api/v2/scores/{score_id}/download"
         return await self._request("GET", url)
 
     @prepare_token

diff --git a/tests/data/v2/get_score_replay_lazer_200.osr b/tests/data/v2/get_score_replay_lazer_200.osr
diff --git a/tests/data/v2/get_score_replay_lazer_404.json b/tests/data/v2/get_score_replay_lazer_404.json
@@ -0,0 +1 @@
+{"error":"Specified Score\\Best\\Osu couldn't be found."}
diff --git a/tests/generate/main.py b/tests/generate/main.py
@@ -551,6 +551,17 @@ def _register_routes(self) -> None:
             f"{DATA_DIR}/v2/get_score_replay_404.json",
             expect_status=404,
         )
+        self._register_route(
+            "GET",
+            f"{BASE_URL}/api/v2/scores/1581778626/download",
+            f"{DATA_DIR}/v2/get_score_replay_lazer_200.osr",
+        )
+        self._register_route(
+            "GET",
+            f"{BASE_URL}/api/v2/scores/0/download",
+            f"{DATA_DIR}/v2/get_score_replay_lazer_404.json",
+            expect_status=404,
+        )
         self._register_route(
             "GET",
             f"{BASE_URL}/api/v2/rankings/osu/performance",

diff --git a/tests/test_v2/test_client.py b/tests/test_v2/test_client.py
@@ -244,7 +244,12 @@ async def test_generated(status_code, content_type, token, mocker):
     generate_test(
         aiosu.v2.Client.get_score_replay,
         STATUS_CAN_404_OCTET,
-        func_kwargs={"score_id": 4220635589, "mode": "osu"},
+        func_kwargs={"legacy_score_id": 4220635589, "mode": "osu"},
+    ),
+    generate_test(
+        aiosu.v2.Client.get_score_replay_lazer,
+        STATUS_CAN_404_OCTET,
+        func_kwargs={"score_id": 1581778626},
     ),
     generate_test(
         aiosu.v2.Client.get_rankings,