mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-10-11 12:18:59 +00:00
viettranx
5623f97961
- Add extension_whitelist.go to deterministically skip binary .bin/.exe/.dll - Add media_summary.go: synthesize docType + generate short summary from vault fields - Update rescan.go: detect document type for PDF/office files + propagate docType - Update safe_walk.go test: .bin now filtered by whitelist, use .txt instead - Update rescan_test.go: add document docType coverage - Add link_types.go: define AUTO_LINK_TASK, AUTO_LINK_DELEGATION link types
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package vault
|
|
|
|
import "testing"
|
|
|
|
func TestIsIncludedExtension(t *testing.T) {
|
|
cases := []struct {
|
|
ext string
|
|
include bool
|
|
want string
|
|
}{
|
|
// Text/code → note
|
|
{".md", true, "note"},
|
|
{".go", true, "note"},
|
|
{".ts", true, "note"},
|
|
{".tsx", true, "note"},
|
|
{".py", true, "note"},
|
|
{".sql", true, "note"},
|
|
// Images → media
|
|
{".png", true, "media"},
|
|
{".jpg", true, "media"},
|
|
{".webp", true, "media"},
|
|
// Video → media
|
|
{".mp4", true, "media"},
|
|
{".mkv", true, "media"},
|
|
// Audio → media
|
|
{".mp3", true, "media"},
|
|
{".wav", true, "media"},
|
|
// Office → document
|
|
{".pdf", true, "document"},
|
|
{".docx", true, "document"},
|
|
{".xlsx", true, "document"},
|
|
{".pptx", true, "document"},
|
|
// Binary / archive / exe → excluded
|
|
{".exe", false, ""},
|
|
{".bin", false, ""},
|
|
{".dll", false, ""},
|
|
{".zip", false, ""},
|
|
{".gz", false, ""}, // filepath.Ext returns only last segment — ".tar.gz" → ".gz"
|
|
{".dmg", false, ""},
|
|
{".iso", false, ""},
|
|
{".so", false, ""},
|
|
{".dylib", false, ""},
|
|
// Empty extension
|
|
{"", false, ""},
|
|
// Unknown
|
|
{".xyz", false, ""},
|
|
}
|
|
for _, c := range cases {
|
|
gotInclude, gotDocType := isIncludedExtension(c.ext)
|
|
if gotInclude != c.include || gotDocType != c.want {
|
|
t.Errorf("isIncludedExtension(%q) = (%v, %q); want (%v, %q)",
|
|
c.ext, gotInclude, gotDocType, c.include, c.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestExtensionWhitelistMapCoverage(t *testing.T) {
|
|
// Sanity: ensure the map has expected canonical entries.
|
|
required := []string{".md", ".png", ".pdf", ".docx", ".mp4", ".mp3"}
|
|
for _, ext := range required {
|
|
if _, ok := extensionDocType[ext]; !ok {
|
|
t.Errorf("extensionDocType missing required extension %q", ext)
|
|
}
|
|
}
|
|
}
|