132 lines
5.3 KiB
Swift
132 lines
5.3 KiB
Swift
import Darwin
|
|
import Foundation
|
|
import Testing
|
|
@testable import Clawdbot
|
|
|
|
@Suite(.serialized) struct CommandResolverTests {
|
|
private func makeDefaults() -> UserDefaults {
|
|
// Use a unique suite to avoid cross-suite concurrency on UserDefaults.standard.
|
|
UserDefaults(suiteName: "CommandResolverTests.\(UUID().uuidString)")!
|
|
}
|
|
|
|
private func makeTempDir() throws -> URL {
|
|
let base = URL(fileURLWithPath: NSTemporaryDirectory(), isDirectory: true)
|
|
let dir = base.appendingPathComponent(UUID().uuidString, isDirectory: true)
|
|
try FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true)
|
|
return dir
|
|
}
|
|
|
|
private func makeExec(at path: URL) throws {
|
|
try FileManager.default.createDirectory(
|
|
at: path.deletingLastPathComponent(),
|
|
withIntermediateDirectories: true)
|
|
FileManager.default.createFile(atPath: path.path, contents: Data("echo ok\n".utf8))
|
|
try FileManager.default.setAttributes([.posixPermissions: 0o755], ofItemAtPath: path.path)
|
|
}
|
|
|
|
@Test func prefersClawdbotBinary() async throws {
|
|
let defaults = self.makeDefaults()
|
|
defaults.set(AppState.ConnectionMode.local.rawValue, forKey: connectionModeKey)
|
|
|
|
let tmp = try makeTempDir()
|
|
CommandResolver.setProjectRoot(tmp.path)
|
|
|
|
let clawdbotPath = tmp.appendingPathComponent("node_modules/.bin/clawdbot")
|
|
try self.makeExec(at: clawdbotPath)
|
|
|
|
let cmd = CommandResolver.clawdbotCommand(subcommand: "gateway", defaults: defaults)
|
|
#expect(cmd.prefix(2).elementsEqual([clawdbotPath.path, "gateway"]))
|
|
}
|
|
|
|
@Test func fallsBackToNodeAndScript() async throws {
|
|
let defaults = self.makeDefaults()
|
|
defaults.set(AppState.ConnectionMode.local.rawValue, forKey: connectionModeKey)
|
|
|
|
let tmp = try makeTempDir()
|
|
CommandResolver.setProjectRoot(tmp.path)
|
|
|
|
let nodePath = tmp.appendingPathComponent("node_modules/.bin/node")
|
|
let scriptPath = tmp.appendingPathComponent("bin/clawdbot.js")
|
|
try self.makeExec(at: nodePath)
|
|
try "#!/bin/sh\necho v22.0.0\n".write(to: nodePath, atomically: true, encoding: .utf8)
|
|
try FileManager.default.setAttributes([.posixPermissions: 0o755], ofItemAtPath: nodePath.path)
|
|
try self.makeExec(at: scriptPath)
|
|
|
|
let cmd = CommandResolver.clawdbotCommand(
|
|
subcommand: "rpc",
|
|
defaults: defaults,
|
|
searchPaths: [tmp.appendingPathComponent("node_modules/.bin").path])
|
|
|
|
#expect(cmd.count >= 3)
|
|
if cmd.count >= 3 {
|
|
#expect(cmd[0] == nodePath.path)
|
|
#expect(cmd[1] == scriptPath.path)
|
|
#expect(cmd[2] == "rpc")
|
|
}
|
|
}
|
|
|
|
@Test func fallsBackToPnpm() async throws {
|
|
let defaults = self.makeDefaults()
|
|
defaults.set(AppState.ConnectionMode.local.rawValue, forKey: connectionModeKey)
|
|
|
|
let tmp = try makeTempDir()
|
|
CommandResolver.setProjectRoot(tmp.path)
|
|
|
|
let pnpmPath = tmp.appendingPathComponent("node_modules/.bin/pnpm")
|
|
try self.makeExec(at: pnpmPath)
|
|
|
|
let cmd = CommandResolver.clawdbotCommand(subcommand: "rpc", defaults: defaults)
|
|
|
|
#expect(cmd.prefix(4).elementsEqual([pnpmPath.path, "--silent", "clawdbot", "rpc"]))
|
|
}
|
|
|
|
@Test func pnpmKeepsExtraArgsAfterSubcommand() async throws {
|
|
let defaults = self.makeDefaults()
|
|
defaults.set(AppState.ConnectionMode.local.rawValue, forKey: connectionModeKey)
|
|
|
|
let tmp = try makeTempDir()
|
|
CommandResolver.setProjectRoot(tmp.path)
|
|
|
|
let pnpmPath = tmp.appendingPathComponent("node_modules/.bin/pnpm")
|
|
try self.makeExec(at: pnpmPath)
|
|
|
|
let cmd = CommandResolver.clawdbotCommand(
|
|
subcommand: "health",
|
|
extraArgs: ["--json", "--timeout", "5"],
|
|
defaults: defaults)
|
|
|
|
#expect(cmd.prefix(5).elementsEqual([pnpmPath.path, "--silent", "clawdbot", "health", "--json"]))
|
|
#expect(cmd.suffix(2).elementsEqual(["--timeout", "5"]))
|
|
}
|
|
|
|
@Test func preferredPathsStartWithProjectNodeBins() async throws {
|
|
let tmp = try makeTempDir()
|
|
CommandResolver.setProjectRoot(tmp.path)
|
|
|
|
let first = CommandResolver.preferredPaths().first
|
|
#expect(first == tmp.appendingPathComponent("node_modules/.bin").path)
|
|
}
|
|
|
|
@Test func buildsSSHCommandForRemoteMode() async throws {
|
|
let defaults = self.makeDefaults()
|
|
defaults.set(AppState.ConnectionMode.remote.rawValue, forKey: connectionModeKey)
|
|
defaults.set("clawd@example.com:2222", forKey: remoteTargetKey)
|
|
defaults.set("/tmp/id_ed25519", forKey: remoteIdentityKey)
|
|
defaults.set("/srv/clawdbot", forKey: remoteProjectRootKey)
|
|
|
|
let cmd = CommandResolver.clawdbotCommand(subcommand: "status", extraArgs: ["--json"], defaults: defaults)
|
|
|
|
#expect(cmd.first == "/usr/bin/ssh")
|
|
#expect(cmd.contains("clawd@example.com"))
|
|
#expect(cmd.contains("-i"))
|
|
#expect(cmd.contains("/tmp/id_ed25519"))
|
|
if let script = cmd.last {
|
|
#expect(script.contains("cd '/srv/clawdbot'"))
|
|
#expect(script.contains("clawdbot"))
|
|
#expect(script.contains("status"))
|
|
#expect(script.contains("--json"))
|
|
#expect(script.contains("CLI="))
|
|
}
|
|
}
|
|
}
|