Fix unit tests

Signed-off-by: jamshale <[email protected]>
openwallet-foundation · Nov 6, 2024 · 35542cd · 35542cd
1 parent aadae16
commit 35542cd
Show file tree

Hide file tree

Showing 9 changed files with 374 additions and 280 deletions.
diff --git a/acapy_agent/anoncreds/default/legacy_indy/tests/test_registry.py b/acapy_agent/anoncreds/default/legacy_indy/tests/test_registry.py
@@ -13,9 +13,6 @@
 )
 from base58 import alphabet
 
-from ......scenarios.examples.anoncreds_issuance_and_revocation.example import (
-    SchemaResult,
-)
 from .....anoncreds.base import AnonCredsSchemaAlreadyExists
 from .....anoncreds.default.legacy_indy import registry as test_module
 from .....anoncreds.issuer import AnonCredsIssuer
@@ -53,7 +50,7 @@
     RevRegDefState,
     RevRegDefValue,
 )
-from ....models.schema import AnonCredsSchema, GetSchemaResult
+from ....models.schema import AnonCredsSchema, GetSchemaResult, SchemaResult
 
 B58 = alphabet if isinstance(alphabet, str) else alphabet.decode("ascii")
 INDY_DID = rf"^(did:sov:)?[{B58}]{{21,22}}$"

diff --git a/acapy_agent/indy/models/cred_abstract.py b/acapy_agent/indy/models/cred_abstract.py
@@ -38,7 +38,7 @@ def __init__(
         self.xr_cap = xr_cap
 
 
-class AnoncredsCorrectnessProofSchema(BaseModelSchema):
+class IndyKeyCorrectnessProofSchema(BaseModelSchema):
     """Indy key correctness proof schema."""
 
     class Meta:
@@ -149,7 +149,7 @@ class Meta:
         },
     )
     key_correctness_proof = fields.Nested(
-        AnoncredsCorrectnessProofSchema(),
+        IndyKeyCorrectnessProofSchema(),
         required=True,
         metadata={"description": "Key correctness proof"},
     )