schema_builder_test.go

package nero_test

import (
	"math/big"
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/stevenferrer/nero"
)

type MyStruct struct {
	ID   *big.Int
	Name string
}

func TestSchemaBuilder(t *testing.T) {
	pkg := "mypkg"
	table := "mytable"
	ms := &MyStruct{}
	schemaBuilder := nero.NewSchemaBuilder(ms).
		PkgName(pkg).Table(table).
		Identity(
			nero.NewFieldBuilder("id", ms.ID).
				Auto().StructField("ID").Build(),
		).
		Fields(nero.NewFieldBuilder("name", ms.Name).Build())

	schema := schemaBuilder.Build()
	assert.Equal(t, pkg, schema.PkgName())
	assert.Equal(t, table, schema.Table())
	assert.NotNil(t, schema.Identity())
	assert.Len(t, schema.Fields(), 1)
	assert.Len(t, schema.Imports(), 2)
	assert.Len(t, schema.Templates(), 2)
	assert.NotNil(t, schema.TypeInfo())
	assert.Equal(t, "MyStruct", schema.TypeName())
	assert.Equal(t, "MyStructs", schema.TypeNamePlural())
	assert.Equal(t, "myStruct", schema.TypeIdentifier())
	assert.Equal(t, "myStructs", schema.TypeIdentifierPlural())

	tmpl := nero.NewPostgresTemplate()
	schema = schemaBuilder.Templates(tmpl).Build()
	assert.Len(t, schema.Templates(), 1)
}