1 // Copyright (C) 2021 Simon Ruderich
3 // This program is free software: you can redistribute it and/or modify
4 // it under the terms of the GNU General Public License as published by
5 // the Free Software Foundation, either version 3 of the License, or
6 // (at your option) any later version.
8 // This program is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 // GNU General Public License for more details.
13 // You should have received a copy of the GNU General Public License
14 // along with this program. If not, see <http://www.gnu.org/licenses/>.
24 "ruderich.org/simon/safcm/testutil"
27 func TestLoadGroups(t *testing.T) {
28 cwd, err := os.Getwd()
34 err = os.Chdir("../testdata/project")
38 hosts, err := LoadHosts()
51 exp map[string][]string
56 "../testdata/project",
83 "all_except_some:remove": {
100 "../testdata/project",
102 GroupOrder: []string{
109 fmt.Errorf("config.yaml: group_order: group \"does-not-exist\" does not exist"),
112 "../testdata/project",
114 GroupOrder: []string{
121 fmt.Errorf("config.yaml: group_order: invalid group name \"special:group\""),
124 "../testdata/project",
126 GroupOrder: []string{
133 fmt.Errorf("config.yaml: group_order: invalid group name \"group:remove\""),
137 "../testdata/group-invalid-all",
141 fmt.Errorf("groups.yaml: group \"all\": conflict with pre-defined group \"all\""),
144 "../testdata/group-invalid-all-remove",
148 fmt.Errorf("groups.yaml: group \"all:remove\": conflict with pre-defined group \"all:remove\""),
151 "../testdata/group-invalid-conflict",
155 fmt.Errorf("groups.yaml: group \"host2\": conflict with existing host"),
158 "../testdata/group-invalid-detected",
162 fmt.Errorf("groups.yaml: group \"detected_linux\": name must not start with \"detected\" (reserved for detected groups)"),
165 "../testdata/group-invalid-member",
169 fmt.Errorf("groups.yaml: group \"group1\": member \"special:member\" must not contain \":\""),
172 "../testdata/group-invalid-missing",
176 fmt.Errorf("groups.yaml: group \"1group2\": group \"does-not-exist\" not found"),
179 "../testdata/group-invalid-name",
183 fmt.Errorf("groups.yaml: group \"invalid.group.name\": name contains invalid characters (must match ^[a-z0-9_-]+$)"),
187 for _, tc := range tests {
188 t.Run(tc.path, func(t *testing.T) {
189 err := os.Chdir(filepath.Join(cwd, tc.path))
194 res, err := LoadGroups(tc.cfg, tc.hosts)
195 testutil.AssertEqual(t, "res", res, tc.exp)
196 testutil.AssertErrorEqual(t, "err", err, tc.expErr)
201 func TestResolveHostGroups(t *testing.T) {
202 cwd, err := os.Getwd()
208 err = os.Chdir("../testdata/project")
212 allHosts, err := LoadHosts()
216 allGroups, err := LoadGroups(&Config{}, allHosts)
277 "host1, detected_mips",
291 "host2, detected_linux",
306 for _, tc := range tests {
307 t.Run(tc.name, func(t *testing.T) {
308 res, err := ResolveHostGroups(tc.host, allGroups,
310 testutil.AssertEqual(t, "res", res, tc.exp)
311 testutil.AssertErrorEqual(t, "err", err, tc.expErr)