mirror of
https://github.com/browseros-ai/BrowserOS.git
synced 2026-05-18 19:16:22 +00:00
100 lines
2.4 KiB
Go
100 lines
2.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"browseros-cli/config"
|
|
)
|
|
|
|
func TestProbeRunningServerUsesDiscoveryBeforeConfig(t *testing.T) {
|
|
home := t.TempDir()
|
|
t.Setenv("HOME", home)
|
|
t.Setenv("USERPROFILE", home)
|
|
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
|
|
t.Setenv("BROWSEROS_URL", "")
|
|
|
|
discoveredServer := newHealthyServer(t)
|
|
configServer := newHealthyServer(t)
|
|
|
|
serverDir := filepath.Join(home, ".browseros")
|
|
if err := os.MkdirAll(serverDir, 0755); err != nil {
|
|
t.Fatalf("os.MkdirAll() error = %v", err)
|
|
}
|
|
data := []byte(fmt.Sprintf(`{"url":%q}`, discoveredServer.URL))
|
|
if err := os.WriteFile(filepath.Join(serverDir, "server.json"), data, 0644); err != nil {
|
|
t.Fatalf("os.WriteFile() error = %v", err)
|
|
}
|
|
if err := config.Save(&config.Config{ServerURL: configServer.URL}); err != nil {
|
|
t.Fatalf("config.Save() error = %v", err)
|
|
}
|
|
|
|
got := probeRunningServer()
|
|
if got != normalizeServerURL(discoveredServer.URL) {
|
|
t.Fatalf("probeRunningServer() = %q, want %q", got, normalizeServerURL(discoveredServer.URL))
|
|
}
|
|
}
|
|
|
|
func TestWaitForServerUsesCommonPortFallback(t *testing.T) {
|
|
home := t.TempDir()
|
|
t.Setenv("HOME", home)
|
|
t.Setenv("USERPROFILE", home)
|
|
|
|
server := newHealthyServer(t)
|
|
port := serverPort(t, server.URL)
|
|
|
|
originalPorts := commonBrowserOSPorts
|
|
commonBrowserOSPorts = []int{port}
|
|
t.Cleanup(func() {
|
|
commonBrowserOSPorts = originalPorts
|
|
})
|
|
|
|
got, ok := waitForServer(100 * time.Millisecond)
|
|
if !ok {
|
|
t.Fatal("waitForServer() ok = false, want true")
|
|
}
|
|
if got != normalizeServerURL(server.URL) {
|
|
t.Fatalf("waitForServer() = %q, want %q", got, normalizeServerURL(server.URL))
|
|
}
|
|
}
|
|
|
|
func newHealthyServer(t *testing.T) *httptest.Server {
|
|
t.Helper()
|
|
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path != "/health" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
}))
|
|
t.Cleanup(server.Close)
|
|
return server
|
|
}
|
|
|
|
func serverPort(t *testing.T, rawURL string) int {
|
|
t.Helper()
|
|
|
|
parsed, err := url.Parse(rawURL)
|
|
if err != nil {
|
|
t.Fatalf("url.Parse() error = %v", err)
|
|
}
|
|
_, portText, err := net.SplitHostPort(parsed.Host)
|
|
if err != nil {
|
|
t.Fatalf("net.SplitHostPort() error = %v", err)
|
|
}
|
|
port, err := strconv.Atoi(portText)
|
|
if err != nil {
|
|
t.Fatalf("strconv.Atoi() error = %v", err)
|
|
}
|
|
return port
|
|
}
|