Files
dingtalk-workspace-cli/test/unit/open_source_policy_test.go
T
2026-03-27 17:55:57 +08:00

80 lines
1.7 KiB
Go

package unit_test
import (
"fmt"
"io/fs"
"os"
"path/filepath"
"runtime"
"strings"
"testing"
)
func TestOpenSourceTreeOmitsEmbeddedHostMarkers(t *testing.T) {
_, filename, _, ok := runtime.Caller(0)
if !ok {
t.Fatal("runtime.Caller(0) failed")
}
root := filepath.Clean(filepath.Join(filepath.Dir(filename), "..", ".."))
forbidden := []string{
"DWS_" + "BUILD_MODE",
"com.dingtalk.scenario." + "wukong",
"WUKONG_" + "SKILLS_DIR",
"Embedded" + "Mode",
"CleanTokenOn" + "Expiry",
"HideAuth" + "LoginCommand",
"EnablePrivate" + "UtilityCommands",
"UseExecutable" + "ConfigDir",
"DeleteExeRelative" + "TokenOnAuthErr",
"WriteToken" + "Marker",
"Token" + "Marker",
"tokenJSON" + "File",
}
var matches []string
err := filepath.WalkDir(root, func(path string, d fs.DirEntry, walkErr error) error {
if walkErr != nil {
return walkErr
}
if d.IsDir() {
switch d.Name() {
case ".git", ".worktrees", "node_modules", "dist":
return filepath.SkipDir
}
return nil
}
if !isScannableSourceFile(path) {
return nil
}
content, err := os.ReadFile(path)
if err != nil {
return fmt.Errorf("read %s: %w", path, err)
}
for _, needle := range forbidden {
if strings.Contains(string(content), needle) {
rel, _ := filepath.Rel(root, path)
matches = append(matches, rel+": "+needle)
}
}
return nil
})
if err != nil {
t.Fatalf("WalkDir() error = %v", err)
}
if len(matches) > 0 {
t.Fatalf("found forbidden proprietary markers in OSS tree:\n%s", strings.Join(matches, "\n"))
}
}
func isScannableSourceFile(path string) bool {
switch filepath.Ext(path) {
case ".go", ".md", ".sh", ".ps1", ".yml", ".yaml", ".tmpl":
return true
default:
return filepath.Base(path) == "Makefile"
}
}