native macOS codings agent orchestrator prowl.onev.cat
Something went wrong. Try again.
9.4 kB · 208 lines
Swift
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209import Darwinimport Foundationimport Testing
@testable import supacode
@Suite(.serialized) struct CodexConfigReadProcessTests { @Test func appServerInputPipeSuppressesSIGPIPE() { let pipe = Pipe() defer { try? pipe.fileHandleForReading.close() try? pipe.fileHandleForWriting.close() } let descriptor = pipe.fileHandleForWriting.fileDescriptor #expect(fcntl(descriptor, F_GETNOSIGPIPE) == 0)
#expect(CodexConfigReadProcess.configureNoSigPipe(descriptor)) #expect(fcntl(descriptor, F_GETNOSIGPIPE) == 1) }
@Test func appServerExitBeforeInputFailsWithoutWaitingForTimeout() async throws { let root = temporaryDirectory("codex-early-exit") let home = root.appending(path: "home", directoryHint: .isDirectory) try FileManager.default.createDirectory(at: home, withIntermediateDirectories: true) defer { try? FileManager.default.removeItem(at: root) } let executable = root.appending(path: "early-exit.sh", directoryHint: .notDirectory) try "#!/bin/sh\nexit 0\n".write(to: executable, atomically: true, encoding: .utf8) try FileManager.default.setAttributes([.posixPermissions: 0o700], ofItemAtPath: executable.path) let process = CodexConfigReadProcess( executableURL: executable, temporaryBaseDirectory: root, timeout: 2 ) await #expect(throws: CodexConfigReadProcessError.processFailed) { try await process.query( CodexConfigQuery(kind: .base, codexHome: home, cwd: root, overrides: []) ) } }
@Test(arguments: [#"{"notify":null}"#, "{}"]) func absentNotifierCompletesWhileServerRemainsOpen(config: String) async throws { let root = temporaryDirectory("codex-absent-notifier") try FileManager.default.createDirectory(at: root, withIntermediateDirectories: true) defer { try? FileManager.default.removeItem(at: root) } let executable = root.appending(path: "no-notifier.sh") try """ #!/bin/sh for request in 1 2 3; do IFS= read -r line || exit 1 done printf '%s\\n' '{"id":2,"result":{"config":\(config)}}' while IFS= read -r line; do :; done """.write(to: executable, atomically: true, encoding: .utf8) try FileManager.default.setAttributes([.posixPermissions: 0o700], ofItemAtPath: executable.path) // This asserts response semantics, not startup latency under parallel CI load. Keep the // server open using shell builtins and allow the same budget as the other response fixtures. let process = CodexConfigReadProcess(executableURL: executable, temporaryBaseDirectory: root, timeout: 15) let transcript = try await process.query( CodexConfigQuery(kind: .base, codexHome: root, cwd: root, overrides: []) ) #expect(try CodexConfigReadProtocol.decodeNotify(from: transcript) == nil) }
@Test func requestStdinStaysOpenUntilTheConfigReadResponseArrives() async throws { // Codex 0.149.1's app-server tears down on stdin EOF and drops any request it has not // answered yet, so the request pipe must outlive the response. let root = temporaryDirectory("codex-eof") let home = root.appending(path: "home", directoryHint: .isDirectory) try FileManager.default.createDirectory(at: home, withIntermediateDirectories: true) defer { try? FileManager.default.removeItem(at: root) } let executable = root.appending(path: "eof-exit-codex.py", directoryHint: .notDirectory) try """ #!/usr/bin/python3 import json, select, sys for _ in range(3): sys.stdin.readline() readable, _, _ = select.select([sys.stdin], [], [], 0.5) if readable and sys.stdin.readline() == "": sys.exit(0) print(json.dumps({"jsonrpc": "2.0", "id": 2, "result": {"config": {"notify": ["/tmp/n", "x"]}}}), flush=True) """.write(to: executable, atomically: true, encoding: .utf8) try FileManager.default.setAttributes([.posixPermissions: 0o700], ofItemAtPath: executable.path) let process = CodexConfigReadProcess( executableURL: executable, temporaryBaseDirectory: root, timeout: 15 )
let transcript = try await process.query( CodexConfigQuery(kind: .base, codexHome: home, cwd: root, overrides: []) )
#expect(try CodexConfigReadProtocol.decodeNotify(from: transcript) == ["/tmp/n", "x"]) }
@Test func profileParserHomeIsOwnerOnlyAndRemovedAfterResponse() async throws { let root = temporaryDirectory("codex-process") let parser = root.appending(path: "parser", directoryHint: .isDirectory) let home = root.appending(path: "home", directoryHint: .isDirectory) let profile = home.appending(path: "selected.config.toml", directoryHint: .notDirectory) let report = root.appending(path: "report.json", directoryHint: .notDirectory) try FileManager.default.createDirectory(at: home, withIntermediateDirectories: true) try #"notify = ["/tmp/profile", "space value", ""]"#.write( to: profile, atomically: true, encoding: .utf8 ) let executable = try makeFakeCodex(in: root, report: report, sleeps: false) defer { try? FileManager.default.removeItem(at: root) } let process = CodexConfigReadProcess( executableURL: executable, temporaryBaseDirectory: parser, timeout: 15 ) let query = CodexConfigQuery( kind: .profile(profile), codexHome: home, cwd: root, overrides: [] )
let transcript = try await process.query(query)
#expect(try CodexConfigReadProtocol.decodeNotify(from: transcript) == ["/tmp/profile", "space value", ""]) let observed = try #require( JSONSerialization.jsonObject(with: Data(contentsOf: report)) as? [String: Any] ) #expect((observed["directory_mode"] as? NSNumber)?.intValue == 0o700) #expect((observed["config_mode"] as? NSNumber)?.intValue == 0o600) #expect((try? FileManager.default.contentsOfDirectory(atPath: parser.path))?.isEmpty == true) }
@Test func cancellationTerminatesParserAndRemovesScratchHomeWithoutDispatchSleep() async throws { let root = temporaryDirectory("codex-cancel") let parser = root.appending(path: "parser", directoryHint: .isDirectory) let home = root.appending(path: "home", directoryHint: .isDirectory) let profile = home.appending(path: "selected.config.toml", directoryHint: .notDirectory) let marker = root.appending(path: "started", directoryHint: .notDirectory) try FileManager.default.createDirectory(at: home, withIntermediateDirectories: true) try "notify = [\"/tmp/profile\"]".write(to: profile, atomically: true, encoding: .utf8) let executable = try makeFakeCodex(in: root, report: marker, sleeps: true) defer { try? FileManager.default.removeItem(at: root) } let process = CodexConfigReadProcess( executableURL: executable, temporaryBaseDirectory: parser, timeout: 30 ) // Detached so the query does not compete with this test's poll loop for // the main actor: a yield-only loop is not guaranteed to let a sibling // main-actor task run, which left the scratch home unobserved until the // 30 s process timeout on macOS 27. let task = Task.detached { try await process.query( CodexConfigQuery( kind: .profile(profile), codexHome: home, cwd: root, overrides: [] ) ) } let spawnDeadline = ContinuousClock.now.advanced(by: .seconds(10)) while ((try? FileManager.default.contentsOfDirectory(atPath: parser.path)) ?? []).isEmpty, ContinuousClock.now < spawnDeadline { await Task.yield() } #expect(((try? FileManager.default.contentsOfDirectory(atPath: parser.path)) ?? []).count == 1)
task.cancel() await #expect(throws: (any Error).self) { try await task.value } #expect((try? FileManager.default.contentsOfDirectory(atPath: parser.path))?.isEmpty == true) }
private func makeFakeCodex(in root: URL, report: URL, sleeps: Bool) throws -> URL { let executable = root.appending(path: "fake-codex.py", directoryHint: .notDirectory) let script = """ #!/usr/bin/python3 import json, os, pathlib, stat, sys, time for _ in range(3): sys.stdin.readline() home = pathlib.Path(os.environ["CODEX_HOME"]) config = home / "config.toml" report = pathlib.Path(\(String(reflecting: report.path(percentEncoded: false)))) if \(sleeps ? "True" : "False"): report.write_text("started") time.sleep(30) text = config.read_text() if config.exists() else "" notify = ["/tmp/profile", "space value", ""] if "space value" in text else None report.write_text(json.dumps({ "directory_mode": stat.S_IMODE(home.stat().st_mode), "config_mode": stat.S_IMODE(config.stat().st_mode) if config.exists() else None })) print(json.dumps({"jsonrpc":"2.0","id":2,"result":{"config":{"notify":notify}}}), flush=True) """ try script.write(to: executable, atomically: true, encoding: .utf8) try FileManager.default.setAttributes([.posixPermissions: 0o700], ofItemAtPath: executable.path) return executable }
private func temporaryDirectory(_ name: String) -> URL { FileManager.default.temporaryDirectory.appending( path: "prowl-tests-\(name)-\(UUID().uuidString)", directoryHint: .isDirectory ) }}