Raise ResultGetError for missing value.

Author: Raphael Krupinski

taskiq_nats/exceptions.py

@@ -5,14 +5,12 @@
 from nats.js import JetStreamContext
 from nats.js.errors import BucketNotFoundError, ObjectNotFoundError
 from nats.js.object_store import ObjectStore
-from taskiq import AsyncResultBackend
+from taskiq import AsyncResultBackend, ResultGetError
 from taskiq.abc.serializer import TaskiqSerializer
 from taskiq.compat import model_dump, model_validate
 from taskiq.result import TaskiqResult
 from taskiq.serializers import PickleSerializer
 
-from taskiq_nats.exceptions import ResultIsMissingError
-
 _ReturnType = TypeVar("_ReturnType")
 
 
@@ -109,7 +107,7 @@ async def get_result(
                 name=task_id,
             )
         except ObjectNotFoundError as exc:
-            raise ResultIsMissingError from exc
+            raise ResultGetError from exc
 
         if not self.keep_results:
             await self.object_store.delete(

taskiq_nats/result_backend.py

@@ -2,10 +2,9 @@
 from typing import Any, List, TypeVar
 
 import pytest
-from taskiq import TaskiqResult
+from taskiq import ResultGetError, TaskiqResult
 
 from taskiq_nats import NATSObjectStoreResultBackend
-from taskiq_nats.exceptions import ResultIsMissingError
 
 _ReturnType = TypeVar("_ReturnType")
 pytestmark = pytest.mark.anyio
@@ -64,7 +63,7 @@ async def test_failure_backend_result(
     task_id: str,
 ) -> None:
     """Test exception raising in `get_result` method."""
-    with pytest.raises(expected_exception=ResultIsMissingError):
+    with pytest.raises(expected_exception=ResultGetError):
         await nats_result_backend.get_result(task_id=task_id)
 
 
@@ -85,7 +84,7 @@ async def test_success_backend_default_result_delete_res(
     )
     await backend.get_result(task_id=task_id)
 
-    with pytest.raises(expected_exception=ResultIsMissingError):
+    with pytest.raises(expected_exception=ResultGetError):
         await backend.get_result(task_id=task_id)
 
     await backend.shutdown()