add universal resolver register test; fix did resource test error

decentralized-identity · May 2, 2024 · ca34b29 · ca34b29
1 parent 8e84ef7
commit ca34b29
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/Tests/Web5Tests/Dids/DIDResourceTests.swift b/Tests/Web5Tests/Dids/DIDResourceTests.swift
@@ -22,7 +22,7 @@ final class DIDResourceTests: XCTestCase {
         let did = try! DIDJWK.create()
         let didDocument = await DIDUniversalResolver.DIDResolver().resolve(didURI: did.uri).didDocument
         let resource = DIDResource.didDocument(didDocument!)
-        let encoded = try! JSONEncoder().encode(didDocument)
+        let encoded = try! JSONEncoder().encode(resource)
         let decoded = try! JSONDecoder().decode(DIDResource.self, from: encoded)
         XCTAssertEqual(decoded, resource)
     }

diff --git a/Tests/Web5Tests/Dids/DIDUniversalResolverTests.swift b/Tests/Web5Tests/Dids/DIDUniversalResolverTests.swift
@@ -90,4 +90,19 @@ final class DIDUniversalResolverTests: XCTestCase {
         XCTAssertNotNil(result.dereferencingMetadata.error)
         XCTAssertEqual(result.dereferencingMetadata.error, DID.Error.notFound.localizedDescription)
     }
+
+    func testRegisterResolver() async {
+
+        struct MockResolver: DIDMethodResolver {
+            public let methodName = "mock"
+            public func resolve(didURI: String) async -> DIDResolutionResult {
+                let metaData = DIDResolutionResult.Metadata(contentType: "this is mock resolver")
+                return DIDResolutionResult(didResolutionMetadata: metaData)
+            }
+        }
+
+        DIDUniversalResolver.register(resolver: MockResolver())
+        let result = await universalResolver.resolve(didURI: "did:mock:abc123")
+        XCTAssertEqual(result.didResolutionMetadata.contentType, "this is mock resolver")
+    }
 }