openclaw/apps/macos/Tests/OpenClawIPCTests/GatewayProcessManagerTests.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
        }
    }
}