| 1 | package bootstrap |
| 2 | |
| 3 | import ( |
| 4 | "context" |
| 5 | "errors" |
| 6 | "os" |
| 7 | "slices" |
| 8 | "strings" |
| 9 | "testing" |
| 10 | "time" |
| 11 | |
| 12 | "reasonix/internal/remote" |
| 13 | ) |
| 14 | |
| 15 | func TestLaunchCommandExportsBrowserBrokerEnvOnly(t *testing.T) { |
| 16 | paths := StatePaths{Dir: "/d", TokenFile: "/d/t", PortFile: "/d/p", PidFile: "/d/i", LogFile: "/d/l"} |
| 17 | cmd := LaunchCommand("/usr/bin/reasonix", "/ws", paths, nil, &BrowserBrokerOptions{ |
| 18 | BaseURL: "http://127.0.0.1:41234", Token: "br'oker $tok", |
| 19 | }) |
| 20 | want := `REASONIX_BROWSER_BROKER='http://127.0.0.1:41234' REASONIX_BROWSER_TOKEN='br'\''oker $tok' $SX nohup '/usr/bin/reasonix' serve` |
| 21 | if !strings.Contains(cmd, want) { |
| 22 | t.Fatalf("launch command lacks the quoted broker environment:\n%s", cmd) |
| 23 | } |
| 24 | if strings.Contains(cmd, "--browser") { |
| 25 | t.Fatalf("broker must not travel in argv:\n%s", cmd) |
| 26 | } |
| 27 | plain := LaunchCommand("/usr/bin/reasonix", "/ws", paths, nil, nil) |
| 28 | if strings.Contains(plain, "REASONIX_BROWSER") { |
| 29 | t.Fatalf("no broker must leave no environment behind:\n%s", plain) |
| 30 | } |
| 31 | if !strings.Contains(plain, "SX=setsid; $SX nohup") { |
| 32 | t.Fatalf("empty prefix changed the launch shape:\n%s", plain) |
| 33 | } |
| 34 | } |
| 35 | |
| 36 | func TestBrowserBrokerSupportedCommandProbesHelpMarker(t *testing.T) { |
| 37 | cmd := BrowserBrokerSupportedCommand("/home/x/'; rm -rf ~; echo '/reasonix") |
| 38 | for _, want := range []string{"serve --help", "'browser-broker'", "echo yes", "echo no", `'\''; rm -rf ~; echo '\''`} { |
| 39 | if !strings.Contains(cmd, want) { |
| 40 | t.Errorf("probe command missing %q:\n%s", want, cmd) |
| 41 | } |
| 42 | } |
| 43 | } |
| 44 | |
| 45 | // launchConn scripts a cold start whose located binary answers the browser |
| 46 | // probe with answer; it records the launch command for inspection. |
| 47 | func launchConn(t *testing.T, root, answer string) (*fakeConn, *string) { |
| 48 | t.Helper() |
| 49 | var launch string |
| 50 | paths := pathsFor(root, root) |
| 51 | conn := newFakeConn(t, root, func(cmd string) (remote.ExecResult, error) { |
| 52 | switch { |
| 53 | case strings.Contains(cmd, "uname"): |
| 54 | return ok("Linux x86_64\n") |
| 55 | case strings.Contains(cmd, "command -v reasonix"): |
| 56 | return ok("/usr/bin/reasonix\nreasonix v9.9.0\nportfile:yes\nsessionevents:yes\ndetachedheal:yes\ncaps:yes\n") |
| 57 | case strings.Contains(cmd, "grep -q -- 'browser-broker'"): |
| 58 | return ok(answer + "\n") |
| 59 | case strings.Contains(cmd, "nohup"): |
| 60 | launch = cmd |
| 61 | _ = os.WriteFile(paths.PortFile, []byte("127.0.0.1:44321\n"), 0o600) |
| 62 | return ok("54321\n") |
| 63 | case strings.Contains(cmd, "ps -p 54321"): |
| 64 | return ok("1\n") |
| 65 | default: |
| 66 | return ok("") |
| 67 | } |
| 68 | }) |
| 69 | return conn, &launch |
| 70 | } |
| 71 | |
| 72 | func TestEnsureServeLaunchesWithBrowserBrokerWhenAdvertised(t *testing.T) { |
| 73 | skipOnWindows(t) |
| 74 | root := t.TempDir() |
| 75 | conn, launch := launchConn(t, root, "yes") |
| 76 | calls := 0 |
| 77 | res, err := EnsureServe(context.Background(), conn, Options{ |
| 78 | Workspace: "~", MinVersion: "1.0.0", Clock: time.Now, |
| 79 | BrowserBroker: func(context.Context) (*BrowserBrokerOptions, error) { |
| 80 | calls++ |
| 81 | return &BrowserBrokerOptions{BaseURL: "http://127.0.0.1:5151", Token: "gen-1"}, nil |
| 82 | }, |
| 83 | }) |
| 84 | if err != nil || res.Reused { |
| 85 | t.Fatalf("EnsureServe: %+v %v", res, err) |
| 86 | } |
| 87 | if calls != 1 { |
| 88 | t.Fatalf("broker callback ran %d times, want once before launch", calls) |
| 89 | } |
| 90 | if !strings.Contains(*launch, "REASONIX_BROWSER_BROKER='http://127.0.0.1:5151' REASONIX_BROWSER_TOKEN='gen-1' $SX nohup") { |
| 91 | t.Fatalf("launch command lacks the broker environment:\n%s", *launch) |
| 92 | } |
| 93 | } |
| 94 | |
| 95 | func TestEnsureServeSkipsBrowserBrokerForOlderServe(t *testing.T) { |
| 96 | skipOnWindows(t) |
| 97 | root := t.TempDir() |
| 98 | conn, launch := launchConn(t, root, "no") |
| 99 | called := false |
| 100 | if _, err := EnsureServe(context.Background(), conn, Options{ |
| 101 | Workspace: "~", MinVersion: "1.0.0", Clock: time.Now, |
| 102 | BrowserBroker: func(context.Context) (*BrowserBrokerOptions, error) { |
| 103 | called = true |
| 104 | return &BrowserBrokerOptions{BaseURL: "http://127.0.0.1:5151", Token: "gen-1"}, nil |
| 105 | }, |
| 106 | }); err != nil { |
| 107 | t.Fatalf("EnsureServe: %v", err) |
| 108 | } |
| 109 | if called { |
| 110 | t.Fatal("broker callback must not run (and no forward be opened) for a serve that does not advertise browser-broker") |
| 111 | } |
| 112 | if strings.Contains(*launch, "REASONIX_BROWSER") { |
| 113 | t.Fatalf("older serve received broker environment:\n%s", *launch) |
| 114 | } |
| 115 | } |
| 116 | |
| 117 | func TestEnsureServeLaunchesWithoutBrowserWhenBrokerFails(t *testing.T) { |
| 118 | skipOnWindows(t) |
| 119 | root := t.TempDir() |
| 120 | conn, launch := launchConn(t, root, "yes") |
| 121 | var steps []string |
| 122 | res, err := EnsureServe(context.Background(), conn, Options{ |
| 123 | Workspace: "~", MinVersion: "1.0.0", Clock: time.Now, |
| 124 | Progress: func(step, detail string) { steps = append(steps, step+":"+detail) }, |
| 125 | BrowserBroker: func(context.Context) (*BrowserBrokerOptions, error) { |
| 126 | return nil, errors.New("reverse tunnel refused") |
| 127 | }, |
| 128 | }) |
| 129 | if err != nil || res.State.PID != 54321 { |
| 130 | t.Fatalf("a failed broker must not block the launch: %+v %v", res, err) |
| 131 | } |
| 132 | if strings.Contains(*launch, "REASONIX_BROWSER") { |
| 133 | t.Fatalf("failed broker leaked environment:\n%s", *launch) |
| 134 | } |
| 135 | if !slices.Contains(steps, "browser_broker:unavailable: reverse tunnel refused") { |
| 136 | t.Fatalf("progress did not report the degraded launch: %v", steps) |
| 137 | } |
| 138 | } |
| 139 |