feat: additional in_memory wallet test

Signed-off-by: Micah Peltier <micah6_8@yahoo.com>
Indicio-tech · Jun 29, 2024 · eed6981 · eed6981
1 parent f917714
commit eed6981
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/aries_cloudagent/wallet/tests/test_in_memory_wallet.py b/aries_cloudagent/wallet/tests/test_in_memory_wallet.py
@@ -549,3 +549,13 @@ async def test_set_did_endpoint_x_not_sov(self, wallet: InMemoryWallet):
         assert "Setting DID endpoint is only allowed for did:sov DIDs" in str(
             context.value
         )
+
+    @pytest.mark.asyncio
+    async def test_assign_and_get_by_kid(self, wallet: InMemoryWallet):
+
+        key = await wallet.create_key(ED25519)
+
+        await wallet.assign_kid_to_key(key.verkey, "test_kid")
+
+        return_key = await wallet.get_key_by_kid("test_kid")
+        assert return_key.verkey == key.verkey