check-safety/codegen_test.go
Alex Dunmow cd88c808b0 initial: standalone check-safety module hoisted from CMS
Static safety/lint runner for the BlockNinja codebase. ~25 invariant
checks across Go and frontend sources. Was at git.dev.alexdunmow.com:block/ninja
in backend/cmd/check-safety/ until the 2026-06-06 consolidation moved
the BlockNinja repos under a shared ~/src/blockninja/ parent.

This repo is the standalone extraction:
- Own go.mod (git.dev.alexdunmow.com/block/check-safety, go 1.26.4)
- Vendored internal/{helpers,theme} from CMS (Go's internal/ rule
  blocks cross-module imports; vendoring is the workaround)
- CLI contract unchanged: `check-safety <target-dir> [--flags]`
- CMS Makefile shells into ../check-safety for safety-check /
  install-safety-checker targets

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-06-06 13:04:02 +08:00

76 lines
2.9 KiB
Go

package main
import (
"os"
"path/filepath"
"strings"
"testing"
)
func TestDiscoverCodegenTargetsIncludesCoreAndPluginConfigs(t *testing.T) {
repoRoot := blockNinjaRepoRoot()
backendDir := filepath.Join(repoRoot, "backend")
pluginRoot := filepath.Join(t.TempDir(), "plugin")
writeTestFile(t, filepath.Join(pluginRoot, "plugin.mod"), "[plugin]\nname = \"example\"\nversion = \"1.0.0\"\n", 0644)
writeTestFile(t, filepath.Join(pluginRoot, "sqlc.yaml"), `version: "2"`, 0644)
writeTestFile(t, filepath.Join(pluginRoot, "buf.yaml"), "version: v2\nmodules:\n - path: proto\n", 0644)
targets := discoverCodegenTargets(repoRoot, backendDir, []backendScanTarget{{root: backendDir, display: "backend"}}, []pluginScanTarget{{root: pluginRoot, display: "plugin"}}, true)
if len(targets) != 4 {
t.Fatalf("discoverCodegenTargets() returned %d targets, want 4: %#v", len(targets), targets)
}
}
func TestRunCodegenChecksWithBinariesRunsSQLCAndBuf(t *testing.T) {
root := t.TempDir()
sqlcBin := writeExecutableFile(t, filepath.Join(root, "sqlc"), "#!/bin/sh\nexit 0\n")
bufArgsFile := filepath.Join(root, "buf-args.txt")
bufBin := writeExecutableFile(t, filepath.Join(root, "buf"), "#!/bin/sh\nprintf '%s\\n' \"$@\" > "+shellQuote(bufArgsFile)+"\nexit 0\n")
completed, failures, err := runCodegenChecksWithBinaries([]codegenTarget{
{label: "backend", workdir: root, stage: "sqlc compile", binary: "sqlc", args: []string{"compile"}},
{label: "repo", workdir: root, stage: "buf generate", binary: "buf", args: []string{"generate"}},
}, sqlcBin, bufBin)
if err != nil {
t.Fatalf("runCodegenChecksWithBinaries() error = %v", err)
}
if len(failures) != 0 {
t.Fatalf("failures = %#v, want none", failures)
}
if len(completed) != 2 {
t.Fatalf("completed = %#v, want 2", completed)
}
bufArgsBytes, err := os.ReadFile(bufArgsFile)
if err != nil {
t.Fatalf("ReadFile(%s): %v", bufArgsFile, err)
}
bufArgs := string(bufArgsBytes)
if !strings.Contains(bufArgs, "-o") {
t.Fatalf("buf args = %q, want -o temp output redirection", bufArgs)
}
}
func TestRunCodegenChecksWithBinariesCapturesFailures(t *testing.T) {
root := t.TempDir()
sqlcBin := writeExecutableFile(t, filepath.Join(root, "sqlc"), "#!/bin/sh\necho \"sqlc failed\" >&2\nexit 1\n")
bufBin := writeExecutableFile(t, filepath.Join(root, "buf"), "#!/bin/sh\nexit 0\n")
completed, failures, err := runCodegenChecksWithBinaries([]codegenTarget{
{label: "backend", workdir: root, stage: "sqlc compile", binary: "sqlc", args: []string{"compile"}},
}, sqlcBin, bufBin)
if err != nil {
t.Fatalf("runCodegenChecksWithBinaries() error = %v", err)
}
if len(completed) != 0 {
t.Fatalf("completed = %#v, want none", completed)
}
if len(failures) != 1 {
t.Fatalf("failures = %#v, want 1", failures)
}
if !strings.Contains(failures[0].output, "sqlc failed") {
t.Fatalf("failure output = %q, want stderr", failures[0].output)
}
}