mirror of https://github.com/openclaw/openclaw.git
127 lines
4.8 KiB
Swift
127 lines
4.8 KiB
Swift
import Foundation
|
|
import OpenClawKit
|
|
import Testing
|
|
@testable import OpenClaw
|
|
|
|
@Suite(.serialized)
|
|
@MainActor
|
|
struct GatewayProcessManagerTests {
|
|
@Test func `clears last failure when health succeeds`() async throws {
|
|
let session = GatewayTestWebSocketSession(
|
|
taskFactory: {
|
|
GatewayTestWebSocketTask(
|
|
sendHook: { task, message, sendIndex in
|
|
guard sendIndex > 0 else { return }
|
|
guard let id = GatewayWebSocketTestSupport.requestID(from: message) else { return }
|
|
task.emitReceiveSuccess(.data(GatewayWebSocketTestSupport.okResponseData(id: id)))
|
|
})
|
|
})
|
|
let url = try #require(URL(string: "ws://example.invalid"))
|
|
let connection = GatewayConnection(
|
|
configProvider: { (url: url, token: nil, password: nil) },
|
|
sessionBox: WebSocketSessionBox(session: session))
|
|
|
|
let manager = GatewayProcessManager.shared
|
|
manager.setTestingConnection(connection)
|
|
manager.setTestingDesiredActive(true)
|
|
manager.setTestingLastFailureReason("health failed")
|
|
defer {
|
|
manager.setTestingConnection(nil)
|
|
manager.setTestingDesiredActive(false)
|
|
manager.setTestingLastFailureReason(nil)
|
|
}
|
|
|
|
let ready = await manager.waitForGatewayReady(timeout: 0.5)
|
|
#expect(ready)
|
|
#expect(manager.lastFailureReason == nil)
|
|
}
|
|
|
|
@Test func `attaches to existing gateway without spawning launchd`() async throws {
|
|
let healthData = Data(
|
|
"""
|
|
{
|
|
"ok": true,
|
|
"ts": 1,
|
|
"durationMs": 0,
|
|
"channels": {
|
|
"telegram": {
|
|
"configured": true,
|
|
"linked": true,
|
|
"authAgeMs": 60000
|
|
}
|
|
},
|
|
"channelOrder": ["telegram"],
|
|
"channelLabels": {
|
|
"telegram": "Telegram"
|
|
},
|
|
"heartbeatSeconds": 30,
|
|
"sessions": {
|
|
"path": "/tmp/sessions",
|
|
"count": 1,
|
|
"recent": []
|
|
}
|
|
}
|
|
""".utf8)
|
|
let session = GatewayTestWebSocketSession(
|
|
taskFactory: {
|
|
GatewayTestWebSocketTask(
|
|
sendHook: { task, message, sendIndex in
|
|
guard sendIndex > 0 else { return }
|
|
guard let id = GatewayWebSocketTestSupport.requestID(from: message) else { return }
|
|
let json = """
|
|
{
|
|
"type": "res",
|
|
"id": "\(id)",
|
|
"ok": true,
|
|
"payload": \(String(decoding: healthData, as: UTF8.self))
|
|
}
|
|
"""
|
|
task.emitReceiveSuccess(.data(Data(json.utf8)))
|
|
})
|
|
})
|
|
let url = try #require(URL(string: "ws://example.invalid"))
|
|
let connection = GatewayConnection(
|
|
configProvider: { (url: url, token: nil, password: nil) },
|
|
sessionBox: WebSocketSessionBox(session: session))
|
|
let port = GatewayEnvironment.gatewayPort()
|
|
let descriptor = PortGuardian.Descriptor(
|
|
pid: 4242,
|
|
command: "openclaw-gateway",
|
|
executablePath: "/tmp/openclaw-gateway")
|
|
|
|
let manager = GatewayProcessManager.shared
|
|
await PortGuardian.shared.setTestingDescriptor(descriptor, forPort: port)
|
|
manager.setTestingConnection(connection)
|
|
manager.setTestingSkipControlChannelRefresh(true)
|
|
manager.setTestingLastFailureReason("stale")
|
|
|
|
func cleanup() async {
|
|
await PortGuardian.shared.setTestingDescriptor(nil, forPort: port)
|
|
manager.setTestingConnection(nil)
|
|
manager.setTestingSkipControlChannelRefresh(false)
|
|
manager.setTestingDesiredActive(false)
|
|
manager.setTestingLastFailureReason(nil)
|
|
}
|
|
|
|
do {
|
|
let attached = await manager._testAttachExistingGatewayIfAvailable()
|
|
#expect(attached)
|
|
#expect(manager.lastFailureReason == nil)
|
|
guard case let .attachedExisting(statusDetails) = manager.status else {
|
|
Issue.record("expected attachedExisting status")
|
|
await cleanup()
|
|
return
|
|
}
|
|
let details = try #require(statusDetails)
|
|
#expect(details.contains("port \(port)"))
|
|
#expect(details.contains("Telegram linked"))
|
|
#expect(details.contains("auth 1m"))
|
|
#expect(details.contains("pid 4242 openclaw-gateway @ /tmp/openclaw-gateway"))
|
|
await cleanup()
|
|
} catch {
|
|
await cleanup()
|
|
throw error
|
|
}
|
|
}
|
|
}
|