test(04-04): close scaffold and CLI command boundary coverage
- Compile and vet every generated artifact in a copied hello workspace - Reject malformed names, traversal, and duplicates through public Make* - Cover CLI argument forms, --no-migration, and one-arg make inside a plugin Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -537,6 +537,11 @@ func TestScaffoldCoreArtifacts(t *testing.T) {
|
||||
t.Fatal("handwritten plugin.go was rewritten")
|
||||
}
|
||||
|
||||
greeterGo := filepath.Join(dir, "plugins", "greeter", "plugin.go")
|
||||
greeterBefore, err := os.ReadFile(greeterGo)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
greeter, err := MakeCommand(ctx, filepath.Join(dir, "plugins", "greeter"), "", "Wave")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
@@ -544,6 +549,13 @@ func TestScaffoldCoreArtifacts(t *testing.T) {
|
||||
if greeter.Hint == "" || !strings.Contains(greeter.Hint, "generatedCommands()") {
|
||||
t.Fatalf("want accessor hint for handwritten plugin, got %q", greeter.Hint)
|
||||
}
|
||||
greeterAfter, err := os.ReadFile(greeterGo)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !bytes.Equal(greeterBefore, greeterAfter) {
|
||||
t.Fatal("handwritten greeter plugin.go was rewritten")
|
||||
}
|
||||
|
||||
if err := AddPlugin(ctx, dir, filepath.Join("plugins", "demo")); err != nil {
|
||||
t.Fatal(err)
|
||||
@@ -570,21 +582,90 @@ func TestScaffoldAllArtifacts(t *testing.T) {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if _, err := MakeModel(ctx, dir, "golem15.demo", "Album", false); err != nil {
|
||||
model, err := MakeModel(ctx, dir, "golem15.demo", "Album", false)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if _, err := MakeMigration(ctx, dir, "golem15.demo", "AddAlbumIndex"); err != nil {
|
||||
wantModel := filepath.Join(pluginDir, "models", "album.go")
|
||||
if len(model.Files) != 2 || model.Files[0] != wantModel {
|
||||
t.Fatalf("make:model files = %v, want %s plus create-table migration", model.Files, wantModel)
|
||||
}
|
||||
albumSrc, err := os.ReadFile(wantModel)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if _, err := MakeCommand(ctx, dir, "golem15.demo", "Purge"); err != nil {
|
||||
for _, want := range []string{
|
||||
"type Album struct",
|
||||
"CreatedAt",
|
||||
"UpdatedAt",
|
||||
`return "golem15_demo_albums"`,
|
||||
} {
|
||||
if !bytes.Contains(albumSrc, []byte(want)) {
|
||||
t.Fatalf("album.go missing %s:\n%s", want, albumSrc)
|
||||
}
|
||||
}
|
||||
createAlbum, err := filepath.Glob(filepath.Join(pluginDir, "updates", "*_create_golem15_demo_albums.go"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(createAlbum) != 1 || createAlbum[0] != model.Files[1] {
|
||||
t.Fatalf("create-table migration files = %v want %v", createAlbum, model.Files[1:])
|
||||
}
|
||||
migSrc, err := os.ReadFile(createAlbum[0])
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
for _, want := range []string{
|
||||
"ID: \"",
|
||||
"Migrate:",
|
||||
"Rollback:",
|
||||
"CREATE TABLE golem15_demo_albums",
|
||||
"DROP TABLE IF EXISTS golem15_demo_albums",
|
||||
} {
|
||||
if !bytes.Contains(migSrc, []byte(want)) {
|
||||
t.Fatalf("model migration missing %s:\n%s", want, migSrc)
|
||||
}
|
||||
}
|
||||
|
||||
widget, err := MakeModel(ctx, pluginDir, "", "Widget", true)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(widget.Files) != 1 || widget.Files[0] != filepath.Join(pluginDir, "models", "widget.go") {
|
||||
t.Fatalf("--no-migration files = %v", widget.Files)
|
||||
}
|
||||
widgetMigs, err := filepath.Glob(filepath.Join(pluginDir, "updates", "*widget*"))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(widgetMigs) != 0 {
|
||||
t.Fatalf("--no-migration wrote %v", widgetMigs)
|
||||
}
|
||||
|
||||
migration, err := MakeMigration(ctx, dir, "golem15.demo", "AddAlbumIndex")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(migration.Files) != 1 || !strings.Contains(migration.Files[0], "add_album_index.go") {
|
||||
t.Fatalf("make:migration files = %v", migration.Files)
|
||||
}
|
||||
|
||||
command, err := MakeCommand(ctx, dir, "golem15.demo", "Purge")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
wantCommand := filepath.Join(pluginDir, "console", "purge.go")
|
||||
if len(command.Files) != 1 || command.Files[0] != wantCommand {
|
||||
t.Fatalf("make:command files = %v, want %s", command.Files, wantCommand)
|
||||
}
|
||||
|
||||
job, err := MakeJob(ctx, dir, "golem15.demo", "Reindex")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(job.Files) != 1 {
|
||||
t.Fatalf("make:job files = %v", job.Files)
|
||||
wantJob := filepath.Join(pluginDir, "jobs", "reindex.go")
|
||||
if len(job.Files) != 1 || job.Files[0] != wantJob {
|
||||
t.Fatalf("make:job files = %v, want %s", job.Files, wantJob)
|
||||
}
|
||||
jobSrc, err := os.ReadFile(job.Files[0])
|
||||
if err != nil {
|
||||
@@ -592,8 +673,10 @@ func TestScaffoldAllArtifacts(t *testing.T) {
|
||||
}
|
||||
for _, want := range []string{
|
||||
"func (ReindexArgs) Kind()",
|
||||
`Kind() string { return "golem15.demo.reindex" }`,
|
||||
"func (reindexJob) Work(",
|
||||
"unexpected args type",
|
||||
"func ReindexJob()",
|
||||
} {
|
||||
if !bytes.Contains(jobSrc, []byte(want)) {
|
||||
t.Fatalf("job.go missing %s:\n%s", want, jobSrc)
|
||||
@@ -607,10 +690,20 @@ func TestScaffoldAllArtifacts(t *testing.T) {
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
wantAdmin := []string{
|
||||
filepath.Join(pluginDir, "controllers", "albums.go"),
|
||||
filepath.Join(pluginDir, "controllers", "albums", "fields.yaml"),
|
||||
filepath.Join(pluginDir, "controllers", "albums", "columns.yaml"),
|
||||
}
|
||||
if len(admin.Files) != 3 {
|
||||
t.Fatalf("make:admin-controller files = %v", admin.Files)
|
||||
}
|
||||
adminSrc, err := os.ReadFile(filepath.Join(pluginDir, "controllers", "albums.go"))
|
||||
for i, want := range wantAdmin {
|
||||
if admin.Files[i] != want {
|
||||
t.Fatalf("admin file[%d] = %s, want %s", i, admin.Files[i], want)
|
||||
}
|
||||
}
|
||||
adminSrc, err := os.ReadFile(wantAdmin[0])
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
@@ -646,6 +739,7 @@ func TestScaffoldAllArtifacts(t *testing.T) {
|
||||
}
|
||||
for _, want := range []string{
|
||||
"&models.Album{}",
|
||||
"&models.Widget{}",
|
||||
"updates.CreateAlbums()",
|
||||
"updates.AddAlbumIndex()",
|
||||
"console.PurgeCommand()",
|
||||
@@ -684,8 +778,13 @@ func TestScaffoldAllArtifacts(t *testing.T) {
|
||||
if err := App(ctx, dir, &buf); err != nil {
|
||||
t.Fatalf("app build: %v\n%s", err, buf.String())
|
||||
}
|
||||
vet := execGo(t, pluginDir, "vet", "./...")
|
||||
if out, err := vet.CombinedOutput(); err != nil {
|
||||
if out, err := execGo(t, dir, "build", "./...").CombinedOutput(); err != nil {
|
||||
t.Fatalf("copied workspace go build: %v\n%s", err, out)
|
||||
}
|
||||
if out, err := execGo(t, dir, "vet", "./...").CombinedOutput(); err != nil {
|
||||
t.Fatalf("copied workspace go vet: %v\n%s", err, out)
|
||||
}
|
||||
if out, err := execGo(t, pluginDir, "vet", "./...").CombinedOutput(); err != nil {
|
||||
t.Fatalf("plugin go vet: %v\n%s", err, out)
|
||||
}
|
||||
}
|
||||
@@ -721,6 +820,45 @@ func TestModelsLeaf(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestMakeArtifactRejectsMalformedNames(t *testing.T) {
|
||||
dir := copyHelloApp(t)
|
||||
ctx := t.Context()
|
||||
if _, err := MakePlugin(ctx, dir, "golem15.demo"); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
for _, name := range []string{"", "album", "../Evil", "Album;rm", `Album$(x)`, "Album/tmp", "2Album", "Album name"} {
|
||||
if _, err := MakeModel(ctx, dir, "golem15.demo", name, true); err == nil {
|
||||
t.Fatalf("make:model %q: want error", name)
|
||||
}
|
||||
if _, err := MakeMigration(ctx, dir, "golem15.demo", name); err == nil {
|
||||
t.Fatalf("make:migration %q: want error", name)
|
||||
}
|
||||
if _, err := MakeCommand(ctx, dir, "golem15.demo", name); err == nil {
|
||||
t.Fatalf("make:command %q: want error", name)
|
||||
}
|
||||
if _, err := MakeJob(ctx, dir, "golem15.demo", name); err == nil {
|
||||
t.Fatalf("make:job %q: want error", name)
|
||||
}
|
||||
if _, err := MakeAdminController(ctx, dir, "golem15.demo", name); err == nil {
|
||||
t.Fatalf("make:admin-controller %q: want error", name)
|
||||
}
|
||||
}
|
||||
|
||||
for _, id := range []string{"golem15.demo/../tmp", "Golem15.Demo", "../evil.plugin", "golem15.demo;rm"} {
|
||||
if _, err := MakeModel(ctx, dir, id, "Album", true); err == nil {
|
||||
t.Fatalf("make:model plugin %q: want error", id)
|
||||
}
|
||||
}
|
||||
|
||||
if _, err := MakeModel(ctx, dir, "golem15.demo", "Album", true); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if _, err := MakeModel(ctx, dir, "golem15.demo", "Album", true); err == nil {
|
||||
t.Fatal("duplicate make:model want error")
|
||||
}
|
||||
}
|
||||
|
||||
func assertScaffoldFiles(t *testing.T, pluginDir string) {
|
||||
t.Helper()
|
||||
assertWinterPluginShape(t, pluginDir)
|
||||
|
||||
Reference in New Issue
Block a user