181 lines
4.8 KiB
Go
181 lines
4.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
// "backup/cmd"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
// Вспомогательная функция для создания временной структуры каталогов
|
|
func setupTestDir(t *testing.T) (string, func()) {
|
|
dir, err := os.MkdirTemp("", "backup_test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Создаем тестовую структуру:
|
|
// /test_dir
|
|
// /subdir
|
|
// file2.txt
|
|
// file1.txt
|
|
// .hidden
|
|
// exclude_me.txt
|
|
subDir := filepath.Join(dir, "subdir")
|
|
if err := os.Mkdir(subDir, 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
files := []string{
|
|
"file1.txt",
|
|
".hidden",
|
|
"exclude_me.txt",
|
|
filepath.Join("subdir", "file2.txt"),
|
|
}
|
|
|
|
for _, f := range files {
|
|
path := filepath.Join(dir, f)
|
|
if err := os.WriteFile(path, []byte("test"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
return dir, func() { os.RemoveAll(dir) }
|
|
}
|
|
|
|
func TestGetPathSize(t *testing.T) {
|
|
t.Run("Directory size calculation", func(t *testing.T) {
|
|
dir, cleanup := setupTestDir(t)
|
|
defer cleanup()
|
|
|
|
size, err := getPathSize(dir, nil)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
// Ожидаемый размер: 4 файла по 4 байта каждый = 16 байт
|
|
if size != 16 {
|
|
t.Errorf("Expected size 16, got %d", size)
|
|
}
|
|
})
|
|
|
|
t.Run("Exclusion patterns", func(t *testing.T) {
|
|
dir, cleanup := setupTestDir(t)
|
|
defer cleanup()
|
|
|
|
exclude := []string{"exclude_me.txt", "*.hidden", "subdir/*"}
|
|
size, err := getPathSize(dir, exclude)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
// Должен остаться только file1.txt (4 байта)
|
|
if size != 4 {
|
|
t.Errorf("Expected size 4 after exclusions, got %d", size)
|
|
}
|
|
})
|
|
|
|
t.Run("Single file size", func(t *testing.T) {
|
|
dir, cleanup := setupTestDir(t)
|
|
defer cleanup()
|
|
file := filepath.Join(dir, "file1.txt")
|
|
|
|
size, err := getPathSize(file, nil)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
if size != 4 {
|
|
t.Errorf("Expected size 4, got %d", size)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestShouldExcludePath(t *testing.T) {
|
|
tests := []struct {
|
|
path string
|
|
patterns []string
|
|
expected bool
|
|
}{
|
|
// Точное совпадение
|
|
{"exclude.txt", []string{"exclude.txt"}, true},
|
|
{"dir/exclude.txt", []string{"exclude.txt"}, true},
|
|
|
|
// Префикс каталога
|
|
{"node_modules/file.js", []string{"node_modules"}, true},
|
|
{"src/node_modules/file.js", []string{"node_modules"}, false}, // Изменено ожидание
|
|
{"src/node_modules", []string{"node_modules"}, true},
|
|
|
|
// Glob-паттерны
|
|
{"temp.jpg", []string{"*.jpg"}, true},
|
|
{"backup.tmp", []string{"*.tmp"}, true},
|
|
{"image.png", []string{"*.jpg", "*.png"}, true},
|
|
{"document.pdf", []string{"*.doc"}, false},
|
|
|
|
// Базовое имя
|
|
{"deeply/nested/.env", []string{".env"}, true},
|
|
{"config/.env", []string{".*"}, true},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
result := shouldExcludePath(test.path, test.patterns)
|
|
if result != test.expected {
|
|
t.Errorf("Test %d: For path '%s' and patterns %v expected %t, got %t",
|
|
i+1, test.path, test.patterns, test.expected, result)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTargetNaming(t *testing.T) {
|
|
tests := []struct {
|
|
source string
|
|
target string
|
|
isDirectory bool
|
|
expected string
|
|
}{
|
|
{source: "test_dir", target: "backup", isDirectory: true, expected: "backup.tar.gz"},
|
|
{source: "test_file.txt", target: "backup_file", isDirectory: false, expected: "backup_file.gz"},
|
|
{source: "test_dir", target: "archive.tar.gz", isDirectory: true, expected: "archive.tar.gz"},
|
|
{source: "test_file.txt", target: "backup.gz", isDirectory: false, expected: "backup.gz"},
|
|
{
|
|
source: "test_dir",
|
|
target: "server:8000/test_dir",
|
|
isDirectory: true,
|
|
expected: "http://server:8000/backup/test_dir.tar.gz",
|
|
},
|
|
{
|
|
source: "test_file.txt",
|
|
target: "server:8000/file",
|
|
isDirectory: false,
|
|
expected: "http://server:8000/backup/file.gz",
|
|
},
|
|
{source: "test_dir", target: "192.168.1.100:8080", isDirectory: true, expected: "192.168.1.100:8080"},
|
|
{source: "test_file.txt", target: "10.0.0.5:9000", isDirectory: false, expected: "10.0.0.5:9000"},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
// Создаем временную структуру для теста
|
|
var tempPath string
|
|
if test.isDirectory {
|
|
tempPath, _ = os.MkdirTemp("", "test_dir")
|
|
defer os.RemoveAll(tempPath)
|
|
} else {
|
|
tempFile, _ := os.CreateTemp("", "test_file.txt")
|
|
tempPath = tempFile.Name()
|
|
tempFile.Close()
|
|
defer os.Remove(tempPath)
|
|
}
|
|
|
|
model, err := initialModel(tempPath, test.target, nil)
|
|
if err != nil {
|
|
t.Fatalf("Error creating model: %v", err)
|
|
}
|
|
|
|
actual := model.target
|
|
if actual != test.expected {
|
|
t.Errorf("For source '%s' (dir:%t) and target '%s' expected '%s', got '%s'",
|
|
test.source, test.isDirectory, test.target, test.expected, actual)
|
|
}
|
|
}
|
|
}
|