|
| 1 | +@testable import Coder_Desktop |
| 2 | +import CoderSDK |
| 3 | +import os |
| 4 | +import Testing |
| 5 | + |
| 6 | +@MainActor |
| 7 | +@Suite |
| 8 | +struct WorkspaceAppTests { |
| 9 | + let logger = Logger(subsystem: "com.coder.Coder-Desktop-Tests", category: "WorkspaceAppTests") |
| 10 | + let baseAccessURL = URL(string: "https://coder.example.com")! |
| 11 | + let sessionToken = "test-session-token" |
| 12 | + let host = "test-workspace.coder.test" |
| 13 | + |
| 14 | + @Test |
| 15 | + func testCreateWorkspaceApp_Success() throws { |
| 16 | + let sdkApp = CoderSDK.WorkspaceApp( |
| 17 | + id: UUID(), |
| 18 | + url: URL(string: "https://localhost:3000/app")!, |
| 19 | + external: false, |
| 20 | + slug: "test-app", |
| 21 | + display_name: "Test App", |
| 22 | + command: nil, |
| 23 | + icon: URL(string: "/icon/test-app.svg")!, |
| 24 | + subdomain: false, |
| 25 | + subdomain_name: nil |
| 26 | + ) |
| 27 | + |
| 28 | + let workspaceApp = try WorkspaceApp( |
| 29 | + sdkApp, |
| 30 | + iconBaseURL: baseAccessURL, |
| 31 | + sessionToken: sessionToken, |
| 32 | + newAppHost: host |
| 33 | + ) |
| 34 | + |
| 35 | + #expect(workspaceApp.slug == "test-app") |
| 36 | + #expect(workspaceApp.displayName == "Test App") |
| 37 | + #expect(workspaceApp.url.absoluteString == "https://test-workspace.coder.test:3000/app") |
| 38 | + #expect(workspaceApp.icon?.absoluteString == "https://coder.example.com/icon/test-app.svg") |
| 39 | + } |
| 40 | + |
| 41 | + @Test |
| 42 | + func testCreateWorkspaceApp_SessionTokenReplacement() throws { |
| 43 | + let sdkApp = CoderSDK.WorkspaceApp( |
| 44 | + id: UUID(), |
| 45 | + url: URL(string: "https://localhost:3000/app?token=$SESSION_TOKEN")!, |
| 46 | + external: false, |
| 47 | + slug: "token-app", |
| 48 | + display_name: "Token App", |
| 49 | + command: nil, |
| 50 | + icon: URL(string: "/icon/test-app.svg")!, |
| 51 | + subdomain: false, |
| 52 | + subdomain_name: nil |
| 53 | + ) |
| 54 | + |
| 55 | + let workspaceApp = try WorkspaceApp( |
| 56 | + sdkApp, |
| 57 | + iconBaseURL: baseAccessURL, |
| 58 | + sessionToken: sessionToken, |
| 59 | + newAppHost: host |
| 60 | + ) |
| 61 | + |
| 62 | + #expect( |
| 63 | + workspaceApp.url.absoluteString == "https://test-workspace.coder.test:3000/app?token=test-session-token" |
| 64 | + ) |
| 65 | + } |
| 66 | + |
| 67 | + @Test |
| 68 | + func testCreateWorkspaceApp_MissingURL() throws { |
| 69 | + let sdkApp = CoderSDK.WorkspaceApp( |
| 70 | + id: UUID(), |
| 71 | + url: nil, |
| 72 | + external: false, |
| 73 | + slug: "no-url-app", |
| 74 | + display_name: "No URL App", |
| 75 | + command: nil, |
| 76 | + icon: nil, |
| 77 | + subdomain: false, |
| 78 | + subdomain_name: nil |
| 79 | + ) |
| 80 | + |
| 81 | + #expect(throws: WorkspaceAppError.missingURL) { |
| 82 | + try WorkspaceApp( |
| 83 | + sdkApp, |
| 84 | + iconBaseURL: baseAccessURL, |
| 85 | + sessionToken: sessionToken, |
| 86 | + newAppHost: host |
| 87 | + ) |
| 88 | + } |
| 89 | + } |
| 90 | + |
| 91 | + @Test |
| 92 | + func testCreateWorkspaceApp_CommandApp() throws { |
| 93 | + let sdkApp = CoderSDK.WorkspaceApp( |
| 94 | + id: UUID(), |
| 95 | + url: URL(string: "https://localhost:3000/app")!, |
| 96 | + external: false, |
| 97 | + slug: "command-app", |
| 98 | + display_name: "Command App", |
| 99 | + command: "echo 'hello'", |
| 100 | + icon: nil, |
| 101 | + subdomain: false, |
| 102 | + subdomain_name: nil |
| 103 | + ) |
| 104 | + |
| 105 | + #expect(throws: WorkspaceAppError.isCommandApp) { |
| 106 | + try WorkspaceApp( |
| 107 | + sdkApp, |
| 108 | + iconBaseURL: baseAccessURL, |
| 109 | + sessionToken: sessionToken, |
| 110 | + newAppHost: host |
| 111 | + ) |
| 112 | + } |
| 113 | + } |
| 114 | + |
| 115 | + @Test |
| 116 | + func testDisplayApps_VSCode() throws { |
| 117 | + let agent = createMockAgent(displayApps: [.vscode, .web_terminal, .ssh_helper, .port_forwarding_helper]) |
| 118 | + |
| 119 | + let apps = agentToApps(logger, agent, host, baseAccessURL, sessionToken) |
| 120 | + |
| 121 | + #expect(apps.count == 1) |
| 122 | + #expect(apps[0].slug == "-vscode") |
| 123 | + #expect(apps[0].displayName == "VS Code") |
| 124 | + #expect(apps[0].url.absoluteString == "vscode://vscode-remote/ssh-remote+test-workspace.coder.test/home/user") |
| 125 | + #expect(apps[0].icon?.absoluteString == "https://coder.example.com/icon/code.svg") |
| 126 | + } |
| 127 | + |
| 128 | + @Test |
| 129 | + func testDisplayApps_VSCodeInsiders() throws { |
| 130 | + let agent = createMockAgent( |
| 131 | + displayApps: [ |
| 132 | + .vscode_insiders, |
| 133 | + .web_terminal, |
| 134 | + .ssh_helper, |
| 135 | + .port_forwarding_helper, |
| 136 | + ] |
| 137 | + ) |
| 138 | + |
| 139 | + let apps = agentToApps(logger, agent, host, baseAccessURL, sessionToken) |
| 140 | + |
| 141 | + #expect(apps.count == 1) |
| 142 | + #expect(apps[0].slug == "-vscode-insiders") |
| 143 | + #expect(apps[0].displayName == "VS Code Insiders") |
| 144 | + #expect(apps[0].icon?.absoluteString == "https://coder.example.com/icon/code.svg") |
| 145 | + #expect( |
| 146 | + apps[0].url.absoluteString == """ |
| 147 | + vscode-insiders://vscode-remote/ssh-remote+test-workspace.coder.test/home/user |
| 148 | + """ |
| 149 | + ) |
| 150 | + } |
| 151 | + |
| 152 | + @Test |
| 153 | + func testAgentToApps_MultipleApps() throws { |
| 154 | + let sdkApp1 = CoderSDK.WorkspaceApp( |
| 155 | + id: UUID(), |
| 156 | + url: URL(string: "https://localhost:3000/app1")!, |
| 157 | + external: false, |
| 158 | + slug: "app1", |
| 159 | + display_name: "App 1", |
| 160 | + command: nil, |
| 161 | + icon: URL(string: "/icon/app1.svg")!, |
| 162 | + subdomain: false, |
| 163 | + subdomain_name: nil |
| 164 | + ) |
| 165 | + |
| 166 | + let sdkApp2 = CoderSDK.WorkspaceApp( |
| 167 | + id: UUID(), |
| 168 | + url: URL(string: "https://localhost:3000/app2")!, |
| 169 | + external: false, |
| 170 | + slug: "app2", |
| 171 | + display_name: "App 2", |
| 172 | + command: nil, |
| 173 | + icon: URL(string: "/icon/app2.svg")!, |
| 174 | + subdomain: false, |
| 175 | + subdomain_name: nil |
| 176 | + ) |
| 177 | + |
| 178 | + // Command app; skipped |
| 179 | + let sdkApp3 = CoderSDK.WorkspaceApp( |
| 180 | + id: UUID(), |
| 181 | + url: URL(string: "https://localhost:3000/app3")!, |
| 182 | + external: false, |
| 183 | + slug: "app3", |
| 184 | + display_name: "App 3", |
| 185 | + command: "echo 'skip me'", |
| 186 | + icon: nil, |
| 187 | + subdomain: false, |
| 188 | + subdomain_name: nil |
| 189 | + ) |
| 190 | + |
| 191 | + let agent = createMockAgent(apps: [sdkApp1, sdkApp2, sdkApp3], displayApps: [.vscode]) |
| 192 | + let apps = agentToApps(logger, agent, host, baseAccessURL, sessionToken) |
| 193 | + |
| 194 | + #expect(apps.count == 3) |
| 195 | + let appSlugs = apps.map(\.slug) |
| 196 | + #expect(appSlugs.contains("app1")) |
| 197 | + #expect(appSlugs.contains("app2")) |
| 198 | + #expect(!appSlugs.contains("app3")) |
| 199 | + #expect(appSlugs.contains("-vscode")) |
| 200 | + } |
| 201 | + |
| 202 | + private func createMockAgent( |
| 203 | + apps: [CoderSDK.WorkspaceApp] = [], |
| 204 | + displayApps: [DisplayApp] = [] |
| 205 | + ) -> CoderSDK.WorkspaceAgent { |
| 206 | + CoderSDK.WorkspaceAgent( |
| 207 | + id: UUID(), |
| 208 | + expanded_directory: "/home/user", |
| 209 | + apps: apps, |
| 210 | + display_apps: displayApps |
| 211 | + ) |
| 212 | + } |
| 213 | +} |
0 commit comments