iso_test.go

package xtractr_test

import (
	"io/fs"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/Unpackerr/iso9660"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"golift.io/xtractr"
)

func TestIso(t *testing.T) {
	t.Parallel()

	testFilesInfo := createTestFiles(t)
	writer, err := iso9660.NewWriter()
	require.NoError(t, err, "failed to create writer")

	defer func() {
		err = writer.Cleanup()
		require.NoError(t, err, "failed to cleanup writer")
	}()

	size := uint64(0)
	walkFS := os.DirFS(testFilesInfo.srcFilesDir)
	walkErr := fs.WalkDir(walkFS, ".", func(path string, entry fs.DirEntry, err error) error {
		require.NoError(t, err, "unexpected")

		if entry.IsDir() {
			return nil
		}

		if path == "." {
			return nil
		}

		fileToAdd, err := walkFS.Open(path)

		require.NoError(t, err, "failed to open file")
		defer safeCloser(t, fileToAdd)

		fStat, err := fileToAdd.Stat()
		require.NoError(t, err, "failed to stat file")

		size += uint64(fStat.Size())

		isoName := strings.TrimPrefix(filepath.ToSlash(path), "/")
		err = writer.AddFile(fileToAdd, isoName)
		require.NoError(t, err, "failed to add file")

		return nil
	})
	require.NoError(t, walkErr, "failed to walk files")

	isoFileName := filepath.Join(testFilesInfo.dstFilesDir, "archive.iso")

	isoFile, err := os.Create(isoFileName)

	require.NoError(t, err, "failed to create ISO file")
	defer safeCloser(t, isoFile)

	err = writer.WriteTo(isoFile, "test")
	require.NoError(t, err, "failed to write ISO")

	xSize, files, archives, err := xtractr.ExtractFile(&xtractr.XFile{
		FilePath:  isoFileName,
		OutputDir: filepath.Clean(testFilesInfo.dstFilesDir),
		FileMode:  0o600,
		DirMode:   0o700,
	})
	require.NoError(t, err)
	assert.Equal(t, testFilesInfo.dataSize, size, "data written does not match predetermined size")
	assert.Equal(t, testFilesInfo.dataSize, xSize, "data extracted does not match predetermined size")
	assert.Len(t, files, testFilesInfo.fileCount)
	assert.Len(t, archives, testFilesInfo.archiveCount)
}