Privatae
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
// serveConsoleAt runs a console with `secret` on a fresh loopback port and
|
|
// records that port in this account's ui.addr, as a running daemon does.
|
|
func serveConsoleAt(t *testing.T, secret string) {
|
|
t.Helper()
|
|
ln, err := net.Listen("tcp", "127.0.0.1:0")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
u, err := newUIServer(fmt.Sprint(ln.Addr().(*net.TCPAddr).Port), secret)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
h := &http.Server{Handler: u.handler()}
|
|
go func() { _ = h.Serve(ln) }()
|
|
t.Cleanup(func() { _ = h.Close() })
|
|
p, err := uiAddrPath()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(p, []byte(ln.Addr().String()), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
// With no Host, `status` decides "running" from the console. A console that
|
|
// answers on the recorded port but belongs to another account's bastion (the
|
|
// system service beside a person's own, say) is not this one running.
|
|
func TestStatusWithNoHostCountsOnlyThisAccountsOwnConsole(t *testing.T) {
|
|
uiTestConfigDir(t)
|
|
if _, err := loadOrCreateUISecret(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
serveConsoleAt(t, "another-accounts-secret-aaaaaaaaaaaaaaaaaaaa")
|
|
if uiDaemonListening() {
|
|
t.Fatal("another account's console on the recorded port read as this bastion running")
|
|
}
|
|
}
|
|
|
|
func TestStatusWithNoHostSeesItsOwnRunningConsole(t *testing.T) {
|
|
uiTestConfigDir(t)
|
|
secret, err := loadOrCreateUISecret()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
serveConsoleAt(t, secret)
|
|
if !uiDaemonListening() {
|
|
t.Fatal("this account's own console, up and answering, did not read as running")
|
|
}
|
|
}
|
|
|
|
func TestStatusWithNoHostAndNoDaemonIsNotRunning(t *testing.T) {
|
|
uiTestConfigDir(t)
|
|
if uiDaemonListening() {
|
|
t.Fatal("nothing recorded and nothing listening read as running")
|
|
}
|
|
}
|