44af6ba759
- Introduce new MCP and Robot command groups in the CLI, allowing for better organization of package management commands. - Add corresponding command descriptions for MCP and Robot functionalities to improve user guidance. - Update the agent command group to include additional commands for enhanced agent management. - Modify .gitignore to exclude specific design markdown files from version control. - Add environment variable for test application path in GitHub workflows to streamline testing setup.
87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package common
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestHashFile(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, "test.txt")
|
|
if err := os.WriteFile(path, []byte("hello world"), 0644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
hash, err := HashFile(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !strings.HasPrefix(hash, "sha256-") {
|
|
t.Errorf("expected sha256- prefix, got %q", hash)
|
|
}
|
|
if len(hash) != 7+64 { // "sha256-" + 64 hex chars
|
|
t.Errorf("unexpected hash length: %d", len(hash))
|
|
}
|
|
|
|
// Same content should produce same hash
|
|
hash2, _ := HashFile(path)
|
|
if hash != hash2 {
|
|
t.Error("same file should produce same hash")
|
|
}
|
|
|
|
// Non-existent file
|
|
_, err = HashFile(filepath.Join(dir, "nope.txt"))
|
|
if err == nil {
|
|
t.Error("expected error for missing file")
|
|
}
|
|
}
|
|
|
|
func TestHashBytes(t *testing.T) {
|
|
h := HashBytes([]byte("hello world"))
|
|
if !strings.HasPrefix(h, "sha256-") {
|
|
t.Errorf("expected sha256- prefix, got %q", h)
|
|
}
|
|
|
|
h2 := HashBytes([]byte("hello world"))
|
|
if h != h2 {
|
|
t.Error("same bytes should produce same hash")
|
|
}
|
|
|
|
h3 := HashBytes([]byte("different"))
|
|
if h == h3 {
|
|
t.Error("different bytes should produce different hash")
|
|
}
|
|
}
|
|
|
|
func TestHashDir(t *testing.T) {
|
|
dir := t.TempDir()
|
|
os.MkdirAll(filepath.Join(dir, "sub"), 0755)
|
|
os.WriteFile(filepath.Join(dir, "a.txt"), []byte("aaa"), 0644)
|
|
os.WriteFile(filepath.Join(dir, "sub", "b.txt"), []byte("bbb"), 0644)
|
|
|
|
// Without prefix
|
|
hashes, err := HashDir(dir, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(hashes) != 2 {
|
|
t.Errorf("expected 2 files, got %d", len(hashes))
|
|
}
|
|
if _, ok := hashes["a.txt"]; !ok {
|
|
t.Error("expected a.txt in hashes")
|
|
}
|
|
if _, ok := hashes["sub/b.txt"]; !ok {
|
|
t.Error("expected sub/b.txt in hashes")
|
|
}
|
|
|
|
// With prefix
|
|
hashes, err = HashDir(dir, "mcps/yao/rag-tools")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, ok := hashes["mcps/yao/rag-tools/a.txt"]; !ok {
|
|
t.Error("expected prefixed path")
|
|
}
|
|
}
|