Files
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
chore: import upstream snapshot with attribution
2026-07-13 12:33:42 +08:00

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")
}