Add SQLBuilder support for SQLite databases.
This commit is contained in:
go-jet
parent
d197956271
commit
e8f4c2b31b
50 changed files with 5851 additions and 75 deletions
298
tests/sqlite/generator_test.go
Normal file
298
tests/sqlite/generator_test.go
Normal file
|
|
@ -0,0 +1,298 @@
|
|||
package sqlite
|
||||
|
||||
import (
|
||||
"github.com/go-jet/jet/v2/generator/sqlite"
|
||||
"github.com/go-jet/jet/v2/internal/testutils"
|
||||
"github.com/go-jet/jet/v2/tests/.gentestdata/sqlite/sakila/model"
|
||||
"github.com/go-jet/jet/v2/tests/internal/utils/repo"
|
||||
"github.com/stretchr/testify/require"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"os/exec"
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestGeneratedModel(t *testing.T) {
|
||||
actor := model.Actor{}
|
||||
|
||||
require.Equal(t, reflect.TypeOf(actor.ActorID).String(), "int32")
|
||||
actorIDField, ok := reflect.TypeOf(actor).FieldByName("ActorID")
|
||||
require.True(t, ok)
|
||||
require.Equal(t, actorIDField.Tag.Get("sql"), "primary_key")
|
||||
require.Equal(t, reflect.TypeOf(actor.FirstName).String(), "string")
|
||||
require.Equal(t, reflect.TypeOf(actor.LastName).String(), "string")
|
||||
require.Equal(t, reflect.TypeOf(actor.LastUpdate).String(), "time.Time")
|
||||
|
||||
filmActor := model.FilmActor{}
|
||||
|
||||
require.Equal(t, reflect.TypeOf(filmActor.FilmID).String(), "int32")
|
||||
filmIDField, ok := reflect.TypeOf(filmActor).FieldByName("FilmID")
|
||||
require.True(t, ok)
|
||||
require.Equal(t, filmIDField.Tag.Get("sql"), "primary_key")
|
||||
|
||||
require.Equal(t, reflect.TypeOf(filmActor.ActorID).String(), "int32")
|
||||
actorIDField, ok = reflect.TypeOf(filmActor).FieldByName("ActorID")
|
||||
require.True(t, ok)
|
||||
require.Equal(t, filmIDField.Tag.Get("sql"), "primary_key")
|
||||
|
||||
staff := model.Staff{}
|
||||
|
||||
require.Equal(t, reflect.TypeOf(staff.Email).String(), "*string")
|
||||
require.Equal(t, reflect.TypeOf(staff.Picture).String(), "*[]uint8")
|
||||
}
|
||||
|
||||
var testDatabaseFilePath = repo.GetTestDataFilePath("/init/sqlite/sakila.db")
|
||||
var genDestDir = repo.GetTestsFilePath("/sqlite/.gen")
|
||||
|
||||
func TestGenerator(t *testing.T) {
|
||||
for i := 0; i < 3; i++ {
|
||||
err := sqlite.GenerateDSN(testDatabaseFilePath, genDestDir)
|
||||
require.NoError(t, err)
|
||||
|
||||
assertGeneratedFiles(t)
|
||||
}
|
||||
|
||||
err := os.RemoveAll(genDestDir)
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestCmdGenerator(t *testing.T) {
|
||||
cmd := exec.Command("jet", "-source=SQLite", "-dsn=file://"+testDatabaseFilePath, "-path="+genDestDir)
|
||||
|
||||
cmd.Stderr = os.Stderr
|
||||
cmd.Stdout = os.Stdout
|
||||
|
||||
err := cmd.Run()
|
||||
require.NoError(t, err)
|
||||
|
||||
assertGeneratedFiles(t)
|
||||
|
||||
err = os.RemoveAll(genDestDir)
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func assertGeneratedFiles(t *testing.T) {
|
||||
// Table SQL Builder files
|
||||
tableSQLBuilderFiles, err := ioutil.ReadDir(genDestDir + "/table")
|
||||
require.NoError(t, err)
|
||||
|
||||
testutils.AssertFileNamesEqual(t, tableSQLBuilderFiles, "actor.go", "address.go", "category.go", "city.go", "country.go",
|
||||
"customer.go", "film.go", "film_actor.go", "film_category.go", "film_text.go", "inventory.go", "language.go",
|
||||
"payment.go", "rental.go", "staff.go", "store.go")
|
||||
|
||||
testutils.AssertFileContent(t, genDestDir+"/table/actor.go", actorSQLBuilderFile)
|
||||
|
||||
// View SQL Builder files
|
||||
viewSQLBuilderFiles, err := ioutil.ReadDir(genDestDir + "/view")
|
||||
require.NoError(t, err)
|
||||
|
||||
testutils.AssertFileNamesEqual(t, viewSQLBuilderFiles, "film_list.go", "sales_by_film_category.go",
|
||||
"customer_list.go", "sales_by_store.go", "staff_list.go")
|
||||
|
||||
testutils.AssertFileContent(t, genDestDir+"/view/film_list.go", filmListSQLBuilderFile)
|
||||
|
||||
// Model files
|
||||
modelFiles, err := ioutil.ReadDir(genDestDir + "/model")
|
||||
require.NoError(t, err)
|
||||
|
||||
testutils.AssertFileNamesEqual(t, modelFiles, "actor.go", "address.go", "category.go", "city.go", "country.go",
|
||||
"customer.go", "film.go", "film_actor.go", "film_category.go", "film_text.go", "inventory.go", "language.go",
|
||||
"payment.go", "rental.go", "staff.go", "store.go",
|
||||
"film_list.go", "sales_by_film_category.go",
|
||||
"customer_list.go", "sales_by_store.go", "staff_list.go")
|
||||
|
||||
testutils.AssertFileContent(t, genDestDir+"/model/address.go", addressModelFile)
|
||||
}
|
||||
|
||||
const actorSQLBuilderFile = `
|
||||
//
|
||||
// Code generated by go-jet DO NOT EDIT.
|
||||
//
|
||||
// WARNING: Changes to this file may cause incorrect behavior
|
||||
// and will be lost if the code is regenerated
|
||||
//
|
||||
|
||||
package table
|
||||
|
||||
import (
|
||||
"github.com/go-jet/jet/v2/sqlite"
|
||||
)
|
||||
|
||||
var Actor = newActorTable("", "actor", "")
|
||||
|
||||
type actorTable struct {
|
||||
sqlite.Table
|
||||
|
||||
//Columns
|
||||
ActorID sqlite.ColumnInteger
|
||||
FirstName sqlite.ColumnString
|
||||
LastName sqlite.ColumnString
|
||||
LastUpdate sqlite.ColumnTimestamp
|
||||
|
||||
AllColumns sqlite.ColumnList
|
||||
MutableColumns sqlite.ColumnList
|
||||
}
|
||||
|
||||
type ActorTable struct {
|
||||
actorTable
|
||||
|
||||
EXCLUDED actorTable
|
||||
}
|
||||
|
||||
// AS creates new ActorTable with assigned alias
|
||||
func (a ActorTable) AS(alias string) *ActorTable {
|
||||
return newActorTable(a.SchemaName(), a.TableName(), alias)
|
||||
}
|
||||
|
||||
// Schema creates new ActorTable with assigned schema name
|
||||
func (a ActorTable) FromSchema(schemaName string) *ActorTable {
|
||||
return newActorTable(schemaName, a.TableName(), a.Alias())
|
||||
}
|
||||
|
||||
func newActorTable(schemaName, tableName, alias string) *ActorTable {
|
||||
return &ActorTable{
|
||||
actorTable: newActorTableImpl(schemaName, tableName, alias),
|
||||
EXCLUDED: newActorTableImpl("", "excluded", ""),
|
||||
}
|
||||
}
|
||||
|
||||
func newActorTableImpl(schemaName, tableName, alias string) actorTable {
|
||||
var (
|
||||
ActorIDColumn = sqlite.IntegerColumn("actor_id")
|
||||
FirstNameColumn = sqlite.StringColumn("first_name")
|
||||
LastNameColumn = sqlite.StringColumn("last_name")
|
||||
LastUpdateColumn = sqlite.TimestampColumn("last_update")
|
||||
allColumns = sqlite.ColumnList{ActorIDColumn, FirstNameColumn, LastNameColumn, LastUpdateColumn}
|
||||
mutableColumns = sqlite.ColumnList{FirstNameColumn, LastNameColumn, LastUpdateColumn}
|
||||
)
|
||||
|
||||
return actorTable{
|
||||
Table: sqlite.NewTable(schemaName, tableName, alias, allColumns...),
|
||||
|
||||
//Columns
|
||||
ActorID: ActorIDColumn,
|
||||
FirstName: FirstNameColumn,
|
||||
LastName: LastNameColumn,
|
||||
LastUpdate: LastUpdateColumn,
|
||||
|
||||
AllColumns: allColumns,
|
||||
MutableColumns: mutableColumns,
|
||||
}
|
||||
}
|
||||
`
|
||||
|
||||
const filmListSQLBuilderFile = `
|
||||
//
|
||||
// Code generated by go-jet DO NOT EDIT.
|
||||
//
|
||||
// WARNING: Changes to this file may cause incorrect behavior
|
||||
// and will be lost if the code is regenerated
|
||||
//
|
||||
|
||||
package view
|
||||
|
||||
import (
|
||||
"github.com/go-jet/jet/v2/sqlite"
|
||||
)
|
||||
|
||||
var FilmList = newFilmListTable("", "film_list", "")
|
||||
|
||||
type filmListTable struct {
|
||||
sqlite.Table
|
||||
|
||||
//Columns
|
||||
Fid sqlite.ColumnInteger
|
||||
Title sqlite.ColumnString
|
||||
Description sqlite.ColumnString
|
||||
Category sqlite.ColumnString
|
||||
Price sqlite.ColumnFloat
|
||||
Length sqlite.ColumnInteger
|
||||
Rating sqlite.ColumnString
|
||||
Actors sqlite.ColumnString
|
||||
|
||||
AllColumns sqlite.ColumnList
|
||||
MutableColumns sqlite.ColumnList
|
||||
}
|
||||
|
||||
type FilmListTable struct {
|
||||
filmListTable
|
||||
|
||||
EXCLUDED filmListTable
|
||||
}
|
||||
|
||||
// AS creates new FilmListTable with assigned alias
|
||||
func (a FilmListTable) AS(alias string) *FilmListTable {
|
||||
return newFilmListTable(a.SchemaName(), a.TableName(), alias)
|
||||
}
|
||||
|
||||
// Schema creates new FilmListTable with assigned schema name
|
||||
func (a FilmListTable) FromSchema(schemaName string) *FilmListTable {
|
||||
return newFilmListTable(schemaName, a.TableName(), a.Alias())
|
||||
}
|
||||
|
||||
func newFilmListTable(schemaName, tableName, alias string) *FilmListTable {
|
||||
return &FilmListTable{
|
||||
filmListTable: newFilmListTableImpl(schemaName, tableName, alias),
|
||||
EXCLUDED: newFilmListTableImpl("", "excluded", ""),
|
||||
}
|
||||
}
|
||||
|
||||
func newFilmListTableImpl(schemaName, tableName, alias string) filmListTable {
|
||||
var (
|
||||
FidColumn = sqlite.IntegerColumn("FID")
|
||||
TitleColumn = sqlite.StringColumn("title")
|
||||
DescriptionColumn = sqlite.StringColumn("description")
|
||||
CategoryColumn = sqlite.StringColumn("category")
|
||||
PriceColumn = sqlite.FloatColumn("price")
|
||||
LengthColumn = sqlite.IntegerColumn("length")
|
||||
RatingColumn = sqlite.StringColumn("rating")
|
||||
ActorsColumn = sqlite.StringColumn("actors")
|
||||
allColumns = sqlite.ColumnList{FidColumn, TitleColumn, DescriptionColumn, CategoryColumn, PriceColumn, LengthColumn, RatingColumn, ActorsColumn}
|
||||
mutableColumns = sqlite.ColumnList{FidColumn, TitleColumn, DescriptionColumn, CategoryColumn, PriceColumn, LengthColumn, RatingColumn, ActorsColumn}
|
||||
)
|
||||
|
||||
return filmListTable{
|
||||
Table: sqlite.NewTable(schemaName, tableName, alias, allColumns...),
|
||||
|
||||
//Columns
|
||||
Fid: FidColumn,
|
||||
Title: TitleColumn,
|
||||
Description: DescriptionColumn,
|
||||
Category: CategoryColumn,
|
||||
Price: PriceColumn,
|
||||
Length: LengthColumn,
|
||||
Rating: RatingColumn,
|
||||
Actors: ActorsColumn,
|
||||
|
||||
AllColumns: allColumns,
|
||||
MutableColumns: mutableColumns,
|
||||
}
|
||||
}
|
||||
`
|
||||
|
||||
const addressModelFile = `
|
||||
//
|
||||
// Code generated by go-jet DO NOT EDIT.
|
||||
//
|
||||
// WARNING: Changes to this file may cause incorrect behavior
|
||||
// and will be lost if the code is regenerated
|
||||
//
|
||||
|
||||
package model
|
||||
|
||||
import (
|
||||
"time"
|
||||
)
|
||||
|
||||
type Address struct {
|
||||
AddressID int32 ` + "`sql:\"primary_key\"`" + `
|
||||
Address string
|
||||
Address2 *string
|
||||
District string
|
||||
CityID int32
|
||||
PostalCode *string
|
||||
Phone string
|
||||
LastUpdate time.Time
|
||||
}
|
||||
`
|
||||
Loading…
Add table
Add a link
Reference in a new issue