systemctl/filtererr_test.go at master · taigrr/systemctl · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
package systemctl

import (
	"errors"
	"testing"
)

func TestFilterErr(t *testing.T) {
	tests := []struct {
		name   string
		stderr string
		want   error
	}{
		{
			name:   "empty stderr",
			stderr: "",
			want:   nil,
		},
		{
			name:   "unit does not exist",
			stderr: "Unit foo.service does not exist, proceeding anyway.",
			want:   ErrDoesNotExist,
		},
		{
			name:   "unit not found",
			stderr: "Unit foo.service not found.",
			want:   ErrDoesNotExist,
		},
		{
			name:   "unit not loaded",
			stderr: "Unit foo.service not loaded.",
			want:   ErrUnitNotLoaded,
		},
		{
			name:   "no such file or directory",
			stderr: "No such file or directory",
			want:   ErrDoesNotExist,
		},
		{
			name:   "interactive authentication required",
			stderr: "Interactive authentication required.",
			want:   ErrInsufficientPermissions,
		},
		{
			name:   "access denied",
			stderr: "Access denied",
			want:   ErrInsufficientPermissions,
		},
		{
			name:   "dbus session bus address",
			stderr: "Failed to connect to bus: $DBUS_SESSION_BUS_ADDRESS not set",
			want:   ErrBusFailure,
		},
		{
			name:   "unit is masked",
			stderr: "Unit foo.service is masked.",
			want:   ErrMasked,
		},
		{
			name:   "generic failed",
			stderr: "Failed to do something unknown",
			want:   ErrUnspecified,
		},
		{
			name:   "does not exist with auth required prioritizes permission error",
			stderr: "Unit nginx.service does not exist, proceeding anyway.\nFailed to mask unit: Interactive authentication required.",
			want:   ErrInsufficientPermissions,
		},
		{
			name:   "does not exist with access denied prioritizes permission error",
			stderr: "Unit foo.service does not exist, proceeding anyway.\nAccess denied",
			want:   ErrInsufficientPermissions,
		},
		{
			name:   "does not exist with bus failure prioritizes bus error",
			stderr: "Unit foo.service does not exist, proceeding anyway.\n$DBUS_SESSION_BUS_ADDRESS not set",
			want:   ErrBusFailure,
		},
		{
			name:   "unrecognized warning",
			stderr: "Warning: something benign happened",
			want:   nil,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := filterErr(tt.stderr)
			if tt.want == nil {
				if got != nil {
					t.Errorf("filterErr(%q) = %v, want nil", tt.stderr, got)
				}
				return
			}
			if !errors.Is(got, tt.want) {
				t.Errorf("filterErr(%q) = %v, want error wrapping %v", tt.stderr, got, tt.want)
			}
		})
	}
}

func TestHasValidUnitSuffix(t *testing.T) {
	tests := []struct {
		unit string
		want bool
	}{
		{"nginx.service", true},
		{"sshd.socket", true},
		{"backup.timer", true},
		{"dev-sda1.device", true},
		{"home.mount", true},
		{"dev-sda1.swap", true},
		{"user.slice", true},
		{"multi-user.target", true},
		{"session-1.scope", true},
		{"foo.automount", true},
		{"backup.path", true},
		{"foo.snapshot", true},
		{"nginx", false},
		{"", false},
		{"foo.bar", false},
		{"foo.services", false},
		{".service", true},
	}

	for _, tt := range tests {
		t.Run(tt.unit, func(t *testing.T) {
			got := HasValidUnitSuffix(tt.unit)
			if got != tt.want {
				t.Errorf("HasValidUnitSuffix(%q) = %v, want %v", tt.unit, got, tt.want)
			}
		})
	}
}