jet/tests/init/init.go

package main

import (
	"database/sql"
	"fmt"
	postgres_generator "github.com/go-jet/jet/generator/postgresgen"
	"github.com/go-jet/jet/tests/dbconfig"
	"io/ioutil"
)

func main() {
	fmt.Println(dbconfig.ConnectString)

	db, err := sql.Open("postgres", dbconfig.ConnectString)
	if err != nil {
		panic("Failed to connect to test db")
	}
	defer func() {
		err := db.Close()
		printOnError(err)
	}()

	schemaNames := []string{
		"dvds",
		"test_sample",
		"chinook",
	}

	for _, schemaName := range schemaNames {
		testSampleSql, err := ioutil.ReadFile("./init/data/" + schemaName + ".sql")

		panicOnError(err)

		_, err = db.Exec(string(testSampleSql))

		err = postgres_generator.Generate("./.gentestdata", postgres_generator.DBConnection{
			Host:       dbconfig.Host,
			Port:       "5432",
			User:       dbconfig.User,
			Password:   dbconfig.Password,
			DBName:     dbconfig.DBName,
			SchemaName: schemaName,
			SslMode:    "disable",
		})

		panicOnError(err)
	}
}

func panicOnError(err error) {
	if err != nil {
		panic(err)
	}
}

func printOnError(err error) {
	if err != nil {
		fmt.Println(err.Error())
	}
}