mirror of
https://github.com/GopeedLab/gopeed.git
synced 2026-11-04 15:49:49 +00:00
118 lines
3.5 KiB
Go
118 lines
3.5 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
package download
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/GopeedLab/gopeed/pkg/base"
|
|
)
|
|
|
|
func TestScript_TriggerOnDone_MoveFile(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
downloadDir := filepath.Join(tmpDir, "downloads")
|
|
destDir := filepath.Join(tmpDir, "moved")
|
|
task, taskPath := createDownloadDoneTask(t, downloadDir, "test.txt")
|
|
srcPath := filepath.FromSlash(taskPath)
|
|
destFile := filepath.Join(destDir, "test.txt")
|
|
|
|
scriptPath := getTestScriptPath(t, "move.sh")
|
|
ensureScriptExecutable(t, scriptPath)
|
|
|
|
setupScriptTest(t, func(downloader *Downloader) {
|
|
cfg, _ := downloader.GetConfig()
|
|
cfg.Script = &base.ScriptConfig{
|
|
Enable: true,
|
|
Paths: []string{scriptPath},
|
|
}
|
|
downloader.PutConfig(cfg)
|
|
|
|
t.Setenv("GOPEED_TEST_DEST_DIR", destDir)
|
|
downloader.triggerScripts(ScriptEventDownloadDone, task, nil)
|
|
|
|
waitForFile(t, destFile, 3*time.Second)
|
|
if _, err := os.Stat(srcPath); !os.IsNotExist(err) {
|
|
t.Errorf("Expected source file to be moved, but it still exists: %s", srcPath)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestScript_MultipleScripts(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
outputFile1 := filepath.Join(tmpDir, "output1.txt")
|
|
outputFile2 := filepath.Join(tmpDir, "output2.txt")
|
|
scriptPath1 := getTestScriptPath(t, "write_output1.sh")
|
|
ensureScriptExecutable(t, scriptPath1)
|
|
scriptPath2 := getTestScriptPath(t, "write_output2.sh")
|
|
ensureScriptExecutable(t, scriptPath2)
|
|
|
|
setupScriptTest(t, func(downloader *Downloader) {
|
|
cfg, _ := downloader.GetConfig()
|
|
cfg.Script = &base.ScriptConfig{
|
|
Enable: true,
|
|
Paths: []string{scriptPath1, scriptPath2},
|
|
}
|
|
downloader.PutConfig(cfg)
|
|
|
|
downloadDir := filepath.Join(tmpDir, "downloads")
|
|
task, _ := createDownloadDoneTask(t, downloadDir, "multi.txt")
|
|
|
|
t.Setenv("GOPEED_TEST_OUTPUT_FILE_1", outputFile1)
|
|
t.Setenv("GOPEED_TEST_OUTPUT_FILE_2", outputFile2)
|
|
downloader.triggerScripts(ScriptEventDownloadDone, task, nil)
|
|
|
|
waitForFile(t, outputFile1, 3*time.Second)
|
|
waitForFile(t, outputFile2, 3*time.Second)
|
|
})
|
|
}
|
|
|
|
func TestScript_EnvironmentVariables(t *testing.T) {
|
|
tmpDir := t.TempDir()
|
|
scriptPath := getTestScriptPath(t, "env_dump.sh")
|
|
ensureScriptExecutable(t, scriptPath)
|
|
outputFile := filepath.Join(tmpDir, "env_output.txt")
|
|
|
|
setupScriptTest(t, func(downloader *Downloader) {
|
|
cfg, _ := downloader.GetConfig()
|
|
cfg.Script = &base.ScriptConfig{
|
|
Enable: true,
|
|
Paths: []string{scriptPath},
|
|
}
|
|
downloader.PutConfig(cfg)
|
|
|
|
downloadDir := filepath.Join(tmpDir, "downloads")
|
|
task, taskPath := createDownloadDoneTask(t, downloadDir, "env.txt")
|
|
|
|
t.Setenv("GOPEED_TEST_OUTPUT_FILE", outputFile)
|
|
downloader.triggerScripts(ScriptEventDownloadDone, task, nil)
|
|
|
|
waitForFile(t, outputFile, 3*time.Second)
|
|
content, err := os.ReadFile(outputFile)
|
|
if err != nil {
|
|
t.Fatalf("Failed to read output file: %v", err)
|
|
}
|
|
|
|
output := string(content)
|
|
if !strings.Contains(output, "GOPEED_EVENT=DOWNLOAD_DONE") {
|
|
t.Errorf("Expected GOPEED_EVENT in output, got: %s", output)
|
|
}
|
|
if !strings.Contains(output, "GOPEED_TASK_ID="+task.ID) {
|
|
t.Errorf("Expected GOPEED_TASK_ID in output, got: %s", output)
|
|
}
|
|
if !strings.Contains(output, "GOPEED_TASK_NAME="+task.Name()) {
|
|
t.Errorf("Expected GOPEED_TASK_NAME in output, got: %s", output)
|
|
}
|
|
if !strings.Contains(output, "GOPEED_TASK_STATUS="+string(task.Status)) {
|
|
t.Errorf("Expected GOPEED_TASK_STATUS in output, got: %s", output)
|
|
}
|
|
if !strings.Contains(output, "GOPEED_TASK_PATH="+taskPath) {
|
|
t.Errorf("Expected GOPEED_TASK_PATH in output, got: %s", output)
|
|
}
|
|
})
|
|
}
|