ignitions_test.go

package neco

import (
	"fmt"
	"os"
	"path/filepath"
	"reflect"
	"regexp"
	"sort"
	"strings"
	"testing"

	sabakan "github.com/cybozu-go/sabakan/v3/client"
	"github.com/google/go-cmp/cmp"
	"sigs.k8s.io/yaml"
)

const (
	testRoleDir = "ignitions/roles"
)

func checkForFlag(filestateInDir map[string]bool) error {
	for key, value := range filestateInDir {
		if !value {
			return fmt.Errorf("%s file is not included in the configuration file\n", key)
		}
	}
	return nil
}

func testIgnitionTemplates(path string, filestateInDir map[string]bool) error {
	data, err := os.ReadFile(path)
	if err != nil {
		return err
	}

	t := &sabakan.TemplateSource{}
	err = yaml.Unmarshal(data, t)
	if err != nil {
		return err
	}

	if len(t.Include) != 0 {
		abs, err := filepath.Abs(filepath.Join(filepath.Dir(path), t.Include))
		if err != nil {
			return err
		}

		_, err = os.Stat(abs)
		if err != nil {
			return err
		}
		err = testIgnitionTemplates(abs, filestateInDir)
		if err != nil {
			return err
		}
	}

	if len(t.Passwd) != 0 {
		abs, err := filepath.Abs(filepath.Join(filepath.Dir(path), t.Passwd))
		if err != nil {
			return err
		}

		_, err = os.Stat(abs)
		if err != nil {
			return err
		}
	}

	if len(t.Files) != 0 {
		filesDir, err := filepath.Abs(filepath.Join(filepath.Dir(path), "files"))
		if err != nil {
			return err
		}

		var filelistInYAML []string
		for _, f := range t.Files {
			realPath := filepath.Join(filesDir, f)
			filelistInYAML = append(filelistInYAML, realPath)
		}
		sort.Strings(filelistInYAML)

		err = filepath.Walk(filesDir, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return err
			}
			if info.IsDir() {
				return nil
			}
			if _, ok := filestateInDir[path]; !ok {
				filestateInDir[path] = false
			}
			return nil
		})

		if err != nil {
			return err
		}

		for _, f := range filelistInYAML {
			if _, ok := filestateInDir[f]; ok {
				filestateInDir[f] = true
			} else {
				return fmt.Errorf("file in %s and file tree differ\n", f)
			}
		}
	}

	// remote_files does not use in Neco.
	//if len(t.RemoteFiles) != 0 {
	//	...
	//}

	if len(t.Systemd) != 0 {
		systemdDir, err := filepath.Abs(filepath.Join(filepath.Dir(path), "systemd"))
		if err != nil {
			return err
		}

		err = filepath.Walk(systemdDir, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return err
			}
			if info.IsDir() {
				return nil
			}
			if _, ok := filestateInDir[path]; !ok {
				filestateInDir[path] = false
			}
			return nil
		})
		if err != nil {
			return err
		}

		var filelistInYAML []string
		for _, f := range t.Systemd {
			realPath := filepath.Join(systemdDir, f.Name)
			filelistInYAML = append(filelistInYAML, realPath)
		}
		sort.Strings(filelistInYAML)
		for _, s := range filelistInYAML {
			if _, ok := filestateInDir[s]; ok {
				filestateInDir[s] = true
			} else {
				fmt.Printf("%s is pre-installed service.\n", s)
			}
		}
	}

	if len(t.Networkd) != 0 {
		networkdDir, err := filepath.Abs(filepath.Join(filepath.Dir(path), "networkd"))
		if err != nil {
			return err
		}

		var filelistInYAML []string
		for _, f := range t.Networkd {
			realPath := filepath.Join(networkdDir, f)
			filelistInYAML = append(filelistInYAML, realPath)
		}
		sort.Strings(filelistInYAML)

		var filelistInDir []string
		err = filepath.Walk(networkdDir, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return err
			}
			if info.IsDir() {
				return nil
			}
			filelistInDir = append(filelistInDir, path)
			return nil
		})
		if err != nil {
			return err
		}

		sort.Strings(filelistInDir)

		if !reflect.DeepEqual(filelistInYAML, filelistInDir) {
			return fmt.Errorf("files in %s and file tree %s differ\n%v", path, networkdDir, cmp.Diff(filelistInYAML, filelistInDir))
		}
	}

	return nil
}

func TestNecoIgnitionTemplates(t *testing.T) {
	siteYAMLsByRole := make(map[string][]string)

	t.Parallel()
	err := filepath.Walk(testRoleDir, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		if info.IsDir() {
			return nil
		}
		if s, _ := regexp.MatchString(`site(-.*|)\.yml`, info.Name()); s {
			role := strings.Split(path, "/")[2]
			siteYAMLsByRole[role] = append(siteYAMLsByRole[role], path)
		}
		return nil
	})
	if err != nil {
		t.Fatal(err)
	}

	for _, siteYAMLs := range siteYAMLsByRole {
		filestateInDir := make(map[string]bool)
		for _, sy := range siteYAMLs {
			err := testIgnitionTemplates(sy, filestateInDir)
			if err != nil {
				t.Fatal(err)
			}
		}
		if err := checkForFlag(filestateInDir); err != nil {
			t.Fatal(err)
		}
	}
}