feat(tests): add tests for zip package

Signed-off-by: Alexander Sharov <kvendingoldo@yandex.ru>

Author: kvendingoldo

pkg/zip/zip_test.go

@@ -0,0 +1,144 @@
+package zip
+
+import (
+	"archive/zip"
+	"bytes"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestUnzipToDir(t *testing.T) {
+	// Create a temporary directory for testing
+	tempDir, err := os.MkdirTemp("", "zip-test-*")
+	require.NoError(t, err)
+	defer os.RemoveAll(tempDir)
+
+	// Create a test zip file in memory
+	buf := new(bytes.Buffer)
+	zipWriter := zip.NewWriter(buf)
+
+	// Add some test files to the zip
+	files := []struct {
+		name    string
+		content string
+	}{
+		{"test1.txt", "Hello World"},
+		{"subdir/test2.txt", "Test Content"},
+		{"subdir/nested/test3.txt", "Nested Content"},
+	}
+
+	for _, file := range files {
+		writer, err := zipWriter.Create(file.name)
+		require.NoError(t, err)
+		_, err = writer.Write([]byte(file.content))
+		require.NoError(t, err)
+	}
+
+	require.NoError(t, zipWriter.Close())
+
+	// Test cases
+	tests := []struct {
+		name     string
+		filter   func(string) bool
+		expected map[string]string
+	}{
+		{
+			name:   "no filter",
+			filter: func(string) bool { return true },
+			expected: map[string]string{
+				"test1.txt":               "Hello World",
+				"subdir/test2.txt":        "Test Content",
+				"subdir/nested/test3.txt": "Nested Content",
+			},
+		},
+		{
+			name:   "filter txt files",
+			filter: func(path string) bool { return filepath.Ext(path) == ".txt" },
+			expected: map[string]string{
+				"test1.txt":               "Hello World",
+				"subdir/test2.txt":        "Test Content",
+				"subdir/nested/test3.txt": "Nested Content",
+			},
+		},
+		{
+			name:   "filter specific file",
+			filter: func(path string) bool { return path == "test1.txt" },
+			expected: map[string]string{
+				"test1.txt": "Hello World",
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// Create a new temp dir for each test case
+			testDir, err := os.MkdirTemp(tempDir, "test-*")
+			require.NoError(t, err)
+
+			// Unzip the test data
+			err = UnzipToDir(buf.Bytes(), testDir, tt.filter)
+			require.NoError(t, err)
+
+			// Verify the extracted files
+			for path, expectedContent := range tt.expected {
+				fullPath := filepath.Join(testDir, path)
+				content, err := os.ReadFile(fullPath)
+				require.NoError(t, err)
+				assert.Equal(t, expectedContent, string(content))
+			}
+		})
+	}
+}
+
+func TestSanitizeArchivePath(t *testing.T) {
+	tests := []struct {
+		name     string
+		dirPath  string
+		fileName string
+		want     string
+		wantErr  bool
+	}{
+		{
+			name:     "valid path",
+			dirPath:  "/tmp/test",
+			fileName: "file.txt",
+			want:     "/tmp/test/file.txt",
+			wantErr:  false,
+		},
+		{
+			name:     "valid nested path",
+			dirPath:  "/tmp/test",
+			fileName: "subdir/file.txt",
+			want:     "/tmp/test/subdir/file.txt",
+			wantErr:  false,
+		},
+		{
+			name:     "path traversal attempt",
+			dirPath:  "/tmp/test",
+			fileName: "../file.txt",
+			wantErr:  true,
+		},
+		{
+			name:     "absolute path attempt",
+			dirPath:  "/tmp/test",
+			fileName: "/etc/passwd",
+			wantErr:  true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := sanitizeArchivePath(tt.dirPath, tt.fileName)
+			if tt.wantErr {
+				assert.Error(t, err)
+				return
+			}
+			assert.NoError(t, err)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}