98 lines
2.9 KiB
Swift
98 lines
2.9 KiB
Swift
import Foundation
|
|
import Testing
|
|
@testable import Clawdis
|
|
|
|
@Suite
|
|
struct PiOAuthStoreTests {
|
|
@Test
|
|
func returnsMissingWhenFileAbsent() {
|
|
let url = FileManager.default.temporaryDirectory
|
|
.appendingPathComponent("clawdis-oauth-\(UUID().uuidString)")
|
|
.appendingPathComponent("oauth.json")
|
|
#expect(PiOAuthStore.anthropicOAuthStatus(at: url) == .missingFile)
|
|
}
|
|
|
|
@Test
|
|
func usesEnvOverrideForPiAgentDir() throws {
|
|
let key = "PI_CODING_AGENT_DIR"
|
|
let previous = ProcessInfo.processInfo.environment[key]
|
|
defer {
|
|
if let previous {
|
|
setenv(key, previous, 1)
|
|
} else {
|
|
unsetenv(key)
|
|
}
|
|
}
|
|
|
|
let dir = FileManager.default.temporaryDirectory
|
|
.appendingPathComponent("clawdis-pi-agent-\(UUID().uuidString)", isDirectory: true)
|
|
setenv(key, dir.path, 1)
|
|
|
|
#expect(PiOAuthStore.oauthDir().standardizedFileURL == dir.standardizedFileURL)
|
|
}
|
|
|
|
@Test
|
|
func acceptsPiFormatTokens() throws {
|
|
let url = try self.writeOAuthFile([
|
|
"anthropic": [
|
|
"type": "oauth",
|
|
"refresh": "r1",
|
|
"access": "a1",
|
|
"expires": 1_234_567_890,
|
|
],
|
|
])
|
|
|
|
#expect(PiOAuthStore.anthropicOAuthStatus(at: url).isConnected)
|
|
}
|
|
|
|
@Test
|
|
func acceptsTokenKeyVariants() throws {
|
|
let url = try self.writeOAuthFile([
|
|
"anthropic": [
|
|
"type": "oauth",
|
|
"refresh_token": "r1",
|
|
"access_token": "a1",
|
|
],
|
|
])
|
|
|
|
#expect(PiOAuthStore.anthropicOAuthStatus(at: url).isConnected)
|
|
}
|
|
|
|
@Test
|
|
func reportsMissingProviderEntry() throws {
|
|
let url = try self.writeOAuthFile([
|
|
"other": [
|
|
"type": "oauth",
|
|
"refresh": "r1",
|
|
"access": "a1",
|
|
],
|
|
])
|
|
|
|
#expect(PiOAuthStore.anthropicOAuthStatus(at: url) == .missingProviderEntry)
|
|
}
|
|
|
|
@Test
|
|
func reportsMissingTokens() throws {
|
|
let url = try self.writeOAuthFile([
|
|
"anthropic": [
|
|
"type": "oauth",
|
|
"refresh": "",
|
|
"access": "a1",
|
|
],
|
|
])
|
|
|
|
#expect(PiOAuthStore.anthropicOAuthStatus(at: url) == .missingTokens)
|
|
}
|
|
|
|
private func writeOAuthFile(_ json: [String: Any]) throws -> URL {
|
|
let dir = FileManager.default.temporaryDirectory
|
|
.appendingPathComponent("clawdis-oauth-\(UUID().uuidString)", isDirectory: true)
|
|
try FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true)
|
|
|
|
let url = dir.appendingPathComponent("oauth.json")
|
|
let data = try JSONSerialization.data(withJSONObject: json, options: [.prettyPrinted, .sortedKeys])
|
|
try data.write(to: url, options: [.atomic])
|
|
return url
|
|
}
|
|
}
|