VERSE
1e04949242
This project is starting to grow, and I've seen flaws in the initial project structure design I think I'll restructure the entire project directory for better scalability and also cleaner
95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package spec
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestFileGenerator(t *testing.T) {
|
|
// Setup mock files
|
|
tmp := t.TempDir()
|
|
_ = os.MkdirAll(filepath.Join(tmp, "src"), 0755)
|
|
_ = os.WriteFile(filepath.Join(tmp, "main.go"), []byte(""), 0644)
|
|
_ = os.WriteFile(filepath.Join(tmp, "README.md"), []byte(""), 0644)
|
|
_ = os.WriteFile(filepath.Join(tmp, ".hidden"), []byte(""), 0644)
|
|
_ = os.WriteFile(filepath.Join(tmp, "src/utils.go"), []byte(""), 0644)
|
|
|
|
oldWd, _ := os.Getwd()
|
|
_ = os.Chdir(tmp)
|
|
defer func() { _ = os.Chdir(oldWd) }()
|
|
|
|
|
|
t.Run("dirOnly shows only dirs", func(t *testing.T) {
|
|
gen := FileGenerator("/")
|
|
results := gen([]string{"cd", ""}, "cd ", "")
|
|
for _, r := range results {
|
|
if !strings.HasSuffix(r.Cmd, "/") {
|
|
t.Errorf("Expected only directories, got %q", r.Cmd)
|
|
}
|
|
}
|
|
})
|
|
|
|
|
|
t.Run("Filter extension", func(t *testing.T) {
|
|
gen := FileGenerator(".go")
|
|
results := gen([]string{"ls", ""}, "ls ", "")
|
|
foundMain := false
|
|
for _, r := range results {
|
|
if r.Cmd == "main.go" {
|
|
foundMain = true
|
|
}
|
|
if r.Cmd == "README.md" {
|
|
t.Errorf("Did not expect README.md when filtering for .go")
|
|
}
|
|
}
|
|
if !foundMain {
|
|
t.Errorf("FileGenerator(\".go\") did not suggest main.go")
|
|
}
|
|
})
|
|
|
|
|
|
t.Run("Nested path", func(t *testing.T) {
|
|
gen := FileGenerator()
|
|
results := gen([]string{"ls", "src/u"}, "ls src/u", "src/u")
|
|
foundUtils := false
|
|
for _, r := range results {
|
|
if r.Cmd == "src/utils.go" {
|
|
foundUtils = true
|
|
}
|
|
}
|
|
if !foundUtils {
|
|
t.Errorf("Did not find src/utils.go for nested path src/u")
|
|
}
|
|
})
|
|
|
|
|
|
t.Run("Deep scan 1 level", func(t *testing.T) {
|
|
gen := FileGenerator()
|
|
results := gen([]string{"ls", "src/"}, "ls src/", "src/")
|
|
foundUtils := false
|
|
for _, r := range results {
|
|
if r.Cmd == "src/utils.go" {
|
|
foundUtils = true
|
|
}
|
|
}
|
|
if !foundUtils {
|
|
t.Errorf("Deep scan did not find src/utils.go")
|
|
}
|
|
})
|
|
|
|
|
|
|
|
|
|
t.Run("Hidden files are skipped", func(t *testing.T) {
|
|
gen := FileGenerator()
|
|
results := gen([]string{"ls", ""}, "ls ", "")
|
|
for _, r := range results {
|
|
if strings.HasPrefix(r.Cmd, ".") {
|
|
t.Errorf("Hidden file %q should be skipped", r.Cmd)
|
|
}
|
|
}
|
|
})
|
|
}
|