Files
steipete--codexbar/TestsLinux/AntigravityCLIStrategyLinuxTests.swift
T
2026-07-13 12:22:33 +08:00

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