cp_test.go

package kube

import (
	"context"
	"os"
	"testing"

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

func TestCopy(t *testing.T) {
	var containerName string

	podName, containerName, podNamespace := getResourceNames(t)

	t.Run("should upload error", func(t *testing.T) {
		err := mockcli.Upload(context.TODO(), podName, containerName, podNamespace, "testdata/noexists.txt", "/testdata")
		require.ErrorContains(t, err, "doesn't exist in local filesystem")
	})

	t.Run("should upload file to sub dir error", func(t *testing.T) {
		err := mockcli.Upload(context.TODO(), podName, containerName, podNamespace, "testdata/upload.txt", "/testdata/sub/upload.txt")
		require.ErrorContains(t, err, "/testdata/sub: Cannot open: No such file or directory")
	})

	t.Run("upload file to dir", func(t *testing.T) {
		err := mockcli.Upload(context.TODO(), podName, containerName, podNamespace, "testdata/upload.txt", "/testdata")
		require.NoError(t, err)

		out, _, err := mockcli.Exec(podName, containerName, podNamespace, "ls", "/testdata")
		require.NoError(t, err)
		assert.Contains(t, out, "upload.txt")

	})

	t.Run("upload dir", func(t *testing.T) {
		err := mockcli.Upload(context.TODO(), podName, containerName, podNamespace, "testdata/uploaddir", "/testdata")
		require.NoError(t, err)

		out, _, err := mockcli.Exec(podName, containerName, podNamespace, "ls", "-l", "/testdata/testdata")
		require.NoError(t, err)
		assert.Contains(t, out, "uploaddir")
	})

	t.Run("should download err", func(t *testing.T) {
		err := mockcli.Download(context.TODO(), podName, containerName, podNamespace, "testdata/noexists.txt", "testdata")
		require.ErrorContains(t, err, " testdata/noexists.txt: Cannot stat: No such file or directory")
	})

	t.Run("download file", func(t *testing.T) {
		err := mockcli.Download(context.TODO(), podName, containerName, podNamespace, "testdata/upload.txt", "testdata/downdir")
		require.NoError(t, err)
		_, err = os.Stat("testdata/downdir/upload.txt")
		require.NoError(t, err)
	})

	t.Run("download dir", func(t *testing.T) {
		err := mockcli.Download(context.TODO(), podName, containerName, podNamespace, "testdata/testdata/uploaddir", "testdata/downdir")
		require.NoError(t, err)
		_, err = os.Stat("testdata/downdir/uploaddir")
		require.NoError(t, err)
		_, err = os.Stat("testdata/downdir/uploaddir/upload.txt")
		require.NoError(t, err)
		_, _, err = mockcli.Exec(podName, containerName, podNamespace, "rm", "-rf", "/testdata")
		require.NoError(t, err)
	})
}