This commit is contained in:
@@ -0,0 +1,160 @@
|
||||
package bfcodegen
|
||||
|
||||
import (
|
||||
"git.blackforestbytes.com/BlackForestBytes/goext/langext"
|
||||
"git.blackforestbytes.com/BlackForestBytes/goext/tst"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestProcessCSIDFileSimple(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
|
||||
src := `package mymodels
|
||||
|
||||
type UserID string // @csid:type [USR]
|
||||
type OrderID string // @csid:type [ORD]
|
||||
`
|
||||
fp := writeTestFile(t, dir, "models.go", src)
|
||||
|
||||
ids, pkg, err := processCSIDFile(dir, fp, false)
|
||||
tst.AssertNoErr(t, err)
|
||||
tst.AssertEqual(t, pkg, "mymodels")
|
||||
tst.AssertEqual(t, len(ids), 2)
|
||||
tst.AssertEqual(t, ids[0].Name, "UserID")
|
||||
tst.AssertEqual(t, ids[0].Prefix, "USR")
|
||||
tst.AssertEqual(t, ids[1].Name, "OrderID")
|
||||
tst.AssertEqual(t, ids[1].Prefix, "ORD")
|
||||
tst.AssertEqual(t, ids[0].FileRelative, "models.go")
|
||||
}
|
||||
|
||||
func TestProcessCSIDFilePrefixMustBeUppercase(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
|
||||
// lowercase prefix should not match the regex (only [A-Z0-9]{3})
|
||||
src := `package x
|
||||
|
||||
type FooID string // @csid:type [usr]
|
||||
`
|
||||
fp := writeTestFile(t, dir, "x.go", src)
|
||||
|
||||
ids, pkg, err := processCSIDFile(dir, fp, false)
|
||||
tst.AssertNoErr(t, err)
|
||||
tst.AssertEqual(t, pkg, "x")
|
||||
tst.AssertEqual(t, len(ids), 0)
|
||||
}
|
||||
|
||||
func TestProcessCSIDFileGeneratedHeaderSkipped(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
|
||||
src := `// Code generated by csid-generate.go DO NOT EDIT.
|
||||
|
||||
package x
|
||||
|
||||
type SkipMeID string // @csid:type [SKP]
|
||||
`
|
||||
fp := writeTestFile(t, dir, "skip.go", src)
|
||||
|
||||
ids, pkg, err := processCSIDFile(dir, fp, false)
|
||||
tst.AssertNoErr(t, err)
|
||||
tst.AssertEqual(t, pkg, "")
|
||||
tst.AssertEqual(t, len(ids), 0)
|
||||
}
|
||||
|
||||
func TestGenerateCharsetIDSpecsEndToEnd(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
|
||||
src1 := `package models
|
||||
|
||||
type EntityID string // @csid:type [ENT]
|
||||
type UserID string // @csid:type [USR]
|
||||
`
|
||||
writeTestFile(t, dir, "a_models.go", src1)
|
||||
|
||||
src2 := `package models
|
||||
|
||||
type OrderID string // @csid:type [ORD]
|
||||
`
|
||||
writeTestFile(t, dir, "b_models.go", src2)
|
||||
|
||||
dest := filepath.Join(dir, "csid_gen.go")
|
||||
|
||||
err := GenerateCharsetIDSpecs(dir, dest, CSIDGenOptions{DebugOutput: langext.PFalse})
|
||||
tst.AssertNoErr(t, err)
|
||||
|
||||
out, err := os.ReadFile(dest)
|
||||
tst.AssertNoErr(t, err)
|
||||
outStr := string(out)
|
||||
|
||||
tst.AssertTrue(t, strings.Contains(outStr, "package models"))
|
||||
tst.AssertTrue(t, strings.Contains(outStr, "ChecksumCharsetIDGenerator"))
|
||||
tst.AssertTrue(t, strings.Contains(outStr, "func NewUserID()"))
|
||||
tst.AssertTrue(t, strings.Contains(outStr, "func NewOrderID()"))
|
||||
tst.AssertTrue(t, strings.Contains(outStr, "func NewEntityID()"))
|
||||
tst.AssertTrue(t, strings.Contains(outStr, `prefixUserID`) && strings.Contains(outStr, `"USR"`))
|
||||
tst.AssertTrue(t, strings.Contains(outStr, `prefixOrderID`) && strings.Contains(outStr, `"ORD"`))
|
||||
tst.AssertTrue(t, strings.Contains(outStr, `prefixEntityID`) && strings.Contains(outStr, `"ENT"`))
|
||||
}
|
||||
|
||||
func TestGenerateCharsetIDSpecsIdempotentWhenUnchanged(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
|
||||
src := `package models
|
||||
|
||||
type SomeID string // @csid:type [SOM]
|
||||
`
|
||||
writeTestFile(t, dir, "models.go", src)
|
||||
dest := filepath.Join(dir, "csid_gen.go")
|
||||
|
||||
err := GenerateCharsetIDSpecs(dir, dest, CSIDGenOptions{})
|
||||
tst.AssertNoErr(t, err)
|
||||
|
||||
content1, err := os.ReadFile(dest)
|
||||
tst.AssertNoErr(t, err)
|
||||
|
||||
err = GenerateCharsetIDSpecs(dir, dest, CSIDGenOptions{})
|
||||
tst.AssertNoErr(t, err)
|
||||
|
||||
content2, err := os.ReadFile(dest)
|
||||
tst.AssertNoErr(t, err)
|
||||
|
||||
tst.AssertEqual(t, string(content1), string(content2))
|
||||
}
|
||||
|
||||
func TestGenerateCharsetIDSpecsErrorsWithoutPackage(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
|
||||
src := `// Code generated by csid-generate.go DO NOT EDIT.
|
||||
|
||||
package x
|
||||
|
||||
type SkippedID string // @csid:type [SKP]
|
||||
`
|
||||
writeTestFile(t, dir, "z.go", src)
|
||||
dest := filepath.Join(dir, "csid_gen.go")
|
||||
|
||||
err := GenerateCharsetIDSpecs(dir, dest, CSIDGenOptions{})
|
||||
tst.AssertTrue(t, err != nil)
|
||||
}
|
||||
|
||||
func TestGenerateCharsetIDSpecsMissingDir(t *testing.T) {
|
||||
dir := filepath.Join(t.TempDir(), "definitely-missing")
|
||||
err := GenerateCharsetIDSpecs(dir, filepath.Join(dir, "csid_gen.go"), CSIDGenOptions{})
|
||||
tst.AssertTrue(t, err != nil)
|
||||
}
|
||||
|
||||
func TestFmtCSIDOutputContainsAllNames(t *testing.T) {
|
||||
ids := []CSIDDef{
|
||||
{File: "a.go", FileRelative: "a.go", Name: "AlphaID", Prefix: "ALP"},
|
||||
{File: "b.go", FileRelative: "b.go", Name: "BetaID", Prefix: "BET"},
|
||||
}
|
||||
out := fmtCSIDOutput("CHK_XYZ", ids, "models")
|
||||
tst.AssertTrue(t, strings.Contains(out, "package models"))
|
||||
tst.AssertTrue(t, strings.Contains(out, "CHK_XYZ"))
|
||||
tst.AssertTrue(t, strings.Contains(out, "AlphaID"))
|
||||
tst.AssertTrue(t, strings.Contains(out, "BetaID"))
|
||||
tst.AssertTrue(t, strings.Contains(out, `prefixAlphaID`) && strings.Contains(out, `"ALP"`))
|
||||
tst.AssertTrue(t, strings.Contains(out, `prefixBetaID`) && strings.Contains(out, `"BET"`))
|
||||
}
|
||||
Reference in New Issue
Block a user