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
83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package languages
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/zzet/gortex/internal/graph"
|
|
)
|
|
|
|
func TestZigExtractor_Functions(t *testing.T) {
|
|
src := []byte(`
|
|
pub fn add(a: i32, b: i32) i32 {
|
|
return a + b;
|
|
}
|
|
|
|
fn helper() void {
|
|
doWork();
|
|
}
|
|
`)
|
|
e := NewZigExtractor()
|
|
result, err := e.Extract("math.zig", src)
|
|
require.NoError(t, err)
|
|
|
|
funcs := nodesOfKind(result.Nodes, graph.KindFunction)
|
|
names := make([]string, len(funcs))
|
|
for i, f := range funcs {
|
|
names[i] = f.Name
|
|
}
|
|
assert.Contains(t, names, "add")
|
|
assert.Contains(t, names, "helper")
|
|
}
|
|
|
|
func TestZigExtractor_TypesAndImports(t *testing.T) {
|
|
src := []byte(`
|
|
const std = @import("std");
|
|
const os = @import("os");
|
|
|
|
pub const Point = struct {
|
|
x: f64,
|
|
y: f64,
|
|
};
|
|
|
|
const Color = enum {
|
|
red,
|
|
green,
|
|
blue,
|
|
};
|
|
`)
|
|
e := NewZigExtractor()
|
|
result, err := e.Extract("types.zig", src)
|
|
require.NoError(t, err)
|
|
|
|
types := nodesOfKind(result.Nodes, graph.KindType)
|
|
typeNames := make([]string, len(types))
|
|
for i, n := range types {
|
|
typeNames[i] = n.Name
|
|
}
|
|
assert.Contains(t, typeNames, "Point")
|
|
assert.Contains(t, typeNames, "Color")
|
|
|
|
imports := edgesOfKind(result.Edges, graph.EdgeImports)
|
|
assert.Len(t, imports, 2)
|
|
}
|
|
|
|
func TestZigExtractor_Variables(t *testing.T) {
|
|
src := []byte(`
|
|
const MAX_SIZE = 1024;
|
|
var counter = 0;
|
|
`)
|
|
e := NewZigExtractor()
|
|
result, err := e.Extract("config.zig", src)
|
|
require.NoError(t, err)
|
|
|
|
vars := nodesOfKind(result.Nodes, graph.KindVariable)
|
|
varNames := make([]string, len(vars))
|
|
for i, v := range vars {
|
|
varNames[i] = v.Name
|
|
}
|
|
assert.Contains(t, varNames, "MAX_SIZE")
|
|
assert.Contains(t, varNames, "counter")
|
|
}
|