added a test for wwctl node add

Signed-off-by: Christian Goll <cgoll@suse.de>

internal/pkg/node/constructors.go

@@ -16,6 +16,8 @@ import (
 var ConfigFile string
 var DefaultConfig string
 
+var cachedDB NodeYaml
+
 // used as fallback if DefaultConfig can't be read
 var FallBackConf = `---
 defaultnode:
@@ -44,9 +46,18 @@ func init() {
 	if DefaultConfig == "" {
 		DefaultConfig = path.Join(conf.Paths.Sysconfdir, "warewulf/defaults.conf")
 	}
+	cachedDB.current = false
+	cachedDB.persist = true
 }
 
+/*
+Creates a new nodeDb object from the actual configuration
+*/
 func New() (NodeYaml, error) {
+	if cachedDB.current {
+		wwlog.Debug("Returning cached object")
+		return cachedDB, nil
+	}
 	var ret NodeYaml
 
 	wwlog.Verbose("Opening node configuration file: %s", ConfigFile)
@@ -62,10 +73,40 @@ func New() (NodeYaml, error) {
 	}
 
 	wwlog.Debug("Returning node object")
-
+	cachedDB = ret
+	cachedDB.current = true
 	return ret, nil
 }
 
+/*
+Creates a database object from a given buffer, always create
+a new object, never return the cached one.
+*/
+func TestNew(buffer []byte) (db NodeYaml, err error) {
+	db.NodeProfiles = make(map[string]*NodeConf)
+	db.Nodes = make(map[string]*NodeConf)
+	err = yaml.Unmarshal(buffer, &db)
+	db.persist = false
+	cachedDB = db
+	cachedDB.current = true
+	wwlog.Debug("Created cached object")
+	return
+}
+
+func (config *NodeYaml) DBDump() (buffer []byte) {
+	for _, n := range config.Nodes {
+		n.Flatten()
+	}
+	for _, p := range config.NodeProfiles {
+		p.Flatten()
+	}
+	buffer, err := yaml.Marshal(config)
+	if err != nil {
+		wwlog.Warn("porblems on dumping nodedb: %s", err)
+	}
+	return
+}
+
 /*
 Get all the nodes of a configuration. This function also merges
 the nodes with the given profiles and set the default values

internal/pkg/node/datastructure.go

@@ -10,6 +10,8 @@ type NodeYaml struct {
 	WWInternal   int `yaml:"WW_INTERNAL"`
 	NodeProfiles map[string]*NodeConf
 	Nodes        map[string]*NodeConf
+	current      bool
+	persist      bool
 }
 
 /*

internal/pkg/node/modifiers.go

@@ -127,7 +127,10 @@ func (config *NodeYaml) Persist() error {
 	if err != nil {
 		return err
 	}
-
+	// Can't persist for unit tests
+	if !config.persist {
+		return err
+	}
 	file, err := os.OpenFile(ConfigFile, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
 	if err != nil {
 		wwlog.Error("%s", err)

internal/pkg/warewulfconf/constructors.go

@@ -30,8 +30,8 @@ func New() (conf ControllerConf) {
 		conf.Nfs = new(NfsConf)
 		conf.Paths = new(BuildConfig)
 		_ = defaults.Set(&conf)
-
 		cachedConf = conf
+		cachedConf.readConf = false
 		cachedConf.current = true
 
 	} else {
@@ -76,6 +76,7 @@ func (conf *ControllerConf) Read(data []byte) (err error) {
 	}
 	cachedConf = *conf
 	cachedConf.current = true
+	cachedConf.readConf = true
 	return
 }
 
@@ -148,3 +149,10 @@ func (conf *ControllerConf) SetDynamicDefaults() (err error) {
 	cachedConf.current = true
 	return
 }
+
+/*
+Return if configuration was read from disk
+*/
+func (conf *ControllerConf) Initialized() bool {
+	return conf.readConf
+}

internal/pkg/warewulfconf/datastructure.go

@@ -20,6 +20,7 @@ type ControllerConf struct {
 	MountsContainer []*MountEntry `yaml:"container mounts" default:"[{\"source\": \"/etc/resolv.conf\", \"dest\": \"/etc/resolv.conf\"}]"`
 	Paths           *BuildConfig  `yaml:"paths"`
 	current         bool
+	readConf        bool
 }
 
 type WarewulfConf struct {

internal/pkg/warewulfd/daemon.go

@@ -23,6 +23,23 @@ const (
 
 var loginit bool
 
+// allow to run without daemon for tests
+var nodaemon bool
+
+func init() {
+	nodaemon = false
+}
+
+// run without daemon
+func SetNoDaemon() {
+	nodaemon = true
+}
+
+// run with daemon
+func SetDaemon() {
+	nodaemon = false
+}
+
 func DaemonFormatter(logLevel int, rec *wwlog.LogRecord) string {
 	return "[" + rec.Time.Format(time.UnixDate) + "] " + wwlog.DefaultFormatter(logLevel, rec)
 }
@@ -66,6 +83,10 @@ func DaemonInitLogging() error {
 }
 
 func DaemonStart() error {
+	if nodaemon {
+		return nil
+	}
+
 	if os.Getenv("WAREWULFD_BACKGROUND") == "1" {
 		err := RunServer()
 		if err != nil {
@@ -116,6 +137,10 @@ func DaemonStart() error {
 }
 
 func DaemonStatus() error {
+	if nodaemon {
+		return nil
+	}
+
 	if !util.IsFile(WAREWULFD_PIDFILE) {
 		return errors.New("Warewulf server is not running")
 	}
@@ -142,6 +167,9 @@ func DaemonStatus() error {
 }
 
 func DaemonReload() error {
+	if nodaemon {
+		return nil
+	}
 	if !util.IsFile(WAREWULFD_PIDFILE) {
 		return errors.New("Warewulf server is not running")
 	}