SG Command
88 lines
2.8 KiB
Go
88 lines
2.8 KiB
Go
package template
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.warky.dev/wdevs/relspecgo/pkg/models"
|
|
"git.warky.dev/wdevs/relspecgo/pkg/writers"
|
|
)
|
|
|
|
func TestWriterTableIndexValuesDeterministic(t *testing.T) {
|
|
dir := t.TempDir()
|
|
templatePath := filepath.Join(dir, "indexes.tmpl")
|
|
outputDir := filepath.Join(dir, "out")
|
|
outputPath := filepath.Join(outputDir, "accounts.txt")
|
|
templateBody := "{{range values .Table.Indexes}}{{.Name}}:{{join .Columns \",\"}}\n{{end}}"
|
|
|
|
if err := os.MkdirAll(outputDir, 0755); err != nil {
|
|
t.Fatalf("create output dir: %v", err)
|
|
}
|
|
if err := os.WriteFile(templatePath, []byte(templateBody), 0644); err != nil {
|
|
t.Fatalf("write template: %v", err)
|
|
}
|
|
|
|
db := databaseWithMultipleIndexes()
|
|
var first []byte
|
|
const runs = 100
|
|
for i := 0; i < runs; i++ {
|
|
writer, err := NewWriter(&writers.WriterOptions{
|
|
OutputPath: outputDir,
|
|
Metadata: map[string]interface{}{
|
|
"template_path": templatePath,
|
|
"mode": string(TableMode),
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("new writer: %v", err)
|
|
}
|
|
|
|
if err := writer.WriteDatabase(db); err != nil {
|
|
t.Fatalf("write database run %d: %v", i, err)
|
|
}
|
|
|
|
got, err := os.ReadFile(outputPath)
|
|
if err != nil {
|
|
t.Fatalf("read output run %d: %v", i, err)
|
|
}
|
|
|
|
if i == 0 {
|
|
first = got
|
|
continue
|
|
}
|
|
if string(got) != string(first) {
|
|
t.Fatalf("run %d output differed from first run\nfirst:\n%s\nrun %d:\n%s", i, first, i, got)
|
|
}
|
|
}
|
|
|
|
want := strings.Join([]string{
|
|
"idx_accounts_email:email",
|
|
"idx_accounts_last_login:last_login",
|
|
"idx_accounts_name:name",
|
|
"idx_accounts_status:status",
|
|
"idx_accounts_tenant:tenant_id",
|
|
"",
|
|
}, "\n")
|
|
if string(first) != want {
|
|
t.Fatalf("unexpected index order\nwant:\n%s\ngot:\n%s", want, first)
|
|
}
|
|
}
|
|
|
|
func databaseWithMultipleIndexes() *models.Database {
|
|
db := models.InitDatabase("test")
|
|
schema := models.InitSchema("public")
|
|
table := models.InitTable("accounts", "public")
|
|
|
|
table.Indexes["idx_accounts_status"] = &models.Index{Name: "idx_accounts_status", Table: table.Name, Schema: schema.Name, Columns: []string{"status"}}
|
|
table.Indexes["idx_accounts_email"] = &models.Index{Name: "idx_accounts_email", Table: table.Name, Schema: schema.Name, Columns: []string{"email"}}
|
|
table.Indexes["idx_accounts_tenant"] = &models.Index{Name: "idx_accounts_tenant", Table: table.Name, Schema: schema.Name, Columns: []string{"tenant_id"}}
|
|
table.Indexes["idx_accounts_name"] = &models.Index{Name: "idx_accounts_name", Table: table.Name, Schema: schema.Name, Columns: []string{"name"}}
|
|
table.Indexes["idx_accounts_last_login"] = &models.Index{Name: "idx_accounts_last_login", Table: table.Name, Schema: schema.Name, Columns: []string{"last_login"}}
|
|
|
|
schema.Tables = append(schema.Tables, table)
|
|
db.Schemas = append(db.Schemas, schema)
|
|
return db
|
|
}
|