Ant-Browser/backend/internal/automation/script_template_test.go

package automation

import (
	"bytes"
	"path/filepath"
	"testing"
)

func TestMarshalScriptTemplateRoundTripsAdditionalFiles(t *testing.T) {
	templateData, err := MarshalScriptTemplate(ImportedBundle{
		Record: ScriptRecord{
			ID:           "template-roundtrip",
			Name:         "模板导出",
			Description:  "包含额外文件",
			Type:         "playwright-cdp",
			Status:       "ready",
			EntryFile:    "scripts/index.cjs",
			SelectorText: `{"code":"DEMO_TEMPLATE"}`,
			ParamsText:   `{"url":"https://example.com"}`,
			ScriptText:   "const helper = require('./helpers/helper.cjs')\nmodule.exports.run = async () => helper.run()",
		},
		Files: []ImportedBundleFile{
			{
				Path:    "scripts/index.cjs",
				Content: []byte("const helper = require('./helpers/helper.cjs')\nmodule.exports.run = async () => helper.run()"),
			},
			{
				Path:    "scripts/helpers/helper.cjs",
				Content: []byte("module.exports.run = async () => ({ ok: true })"),
			},
			{
				Path:    "manifest.json",
				Content: []byte(`{"custom":true}`),
			},
			{
				Path:    "assets/raw.bin",
				Content: []byte{0x00, 0x01, 0x02, 0xff},
			},
		},
	})
	if err != nil {
		t.Fatalf("MarshalScriptTemplate returned error: %v", err)
	}

	imported, err := ImportBundleFromBytes("template.json", templateData, "文本导入")
	if err != nil {
		t.Fatalf("ImportBundleFromBytes returned error: %v", err)
	}

	if imported.Record.EntryFile != "scripts/index.cjs" {
		t.Fatalf("unexpected entry file: %s", imported.Record.EntryFile)
	}
	if !bytes.Equal([]byte(imported.Record.ScriptText), []byte("const helper = require('./helpers/helper.cjs')\nmodule.exports.run = async () => helper.run()")) {
		t.Fatalf("unexpected script text: %q", imported.Record.ScriptText)
	}
	if !hasBundleFile(imported.Files, "scripts/helpers/helper.cjs", []byte("module.exports.run = async () => ({ ok: true })")) {
		t.Fatalf("expected helper file to round-trip, got %+v", imported.Files)
	}
	if !hasBundleFile(imported.Files, "manifest.json", []byte(`{"custom":true}`)) {
		t.Fatalf("expected manifest.json to round-trip as a regular file, got %+v", imported.Files)
	}
	if !hasBundleFile(imported.Files, "assets/raw.bin", []byte{0x00, 0x01, 0x02, 0xff}) {
		t.Fatalf("expected binary file to round-trip, got %+v", imported.Files)
	}
}

func TestScriptStoreExportBundleIncludesNestedFiles(t *testing.T) {
	store := NewScriptStore(filepath.Join(t.TempDir(), "data", "automation", "scripts"))

	if _, err := store.ImportBundle(ImportedBundle{
		Record: ScriptRecord{
			ID:         "export-bundle",
			Name:       "导出 bundle",
			Type:       "playwright-cdp",
			Status:     "ready",
			EntryFile:  "scripts/index.cjs",
			ScriptText: "const helper = require('./helpers/helper.cjs')\nmodule.exports.run = async () => helper.run()",
		},
		Files: []ImportedBundleFile{
			{
				Path:    "scripts/index.cjs",
				Content: []byte("const helper = require('./helpers/helper.cjs')\nmodule.exports.run = async () => helper.run()"),
			},
			{
				Path:    "scripts/helpers/helper.cjs",
				Content: []byte("module.exports.run = async () => ({ ok: true })"),
			},
		},
	}); err != nil {
		t.Fatalf("ImportBundle returned error: %v", err)
	}

	exported, err := store.ExportBundle("export-bundle")
	if err != nil {
		t.Fatalf("ExportBundle returned error: %v", err)
	}

	if exported.Record.ID != "export-bundle" {
		t.Fatalf("unexpected exported record: %+v", exported.Record)
	}
	if !hasBundleFile(exported.Files, "scripts/index.cjs", []byte("const helper = require('./helpers/helper.cjs')\nmodule.exports.run = async () => helper.run()")) {
		t.Fatalf("expected entry file in exported bundle, got %+v", exported.Files)
	}
	if !hasBundleFile(exported.Files, "scripts/helpers/helper.cjs", []byte("module.exports.run = async () => ({ ok: true })")) {
		t.Fatalf("expected helper file in exported bundle, got %+v", exported.Files)
	}
}

func hasBundleFile(files []ImportedBundleFile, targetPath string, expectedContent []byte) bool {
	for _, file := range files {
		if file.Path == targetPath && bytes.Equal(file.Content, expectedContent) {
			return true
		}
	}
	return false
}