🧪 Update tests

Signed-off-by: ff137 <ff137@proton.me>

Author: ff137

acapy_agent/revocation_anoncreds/models/tests/test_issuer_cred_rev_record.py

@@ -1,6 +1,5 @@
 from unittest import IsolatedAsyncioTestCase
 
-from ....storage.base import StorageNotFoundError
 from ....utils.testing import create_test_profile
 from .. import issuer_cred_rev_record as test_module
 from ..issuer_cred_rev_record import IssuerCredRevRecord
@@ -84,13 +83,30 @@ async def test_rec_ops(self):
                 )
             )
 
+            await recs[1].set_state(  # Save extra record
+                session,
+                IssuerCredRevRecord.STATE_REVOKED,
+            )
+            # Fetch cred rev id as string
+            assert await IssuerCredRevRecord.retrieve_by_ids(
+                session, rev_reg_id=REV_REG_ID, cred_rev_id="1"
+            ) == [recs[0]]
+
+            # Fetch cred rev id as list
+            assert await IssuerCredRevRecord.retrieve_by_ids(
+                session, rev_reg_id=REV_REG_ID, cred_rev_id=["2"]
+            ) == [recs[1]]
+
+            # Fetch both
             assert (
                 await IssuerCredRevRecord.retrieve_by_ids(
-                    session, rev_reg_id=REV_REG_ID, cred_rev_id="1"
+                    session, rev_reg_id=REV_REG_ID, cred_rev_id=["1", "2"]
                 )
-                == recs[0]
-            )
-            with self.assertRaises(StorageNotFoundError):
+            ) == recs
+
+            # Fetch cred rev id that doesn't exist
+            assert (
                 await IssuerCredRevRecord.retrieve_by_ids(
-                    session, rev_reg_id=REV_REG_ID, cred_rev_id="2"
+                    session, rev_reg_id=REV_REG_ID, cred_rev_id=["3"]
                 )
+            ) == []

acapy_agent/revocation_anoncreds/tests/test_routes.py

@@ -348,9 +348,9 @@ async def test_get_cred_rev_record(self):
                 test_module.web, "json_response", mock.Mock()
             ) as mock_json_response,
         ):
-            mock_retrieve.return_value = mock.MagicMock(
-                serialize=mock.MagicMock(return_value="dummy")
-            )
+            mock_retrieve.return_value = [
+                mock.MagicMock(serialize=mock.MagicMock(return_value="dummy"))
+            ]
             result = await test_module.get_cred_rev_record(self.request)
 
             mock_json_response.assert_called_once_with({"result": "dummy"})