Improve code coverage.

This commit is contained in:
go-jet 2019-09-20 13:55:07 +02:00
parent acd9d20339
commit f5287628aa
15 changed files with 155 additions and 79 deletions

View file

@ -6,6 +6,7 @@ import (
"fmt"
"github.com/go-jet/jet/generator/mysql"
"github.com/go-jet/jet/generator/postgres"
"github.com/go-jet/jet/internal/utils"
"github.com/go-jet/jet/tests/dbconfig"
_ "github.com/go-sql-driver/mysql"
_ "github.com/lib/pq"
@ -60,7 +61,7 @@ func initMySQLDB() {
cmd.Stdout = os.Stdout
err := cmd.Run()
panicOnError(err)
utils.PanicOnError(err)
err = mysql.Generate("./.gentestdata/mysql", mysql.DBConnection{
Host: dbconfig.MySqLHost,
@ -70,7 +71,7 @@ func initMySQLDB() {
DBName: dbName,
})
panicOnError(err)
utils.PanicOnError(err)
}
}
@ -104,22 +105,16 @@ func initPostgresDB() {
SchemaName: schemaName,
SslMode: "disable",
})
panicOnError(err)
utils.PanicOnError(err)
}
}
func execFile(db *sql.DB, sqlFilePath string) {
testSampleSql, err := ioutil.ReadFile(sqlFilePath)
panicOnError(err)
utils.PanicOnError(err)
_, err = db.Exec(string(testSampleSql))
panicOnError(err)
}
func panicOnError(err error) {
if err != nil {
panic(err)
}
utils.PanicOnError(err)
}
func printOnError(err error) {