cmd/gitbay-runner/config_test.go

 1package main
 2
 3import (
 4	"flag"
 5	"os"
 6	"path/filepath"
 7	"strings"
 8	"testing"
 9)
10
11// A config file sets the flags' values; a flag on the command line wins.
12func TestConfigFileFeedsFlagsAndFlagsOverride(t *testing.T) {
13	dir := t.TempDir()
14	path := filepath.Join(dir, "config.toml")
15	os.WriteFile(path, []byte("remote = \"git@example.test\"\npoll = \"9s\"\nuntrusted = true\nidentity = \"/k\"\njobs = 2\n"), 0o600)
16
17	values, found, err := loadConfig(path)
18	if err != nil || !found {
19		t.Fatalf("loadConfig: found=%v err=%v", found, err)
20	}
21	fs := flag.NewFlagSet("t", flag.ContinueOnError)
22	remote := fs.String("remote", "git@gitbay.org", "")
23	poll := fs.Duration("poll", 0, "")
24	untrusted := fs.Bool("untrusted", false, "")
25	identity := fs.String("identity", "", "")
26	jobs := fs.Int("jobs", 1, "")
27	if err := applyConfig(fs, values); err != nil {
28		t.Fatal(err)
29	}
30	if err := fs.Parse([]string{"-poll", "3s"}); err != nil {
31		t.Fatal(err)
32	}
33	if *remote != "git@example.test" || poll.String() != "3s" || !*untrusted || *identity != "/k" || *jobs != 2 {
34		t.Fatalf("remote=%s poll=%s untrusted=%v identity=%s jobs=%d", *remote, poll, *untrusted, *identity, *jobs)
35	}
36	if _, found, err := loadConfig(filepath.Join(dir, "missing.toml")); found || err != nil {
37		t.Fatalf("missing file: found=%v err=%v", found, err)
38	}
39	if _, _, err := loadConfig(path); err != nil {
40		t.Fatal(err)
41	}
42	os.WriteFile(path, []byte("nonsense = \"x\"\n"), 0o600)
43	if _, _, err := loadConfig(path); err == nil {
44		t.Fatal("an unknown key was accepted")
45	}
46}
47
48func TestConfigPathFromArgs(t *testing.T) {
49	for _, tc := range []struct {
50		args []string
51		want string
52	}{
53		{nil, "/def"},
54		{[]string{"-once"}, "/def"},
55		{[]string{"-config", "/a"}, "/a"},
56		{[]string{"--config", "/b", "-once"}, "/b"},
57		{[]string{"-config=/c"}, "/c"},
58	} {
59		if got := configPathFromArgs(tc.args, "/def"); got != tc.want {
60			t.Errorf("%v: got %s want %s", tc.args, got, tc.want)
61		}
62	}
63}
64
65func TestConfigDirHonoursXDG(t *testing.T) {
66	t.Setenv("XDG_CONFIG_HOME", "/x")
67	if got := configDir(); got != "/x/gitbay-runner" {
68		t.Fatalf("got %s", got)
69	}
70	t.Setenv("XDG_CONFIG_HOME", "")
71	t.Setenv("HOME", "/h")
72	if got := configDir(); got != "/h/.config/gitbay-runner" {
73		t.Fatalf("got %s", got)
74	}
75}
76
77func TestIdentityOpts(t *testing.T) {
78	if got := identityOpts(""); got != nil {
79		t.Fatalf("empty identity produced %v", got)
80	}
81	got := strings.Join(identityOpts("/k"), " ")
82	want := "-F /dev/null -i /k -o IdentitiesOnly=yes -o StrictHostKeyChecking=accept-new"
83	if got != want {
84		t.Fatalf("got %q, want %q", got, want)
85	}
86}