d2vm/docker_image_test.go

package d2vm

import (
	"bytes"
	"context"
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"go.linka.cloud/d2vm/pkg/docker"
	"go.linka.cloud/d2vm/pkg/exec"
)

func TestDockerImageAsRunSript(t *testing.T) {
	tests := []struct {
		name  string
		image DockerImage
		want  string
	}{
		{
			name: "nothing",
			image: DockerImage{
				DockerImageConfig: DockerImageConfig{
					User:       "",
					WorkingDir: "",
					Env:        nil,
					Entrypoint: nil,
					Cmd:        nil,
				},
			},
			want: `
#!/bin/sh




`,
		},
		{
			name: "tail -f /dev/null",
			image: DockerImage{
				DockerImageConfig: DockerImageConfig{
					User: "root",
					Cmd:  []string{"tail", "-f", "/dev/null"},
				},
			},
			want: `
#!/bin/sh



su root -p -s /bin/sh -c '"tail" "-f" "/dev/null"'
`,
		},
		{
			name: "tail -f /dev/null inside home",
			image: DockerImage{
				DockerImageConfig: DockerImageConfig{
					User:       "root",
					WorkingDir: "/root",
					Cmd:        []string{"tail", "-f", "/dev/null"},
				},
			},
			want: `
#!/bin/sh

cd /root

su root -p -s /bin/sh -c '"tail" "-f" "/dev/null"'
`,
		},
		{
			name: "subshell tail -f /dev/null",
			image: DockerImage{
				DockerImageConfig: DockerImageConfig{
					User:       "root",
					Entrypoint: []string{"/bin/sh", "-c"},
					Cmd:        []string{"tail -f /dev/null"},
				},
			},
			want: `
#!/bin/sh



su root -p -s /bin/sh -c '"/bin/sh" "-c" "tail -f /dev/null"'
`,
		},
		{
			name: "www-data with env",
			image: DockerImage{
				DockerImageConfig: DockerImageConfig{
					User: "www-data",
					Cmd:  []string{"tail", "-f", "/dev/null"},
					Env:  []string{"ENV=PROD", "DB=mysql://user:password@localhost"},
				},
			},
			want: `
#!/bin/sh
export ENV=PROD
export DB=mysql://user:password@localhost



su www-data -p -s /bin/sh -c '"tail" "-f" "/dev/null"'
`,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var w bytes.Buffer
			require.NoError(t, tt.image.AsRunScript(&w))
			assert.Equal(t, tt.want, w.String())
		})
	}
}

func TestImageFlatten(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	const (
		img        = "d2vm-flatten-test"
		dockerfile = `FROM alpine

COPY resolv.conf /etc/
COPY hostname /etc/

RUN rm -rf /etc/apk
`
	)
	exec.SetDebug(true)
	tmp := filepath.Join(os.TempDir(), "d2vm-tests", "image-flatten")
	require.NoError(t, os.MkdirAll(tmp, perm))
	defer os.RemoveAll(tmp)

	require.NoError(t, os.WriteFile(filepath.Join(tmp, "hostname"), []byte("d2vm-flatten-test"), perm))
	require.NoError(t, os.WriteFile(filepath.Join(tmp, "resolv.conf"), []byte("nameserver 8.8.8.8"), perm))
	require.NoError(t, os.WriteFile(filepath.Join(tmp, "Dockerfile"), []byte(dockerfile), perm))
	require.NoError(t, docker.Build(ctx, img, "", tmp))
	defer docker.Remove(ctx, img)

	imgTmp := filepath.Join(tmp, "image")

	i, err := NewImage(ctx, img, imgTmp)
	require.NoError(t, err)

	rootfs := filepath.Join(tmp, "rootfs")
	require.NoError(t, i.Flatten(ctx, rootfs))

	b, err := os.ReadFile(filepath.Join(rootfs, "etc", "resolv.conf"))
	require.NoError(t, err)
	assert.Equal(t, "nameserver 8.8.8.8", string(b))

	b, err = os.ReadFile(filepath.Join(rootfs, "etc", "hostname"))
	require.NoError(t, err)
	assert.Equal(t, "d2vm-flatten-test", string(b))

	_, err = os.Stat(filepath.Join(rootfs, "etc", "apk"))
	assert.Error(t, err)

	require.NoError(t, i.Close())
	_, err = os.Stat(imgTmp)
	assert.Error(t, err)
}