package clean

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/warewulf/warewulf/internal/pkg/testenv"
)

func Test_Clean(t *testing.T) {
	env := testenv.New(t)
	env.WriteFile("etc/warewulf/nodes.conf",
		`nodeprofiles: {}
nodes:
  node1: {}
`)
	env.WriteFile("srv/warewulf/overlays/node1/__SYSTEM__.img", "Fake System")
	env.WriteFile("srv/warewulf/overlays/node2/__SYSTEM__.img", "Fake System")
	env.WriteFile("var/cache/warewulf/test", "Nothing to see here")
	baseCmd := GetCommand()
	err := baseCmd.Execute()
	assert.NoError(t, err)
	assert.FileExists(t, env.GetPath("srv/warewulf/overlays/node1/__SYSTEM__.img"))
	assert.NoFileExists(t, env.GetPath("srv/warewulf/overlays/node2/__SYSTEM__.img"))
	assert.NoFileExists(t, env.GetPath("/var/cache/warewulf/test"))
}