jet/tests/generator_test.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")
}