wehub-resource-sync
a06f331eb8
CI / benchmark (push) Has been skipped
install-script / posix-syntax (push) Successful in 6m1s
CI / build-onnx (push) Failing after 6m43s
init-smoke / dry-run (push) Failing after 15m57s
security / govulncheck (push) Has been cancelled
security / trivy-fs (push) Has been cancelled
CI / test (1.26, ubuntu-latest) (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
CI / test (1.26, macos-latest) (push) Has been cancelled
CI / build-windows (push) Has been cancelled
CI / lint (push) Has been cancelled
install-script / powershell-syntax (push) Has been cancelled
install-script / install (macos-14) (push) Has been cancelled
install-script / install (ubuntu-latest) (push) Has been cancelled
72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
package parser
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zzet/gortex/internal/graph"
|
|
)
|
|
|
|
// mockExtractor implements the Extractor interface for testing.
|
|
type mockExtractor struct {
|
|
lang string
|
|
exts []string
|
|
}
|
|
|
|
func (m *mockExtractor) Language() string { return m.lang }
|
|
func (m *mockExtractor) Extensions() []string { return m.exts }
|
|
func (m *mockExtractor) Extract(filePath string, src []byte) (*ExtractionResult, error) {
|
|
return &ExtractionResult{
|
|
Nodes: []*graph.Node{{ID: filePath + "::mock", Name: "mock", Kind: graph.KindFunction}},
|
|
}, nil
|
|
}
|
|
|
|
func TestRegistry_RegisterAndGet(t *testing.T) {
|
|
r := NewRegistry()
|
|
r.Register(&mockExtractor{lang: "go", exts: []string{".go"}})
|
|
|
|
e, ok := r.GetByLanguage("go")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "go", e.Language())
|
|
|
|
e, ok = r.GetByExtension(".go")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "go", e.Language())
|
|
}
|
|
|
|
func TestRegistry_DetectLanguage(t *testing.T) {
|
|
r := NewRegistry()
|
|
r.Register(&mockExtractor{lang: "go", exts: []string{".go"}})
|
|
r.Register(&mockExtractor{lang: "typescript", exts: []string{".ts", ".tsx"}})
|
|
|
|
lang, ok := r.DetectLanguage("pkg/foo.go")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "go", lang)
|
|
|
|
lang, ok = r.DetectLanguage("src/app.tsx")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "typescript", lang)
|
|
}
|
|
|
|
func TestRegistry_UnknownExtension(t *testing.T) {
|
|
r := NewRegistry()
|
|
r.Register(&mockExtractor{lang: "go", exts: []string{".go"}})
|
|
|
|
_, ok := r.GetByExtension(".txt")
|
|
assert.False(t, ok)
|
|
|
|
_, ok = r.DetectLanguage("readme.md")
|
|
assert.False(t, ok)
|
|
}
|
|
|
|
func TestRegistry_SupportedLanguages(t *testing.T) {
|
|
r := NewRegistry()
|
|
r.Register(&mockExtractor{lang: "go", exts: []string{".go"}})
|
|
r.Register(&mockExtractor{lang: "python", exts: []string{".py"}})
|
|
|
|
langs := r.SupportedLanguages()
|
|
assert.Len(t, langs, 2)
|
|
assert.Contains(t, langs, "go")
|
|
assert.Contains(t, langs, "python")
|
|
}
|