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
78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package languages
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/zzet/gortex/internal/graph"
|
|
)
|
|
|
|
func TestSQLExtractor_CreateTable(t *testing.T) {
|
|
src := []byte(`CREATE TABLE users (
|
|
id INTEGER PRIMARY KEY,
|
|
name TEXT NOT NULL,
|
|
email TEXT UNIQUE
|
|
);
|
|
`)
|
|
e := NewSQLExtractor()
|
|
result, err := e.Extract("schema.sql", src)
|
|
require.NoError(t, err)
|
|
|
|
types := nodesOfKind(result.Nodes, graph.KindType)
|
|
require.Len(t, types, 1)
|
|
assert.Equal(t, "users", types[0].Name)
|
|
|
|
// Columns as variables with EdgeMemberOf.
|
|
vars := nodesOfKind(result.Nodes, graph.KindVariable)
|
|
assert.Len(t, vars, 3) // id, name, email
|
|
|
|
memberEdges := edgesOfKind(result.Edges, graph.EdgeMemberOf)
|
|
assert.Len(t, memberEdges, 3)
|
|
for _, e := range memberEdges {
|
|
assert.Equal(t, "schema.sql::users", e.To)
|
|
}
|
|
}
|
|
|
|
func TestSQLExtractor_MultipleTables(t *testing.T) {
|
|
src := []byte(`CREATE TABLE users (
|
|
id INTEGER PRIMARY KEY
|
|
);
|
|
|
|
CREATE TABLE orders (
|
|
id INTEGER PRIMARY KEY,
|
|
user_id INTEGER
|
|
);
|
|
`)
|
|
e := NewSQLExtractor()
|
|
result, err := e.Extract("schema.sql", src)
|
|
require.NoError(t, err)
|
|
|
|
types := nodesOfKind(result.Nodes, graph.KindType)
|
|
assert.Len(t, types, 2)
|
|
|
|
names := make([]string, len(types))
|
|
for i, t := range types {
|
|
names[i] = t.Name
|
|
}
|
|
assert.Contains(t, names, "users")
|
|
assert.Contains(t, names, "orders")
|
|
}
|
|
|
|
func TestSQLExtractor_FileNode(t *testing.T) {
|
|
src := []byte(`CREATE TABLE t (id INT);`)
|
|
e := NewSQLExtractor()
|
|
result, err := e.Extract("test.sql", src)
|
|
require.NoError(t, err)
|
|
|
|
files := nodesOfKind(result.Nodes, graph.KindFile)
|
|
require.Len(t, files, 1)
|
|
assert.Equal(t, "test.sql", files[0].Name)
|
|
}
|
|
|
|
func TestSQLExtractor_Extensions(t *testing.T) {
|
|
e := NewSQLExtractor()
|
|
assert.Equal(t, "sql", e.Language())
|
|
assert.Equal(t, []string{".sql"}, e.Extensions())
|
|
}
|