summaryrefslogtreecommitdiffhomepage
path: root/packages/sdk/python/tests/test_integration.py
blob: cbb6c399904568212becba66f73502cc5d7b0faa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
import os
import re
import signal
import subprocess
import sys
import time
from pathlib import Path

import httpx
import pytest

from opencode_ai import OpenCodeClient


@pytest.mark.timeout(30)
def test_integration_live_server_endpoints() -> None:
    # Locate repo root by finding sst.config.ts upwards from this file
    here = Path(__file__).resolve()
    p = here
    repo_root = None
    for _ in range(8):
        if (p / "sst.config.ts").exists():
            repo_root = p
            break
        if p.parent == p:
            break
        p = p.parent
    assert repo_root is not None, "Could not locate repo root (sst.config.ts)"

    # Start opencode headless server on a random port
    pkg_opencode = repo_root / "packages" / "opencode"
    cmd = [
        "bun",
        "run",
        "--conditions=development",
        "./src/index.ts",
        "serve",
        "--port",
        "0",
        "--hostname",
        "127.0.0.1",
    ]

    proc = subprocess.Popen(
        cmd,
        cwd=str(pkg_opencode),
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
        text=True,
        bufsize=1,
        universal_newlines=True,
    )

    url = None
    start = time.time()
    assert proc.stdout is not None
    while time.time() - start < 15:
        line = proc.stdout.readline()
        if not line:
            time.sleep(0.05)
            if proc.poll() is not None:
                break
            continue
        m = re.search(r"opencode server listening on (http://[^\s]+)", line)
        if m:
            url = m.group(1)
            break
    assert url, "Server did not report listening URL"

    try:
        client = OpenCodeClient(base_url=url)
        # Basic endpoints (avoid complex config model parsing issues)
        pinfo = client.get_path()
        assert pinfo is not None
        projects = client.list_projects()
        assert projects is not None

        # SSE: should get the initial server.connected event
        it = client.subscribe_events()
        evt = next(it)
        assert isinstance(evt, dict)
        assert evt.get("type") == "server.connected"
    finally:
        # Cleanup server process
        try:
            if proc.poll() is None:
                proc.terminate()
                try:
                    proc.wait(timeout=5)
                except subprocess.TimeoutExpired:
                    proc.kill()
        except Exception:
            pass