Tests clean up.

tests/init/init.go

@@ -0,0 +1,60 @@
+package main
+
+import (
+	"database/sql"
+	"github.com/go-jet/jet/generator"
+	"github.com/go-jet/jet/tests/dbconfig"
+	"io/ioutil"
+)
+
+func main() {
+	db, err := sql.Open("postgres", dbconfig.ConnectString)
+	if err != nil {
+		panic("Failed to connect to test db")
+	}
+	defer db.Close()
+
+	testSampleSql, err := ioutil.ReadFile("./init/data/test_sample.sql")
+
+	panicOnError(err)
+
+	_, err = db.Exec(string(testSampleSql))
+
+	panicOnError(err)
+
+	dvdsSql, err := ioutil.ReadFile("./init/data/dvds.sql")
+
+	panicOnError(err)
+
+	_, err = db.Exec(string(dvdsSql))
+
+	panicOnError(err)
+
+	err = generator.Generate("./.test_files", generator.GeneratorData{
+		Host:       dbconfig.Host,
+		Port:       "5432",
+		User:       dbconfig.User,
+		Password:   dbconfig.Password,
+		DBName:     dbconfig.DBName,
+		SchemaName: "dvds",
+	})
+
+	panicOnError(err)
+
+	err = generator.Generate("./.test_files", generator.GeneratorData{
+		Host:       dbconfig.Host,
+		Port:       "5432",
+		User:       dbconfig.User,
+		Password:   dbconfig.Password,
+		DBName:     dbconfig.DBName,
+		SchemaName: "test_sample",
+	})
+
+	panicOnError(err)
+}
+
+func panicOnError(err error) {
+	if err != nil {
+		panic(err)
+	}
+}