package d2vm import ( "context" "os" "path/filepath" "strings" "testing" "github.com/google/uuid" "github.com/sirupsen/logrus" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "go.linka.cloud/d2vm/pkg/docker" "go.linka.cloud/d2vm/pkg/exec" ) func testSysconfig(t *testing.T, ctx context.Context, img, sysconf, kernel, initrd string) { tmpPath := filepath.Join(os.TempDir(), "d2vm-tests", strings.NewReplacer(":", "-", ".", "-").Replace(img)) require.NoError(t, os.MkdirAll(tmpPath, 0755)) defer os.RemoveAll(tmpPath) logrus.Infof("inspecting image %s", img) r, err := FetchDockerImageOSRelease(ctx, img, tmpPath) require.NoError(t, err) defer docker.Remove(ctx, img) sys, err := sysconfig(r) require.NoError(t, err) assert.Equal(t, sysconf, sys) d, err := NewDockerfile(r, img, "root", "") require.NoError(t, err) logrus.Infof("docker image based on %s", d.Release.Name) p := filepath.Join(tmpPath, docker.FormatImgName(img)) dir := filepath.Dir(p) f, err := os.Create(p) require.NoError(t, err) defer f.Close() require.NoError(t, d.Render(f)) imgUUID := uuid.New().String() logrus.Infof("building kernel enabled image") require.NoError(t, docker.Build(ctx, imgUUID, p, dir)) defer docker.Remove(ctx, imgUUID) require.NoError(t, docker.RunAndRemove(ctx, imgUUID, "test", "-f", kernel)) require.NoError(t, docker.RunAndRemove(ctx, imgUUID, "test", "-f", initrd)) } func TestSyslinuxCfg(t *testing.T) { tests := []struct { image string kernel string initrd string sysconfig string }{ { image: "ubuntu:18.04", kernel: "/vmlinuz", initrd: "/initrd.img", sysconfig: syslinuxCfgDebian, }, { image: "ubuntu:20.04", kernel: "/boot/vmlinuz", initrd: "/boot/initrd.img", sysconfig: syslinuxCfgUbuntu, }, { image: "ubuntu:22.04", kernel: "/boot/vmlinuz", initrd: "/boot/initrd.img", sysconfig: syslinuxCfgUbuntu, }, { image: "ubuntu:latest", kernel: "/boot/vmlinuz", initrd: "/boot/initrd.img", sysconfig: syslinuxCfgUbuntu, }, { image: "debian:9", kernel: "/vmlinuz", initrd: "/initrd.img", sysconfig: syslinuxCfgDebian, }, { image: "debian:10", kernel: "/vmlinuz", initrd: "/initrd.img", sysconfig: syslinuxCfgDebian, }, { image: "debian:11", kernel: "/vmlinuz", initrd: "/initrd.img", sysconfig: syslinuxCfgDebian, }, { image: "debian:latest", kernel: "/vmlinuz", initrd: "/initrd.img", sysconfig: syslinuxCfgDebian, }, { image: "alpine", kernel: "/boot/vmlinuz-virt", initrd: "/boot/initramfs-virt", sysconfig: syslinuxCfgAlpine, }, { image: "centos:8", kernel: "/boot/vmlinuz", initrd: "/boot/initrd.img", sysconfig: syslinuxCfgCentOS, }, { image: "centos:latest", kernel: "/boot/vmlinuz", initrd: "/boot/initrd.img", sysconfig: syslinuxCfgCentOS, }, } ctx, cancel := context.WithCancel(context.Background()) defer cancel() exec.SetDebug(true) for _, test := range tests { t.Run(test.image, func(t *testing.T) { testSysconfig(t, ctx, test.image, test.sysconfig, test.kernel, test.initrd) }) } }