Add prompt tests

WalletConnect · Feb 15, 2022 · 3f7899f · 3f7899f
1 parent 1e5bbd4
commit 3f7899f
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 3 deletions.
diff --git a/Tests/WalletConnectTests/Mocks/MockedNetworkRelayer.swift b/Tests/WalletConnectTests/Mocks/MockedNetworkRelayer.swift
@@ -6,7 +6,9 @@ class MockedNetworkRelayer: NetworkRelaying {
     var onConnect: (() -> ())?
     var onMessage: ((String, String) -> ())?
     var error: Error?
+    var prompt = false
     func publish(topic: String, payload: String, prompt: Bool, completion: @escaping ((Error?) -> ())) -> Int64 {
+        self.prompt = prompt
         completion(error)
         return 0
     }

diff --git a/Tests/WalletConnectTests/Mocks/SerializerMock.swift b/Tests/WalletConnectTests/Mocks/SerializerMock.swift
@@ -8,7 +8,7 @@ import WalletConnectUtils
 
 class SerializerMock: Serializing {
     var deserialized: Any!
-    var serialized: String!
+    var serialized: String = ""
 
     func serialize(topic: String, encodable: Encodable) throws -> String {
         try serialize(json: try encodable.json(), agreementKeys: AgreementSecret(sharedSecret: Data(), publicKey: AgreementPrivateKey().publicKey))

diff --git a/Tests/WalletConnectTests/WCRelayTests.swift b/Tests/WalletConnectTests/WCRelayTests.swift
@@ -71,8 +71,20 @@ class WalletConnectRelayTests: XCTestCase {
         waitForExpectations(timeout: 0.01, handler: nil)
     }
 
-    func testRequestCompletesWithError() {
-        //todo
+    func testPromptOnSessionPayload() {
+        let topic = "fefc3dc39cacbc562ed58f92b296e2d65a6b07ef08992b93db5b3cb86280635a"
+        let request = getWCSessionPayloadRequest()
+        networkRelayer.prompt = false
+        wcRelay.request(topic: topic, payload: request) { _ in }
+        XCTAssertTrue(networkRelayer.prompt)
+    }
+
+    func testNoPromptOnSessionUpgrade() {
+        let topic = "fefc3dc39cacbc562ed58f92b296e2d65a6b07ef08992b93db5b3cb86280635a"
+        let request = getWCSessionUpgrade()
+        networkRelayer.prompt = false
+        wcRelay.request(topic: topic, payload: request) { _ in }
+        XCTAssertTrue(networkRelayer.prompt)
     }
 }
 
@@ -89,6 +101,14 @@ extension WalletConnectRelayTests {
         let wcRequest = WCRequest(id: wcRequestId, method: WCRequest.Method.sessionPayload, params: params)
         return wcRequest
     }
+
+    func getWCSessionUpgrade() -> WCRequest {
+        let wcRequestId: Int64 = 123456
+        let sessionUpgradeParams = SessionType.UpgradeParams(permissions: SessionPermissions(blockchain: SessionPermissions.Blockchain(chains: []), jsonrpc: SessionPermissions.JSONRPC(methods: []), notifications: SessionPermissions.Notifications(types: [])))
+        let params = WCRequest.Params.sessionUpgrade(sessionUpgradeParams)
+        let wcRequest = WCRequest(id: wcRequestId, method: WCRequest.Method.sessionPayload, params: params)
+        return wcRequest
+    }
 }
 
 fileprivate let testPayload =