87 lines
3.0 KiB
Swift
87 lines
3.0 KiB
Swift
import Foundation
|
|
import Testing
|
|
@testable import CodexBarCore
|
|
|
|
#if os(Linux)
|
|
struct AntigravityCLIStrategyLinuxTests {
|
|
@Test
|
|
func `cli local strategy is available with HTTP fallback`() async throws {
|
|
let binaryURL = FileManager.default.temporaryDirectory
|
|
.appendingPathComponent("codexbar-antigravity-\(UUID().uuidString)")
|
|
try Data("#!/bin/sh\n".utf8).write(to: binaryURL)
|
|
try FileManager.default.setAttributes(
|
|
[.posixPermissions: 0o755],
|
|
ofItemAtPath: binaryURL.path)
|
|
defer { try? FileManager.default.removeItem(at: binaryURL) }
|
|
|
|
let context = ProviderFetchContext(
|
|
runtime: .cli,
|
|
sourceMode: .cli,
|
|
includeCredits: false,
|
|
webTimeout: 1,
|
|
webDebugDumpHTML: false,
|
|
verbose: false,
|
|
env: ["ANTIGRAVITY_CLI_PATH": binaryURL.path],
|
|
settings: nil,
|
|
fetcher: UsageFetcher(environment: [:]),
|
|
claudeFetcher: StubClaudeFetcher(),
|
|
browserDetection: BrowserDetection(cacheTTL: 0))
|
|
let isAvailable = await AntigravityCLIHTTPSFetchStrategy().isAvailable(context)
|
|
|
|
#expect(isAvailable)
|
|
}
|
|
|
|
@Test
|
|
func `cli local endpoints include Linux HTTP fallback`() {
|
|
#expect(
|
|
AntigravityStatusProbe.cliEndpoints(ports: [55624]) == [
|
|
AntigravityStatusProbe.AntigravityConnectionEndpoint(
|
|
scheme: "https",
|
|
port: 55624,
|
|
csrfToken: "",
|
|
source: .cliHTTPS),
|
|
AntigravityStatusProbe.AntigravityConnectionEndpoint(
|
|
scheme: "http",
|
|
port: 55624,
|
|
csrfToken: "",
|
|
source: .cliHTTPS),
|
|
])
|
|
}
|
|
|
|
@Test
|
|
func `language server endpoints include Linux HTTP fallback`() {
|
|
#expect(
|
|
AntigravityStatusProbe.connectionCandidates(
|
|
listeningPorts: [64440],
|
|
languageServerCSRFToken: "language-token",
|
|
extensionServerPort: nil,
|
|
extensionServerCSRFToken: nil) == [
|
|
AntigravityStatusProbe.AntigravityConnectionEndpoint(
|
|
scheme: "https",
|
|
port: 64440,
|
|
csrfToken: "language-token",
|
|
source: .languageServer),
|
|
AntigravityStatusProbe.AntigravityConnectionEndpoint(
|
|
scheme: "http",
|
|
port: 64440,
|
|
csrfToken: "language-token",
|
|
source: .languageServer),
|
|
])
|
|
}
|
|
|
|
private struct StubClaudeFetcher: ClaudeUsageFetching {
|
|
func loadLatestUsage(model _: String) async throws -> ClaudeUsageSnapshot {
|
|
throw ClaudeUsageError.parseFailed("stub")
|
|
}
|
|
|
|
func debugRawProbe(model _: String) async -> String {
|
|
"stub"
|
|
}
|
|
|
|
func detectVersion() -> String? {
|
|
nil
|
|
}
|
|
}
|
|
}
|
|
#endif
|