import (
"fmt"
"os"
- "reflect"
+ "path/filepath"
"testing"
- "github.com/google/go-cmp/cmp"
+ "ruderich.org/simon/safcm/testutil"
)
func TestLoadGroups(t *testing.T) {
{
"../testdata/project",
&Config{
- GroupOrder: []string{
+ GroupPriority: []string{
"detected_linux",
"detected_freebsd",
},
"group2:remove": {
"remove",
},
+ "group3": {
+ "host1.example.org",
+ },
+ "group3:remove": {
+ "host2",
+ },
"all_except_some": {
"all",
},
{
"../testdata/project",
&Config{
- GroupOrder: []string{
+ GroupPriority: []string{
"detected_freebsd",
"does-not-exist",
},
},
hosts,
nil,
- fmt.Errorf("config.yaml: group_order: group \"does-not-exist\" does not exist"),
+ fmt.Errorf("config.yaml: group_priority: group \"does-not-exist\" does not exist"),
},
{
"../testdata/project",
&Config{
- GroupOrder: []string{
+ GroupPriority: []string{
"detected_freebsd",
"special:group",
},
},
hosts,
nil,
- fmt.Errorf("config.yaml: group_order: invalid group name \"special:group\""),
+ fmt.Errorf("config.yaml: group_priority: invalid group name \"special:group\""),
},
{
"../testdata/project",
&Config{
- GroupOrder: []string{
+ GroupPriority: []string{
"detected_freebsd",
"group:remove",
},
},
hosts,
nil,
- fmt.Errorf("config.yaml: group_order: invalid group name \"group:remove\""),
+ fmt.Errorf("config.yaml: group_priority: invalid group name \"group:remove\""),
},
{
nil,
fmt.Errorf("groups.yaml: group \"host2\": conflict with existing host"),
},
+ {
+ "../testdata/group-invalid-conflict-remove",
+ &Config{},
+ hosts,
+ nil,
+ fmt.Errorf("groups.yaml: group \"host2:remove\": conflict with existing host"),
+ },
{
"../testdata/group-invalid-detected",
&Config{},
&Config{},
&Hosts{},
nil,
- fmt.Errorf("groups.yaml: group \"1group2\": group \"does-not-exist\" not found"),
+ fmt.Errorf("groups.yaml: group \"1group2\": member \"does-not-exist\" not found"),
},
{
"../testdata/group-invalid-name",
}
for _, tc := range tests {
- err := os.Chdir(tc.path)
- if err != nil {
- t.Fatal(err)
- }
-
- res, err := LoadGroups(tc.cfg, tc.hosts)
+ t.Run(tc.path, func(t *testing.T) {
+ err := os.Chdir(filepath.Join(cwd, tc.path))
+ if err != nil {
+ t.Fatal(err)
+ }
- if !reflect.DeepEqual(tc.exp, res) {
- t.Errorf("%s: res: %s", tc.path,
- cmp.Diff(tc.exp, res))
- }
- // Ugly but the simplest way to compare errors (including nil)
- if fmt.Sprintf("%s", err) != fmt.Sprintf("%s", tc.expErr) {
- t.Errorf("%s: err = %#v, want %#v",
- tc.path, err, tc.expErr)
- }
-
- err = os.Chdir(cwd)
- if err != nil {
- t.Fatal(err)
- }
+ res, err := LoadGroups(tc.cfg, tc.hosts)
+ testutil.AssertEqual(t, "res", res, tc.exp)
+ testutil.AssertErrorEqual(t, "err", err, tc.expErr)
+ })
}
}
[]string{
"all",
"group",
+ "group3",
"host1.example.org",
"remove",
},
[]string{
"all",
"detected_mips",
+ "group3",
"host1.example.org",
"remove",
},
}
for _, tc := range tests {
- res, err := ResolveHostGroups(tc.host, allGroups, tc.detected)
- if !reflect.DeepEqual(tc.exp, res) {
- t.Errorf("%s: res: %s", tc.name,
- cmp.Diff(tc.exp, res))
- }
- // Ugly but the simplest way to compare errors (including nil)
- if fmt.Sprintf("%s", err) != fmt.Sprintf("%s", tc.expErr) {
- t.Errorf("%s: err = %#v, want %#v",
- tc.name, err, tc.expErr)
- }
+ t.Run(tc.name, func(t *testing.T) {
+ res, err := ResolveHostGroups(tc.host, allGroups,
+ tc.detected)
+ testutil.AssertEqual(t, "res", res, tc.exp)
+ testutil.AssertErrorEqual(t, "err", err, tc.expErr)
+ })
+ }
+}
+
+func TestTransitivelyDetectedGroups(t *testing.T) {
+ tests := []struct {
+ name string
+ groups map[string][]string
+ exp []string
+ }{
+
+ {
+ "no detected",
+ map[string][]string{
+ "group-a": {
+ "a",
+ "b",
+ "group-b",
+ },
+ "group-a:remove": {
+ "d",
+ },
+ "group-b": {
+ "c",
+ "d",
+ },
+ },
+ nil,
+ },
+
+ {
+ "detected as direct member",
+ map[string][]string{
+ "group-a": {
+ "a",
+ "b",
+ "detected_foo",
+ },
+ "group-b": {
+ "c",
+ "d",
+ },
+ },
+ []string{
+ "group-a",
+ },
+ },
+
+ {
+ "detected as direct :remove member",
+ map[string][]string{
+ "group-a": {
+ "a",
+ "b",
+ "group-b",
+ },
+ "group-a:remove": {
+ "d",
+ "detected_foo",
+ },
+ "group-b": {
+ "c",
+ "d",
+ },
+ },
+ []string{
+ "group-a",
+ },
+ },
+
+ {
+ "detected as transitive member",
+ map[string][]string{
+ "group-a": {
+ "group-b",
+ },
+ "group-b": {
+ "group-c",
+ },
+ "group-c": {
+ "group-d",
+ "detected_bar",
+ },
+ "group-d": {
+ "group-e",
+ },
+ "group-e": {
+ "detected_foo",
+ },
+ "group-f": {
+ "a",
+ "b",
+ },
+ },
+ []string{
+ "group-a",
+ "group-b",
+ "group-c",
+ "group-d",
+ "group-e",
+ },
+ },
+
+ {
+ "detected as transitive :remove member",
+ map[string][]string{
+ "group-a": {
+ "group-b",
+ },
+ "group-b": {
+ "group-c",
+ },
+ "group-c": {
+ "group-d",
+ },
+ "group-d": {
+ "group-e",
+ },
+ "group-e": {
+ "all",
+ },
+ "group-e:remove": {
+ "detected_foo",
+ },
+ "group-f": {
+ "a",
+ "b",
+ },
+ },
+ []string{
+ "group-a",
+ "group-b",
+ "group-c",
+ "group-d",
+ "group-e",
+ },
+ },
+ }
+
+ for _, tc := range tests {
+ t.Run(tc.name, func(t *testing.T) {
+ res := TransitivelyDetectedGroups(tc.groups)
+ testutil.AssertEqual(t, "res", res, tc.exp)
+ })
}
}