feat: implement multiple services

gandalf-network · Jun 22, 2024 · 6aa9b2c · 6aa9b2c
1 parent 8c77b47
commit 6aa9b2c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/Sources/Example/Example.swift b/Sources/Example/Example.swift
@@ -11,6 +11,7 @@ struct Example {
 func testConnect() async {
     let services: InputData = [
         "uber": .service(Service(traits: ["rating"], activities: ["trip"])),
+        "netflix": .service(Service(activities: ["watch"])),
     ]
     let input = ConnectInput(
         publicKey: "0x02073d3b9daf439c19a267dcfc19bc1ac1aea5066d8c754554b046476099b6fa22",

diff --git a/Sources/GandalfConnect/GandalfConnect.swift b/Sources/GandalfConnect/GandalfConnect.swift
@@ -153,9 +153,9 @@ public class Connect {
 
         let keys = input.keys.map { $0.lowercased() }
 
-        if keys.count > 2 || (keys.count == 2 && !keys.contains("gandalf")) {
+        if keys.count < 1 {
             throw GandalfError(
-                message: "Only one non-Gandalf service is supported per Connect URL",
+                message: "At least one service is needed.",
                 code: .InvalidService
             )
         }

diff --git a/Tests/GandalfConnectTests/GandalfConnectTests.swift b/Tests/GandalfConnectTests/GandalfConnectTests.swift
@@ -89,8 +89,9 @@ final class ConnectTests: XCTestCase {
         let connect = Connect(input: input)
 
         do {
-            _ = try await connect.generateURL()
-            XCTFail("Expected to throw, but did not throw")
+            let generatedURL = try await connect.generateURL()
+            XCTAssertTrue(generatedURL.contains(publicKey))
+            XCTAssertTrue(generatedURL.contains(redirectURL))
         } catch let error as GandalfError {
             XCTAssertEqual(error.code, .InvalidService)
         } catch {