44 lines
1.3 KiB
Go
44 lines
1.3 KiB
Go
package tests
|
|
|
|
import (
|
|
"fmt"
|
|
. "github.com/sub0Zero/.test_files/dvd_rental/dvds/table"
|
|
"github.com/sub0Zero/go-sqlbuilder/generator"
|
|
. "github.com/sub0Zero/go-sqlbuilder/sqlbuilder"
|
|
"gotest.tools/assert"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
folderPath = ".test_files/"
|
|
host = "localhost"
|
|
port = 5432
|
|
user = "postgres"
|
|
password = "postgres"
|
|
dbname = "dvd_rental"
|
|
schemaName = "dvds"
|
|
)
|
|
|
|
//go:generate generator -db "host=localhost port=5432 user=postgres password=postgres dbname=dvd_rental sslmode=disable" -dbName dvd_rental -schema dvds -path .test_files
|
|
|
|
func TestGenerateModel(t *testing.T) {
|
|
connectString := fmt.Sprintf("host=%s port=%d user=%s "+"password=%s dbname=%s sslmode=disable",
|
|
host, port, user, password, dbname)
|
|
|
|
err := generator.Generate(folderPath, connectString, dbname, schemaName)
|
|
|
|
assert.NilError(t, err)
|
|
|
|
err = generator.Generate(folderPath, connectString, dbname, "sport")
|
|
|
|
assert.NilError(t, err)
|
|
}
|
|
|
|
func TestSelectQuery(t *testing.T) {
|
|
query, err := Actor.InnerJoinOn(Store, Eq(Actor.ActorID, Store.StoreID)).
|
|
Select(Store.StoreID, Store.AddressID, Actor.ActorID).String(schemaName)
|
|
|
|
assert.NilError(t, err)
|
|
|
|
assert.Equal(t, query, "SELECT store.store_id,store.address_id,actor.actor_id FROM dvds.actor JOIN dvds.store ON actor.actor_id=store.store_id")
|
|
}
|