81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package generators
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/lxc/distrobuilder/image"
|
|
"github.com/lxc/distrobuilder/shared"
|
|
)
|
|
|
|
func TestHostnameGeneratorRunLXC(t *testing.T) {
|
|
cacheDir := filepath.Join(os.TempDir(), "distrobuilder-test")
|
|
rootfsDir := filepath.Join(cacheDir, "rootfs")
|
|
|
|
setup(t, cacheDir)
|
|
defer teardown(cacheDir)
|
|
|
|
generator := Get("hostname")
|
|
require.Equal(t, HostnameGenerator{}, generator)
|
|
|
|
definition := shared.Definition{
|
|
Image: shared.DefinitionImage{
|
|
Distribution: "ubuntu",
|
|
Release: "artful",
|
|
},
|
|
}
|
|
|
|
image := image.NewLXCImage(cacheDir, "", cacheDir, definition)
|
|
|
|
err := os.MkdirAll(filepath.Join(cacheDir, "rootfs", "etc"), 0755)
|
|
require.NoError(t, err)
|
|
|
|
createTestFile(t, filepath.Join(cacheDir, "rootfs", "etc", "hostname"), "hostname")
|
|
|
|
err = generator.RunLXC(cacheDir, rootfsDir, image,
|
|
shared.DefinitionFile{Path: "/etc/hostname"})
|
|
require.NoError(t, err)
|
|
|
|
validateTestFile(t, filepath.Join(cacheDir, "tmp", "etc", "hostname"), "hostname")
|
|
validateTestFile(t, filepath.Join(cacheDir, "rootfs", "etc", "hostname"), "LXC_NAME\n")
|
|
|
|
err = RestoreFiles(cacheDir, rootfsDir)
|
|
require.NoError(t, err)
|
|
|
|
validateTestFile(t, filepath.Join(cacheDir, "rootfs", "etc", "hostname"), "hostname")
|
|
}
|
|
|
|
func TestHostnameGeneratorRunLXD(t *testing.T) {
|
|
cacheDir := filepath.Join(os.TempDir(), "distrobuilder-test")
|
|
rootfsDir := filepath.Join(cacheDir, "rootfs")
|
|
|
|
setup(t, cacheDir)
|
|
defer teardown(cacheDir)
|
|
|
|
generator := Get("hostname")
|
|
require.Equal(t, HostnameGenerator{}, generator)
|
|
|
|
definition := shared.Definition{
|
|
Image: shared.DefinitionImage{
|
|
Distribution: "ubuntu",
|
|
Release: "artful",
|
|
},
|
|
}
|
|
|
|
image := image.NewLXDImage(cacheDir, "", cacheDir, definition)
|
|
|
|
err := os.MkdirAll(filepath.Join(cacheDir, "rootfs", "etc"), 0755)
|
|
require.NoError(t, err)
|
|
|
|
createTestFile(t, filepath.Join(cacheDir, "rootfs", "etc", "hostname"), "hostname")
|
|
|
|
err = generator.RunLXD(cacheDir, rootfsDir, image,
|
|
shared.DefinitionFile{Path: "/etc/hostname"})
|
|
require.NoError(t, err)
|
|
|
|
validateTestFile(t, filepath.Join(cacheDir, "templates", "hostname.tpl"), "{{ container.name }}\n")
|
|
}
|