host_cmd_test.go RawHistoryFolder

1// SPDX-License-Identifier: MIT
2
3package main
4
5import (
6	"os"
7	"path/filepath"
8	"testing"
9
10	"bnema.dev/ermite-runner/internal/host"
11)
12
13func TestFirstRunToken(t *testing.T) {
14	env := func(m map[string]string) func(string) string {
15		return func(k string) string { return m[k] }
16	}
17	if tok, err := firstRunToken(env(nil)); err != nil || tok != "" {
18		t.Fatalf("no token: %q %v", tok, err)
19	}
20	if tok, err := firstRunToken(env(map[string]string{host.EnvRegisterToken: " t1 \n"})); err != nil || tok != "t1" {
21		t.Fatalf("env: %q %v", tok, err)
22	}
23	p := filepath.Join(t.TempDir(), "token")
24	if err := os.WriteFile(p, []byte("t2\n"), 0o600); err != nil {
25		t.Fatal(err)
26	}
27	// The file wins over the environment.
28	if tok, err := firstRunToken(env(map[string]string{envRegisterTokenFile: p, host.EnvRegisterToken: "t1"})); err != nil || tok != "t2" {
29		t.Fatalf("file: %q %v", tok, err)
30	}
31	if _, err := firstRunToken(env(map[string]string{envRegisterTokenFile: filepath.Join(t.TempDir(), "missing")})); err == nil {
32		t.Fatal("a missing token file was accepted")
33	}
34}