Add more unit tests.

This commit is contained in:
go-jet 2019-09-20 19:24:54 +02:00
parent 74b94f0d8b
commit 65e05021bc
3 changed files with 43 additions and 32 deletions

View file

@ -25,7 +25,7 @@ func TestToGoIdentifier(t *testing.T) {
assert.Equal(t, ToGoIdentifier("My-Table"), "MyTable")
}
func TestErrorCatch(t *testing.T) {
func TestErrorCatchErr(t *testing.T) {
var err error
func() {
@ -36,3 +36,15 @@ func TestErrorCatch(t *testing.T) {
assert.Error(t, err, "newError")
}
func TestErrorCatchNonErr(t *testing.T) {
var err error
func() {
defer ErrorCatch(&err)
panic(11)
}()
assert.Error(t, err, "11")
}

View file

@ -15,22 +15,22 @@ const genTestDirRoot = "./.gentestdata3"
const genTestDir3 = "./.gentestdata3/mysql"
func TestGenerator(t *testing.T) {
err := os.RemoveAll(genTestDir3)
assert.NilError(t, err)
err = mysql.Generate(genTestDir3, mysql.DBConnection{
Host: dbconfig.MySqLHost,
Port: dbconfig.MySQLPort,
User: dbconfig.MySQLUser,
Password: dbconfig.MySQLPassword,
DBName: "dvds",
})
for i := 0; i < 3; i++ {
err := mysql.Generate(genTestDir3, mysql.DBConnection{
Host: dbconfig.MySqLHost,
Port: dbconfig.MySQLPort,
User: dbconfig.MySQLUser,
Password: dbconfig.MySQLPassword,
DBName: "dvds",
})
assert.NilError(t, err)
assert.NilError(t, err)
assertGeneratedFiles(t)
assertGeneratedFiles(t)
}
err = os.RemoveAll(genTestDirRoot)
err := os.RemoveAll(genTestDirRoot)
assert.NilError(t, err)
}

View file

@ -71,27 +71,26 @@ func TestCmdGenerator(t *testing.T) {
func TestGenerator(t *testing.T) {
for i := 0; i < 3; i++ {
err := postgres.Generate(genTestDir2, postgres.DBConnection{
Host: dbconfig.Host,
Port: dbconfig.Port,
User: dbconfig.User,
Password: dbconfig.Password,
SslMode: "disable",
Params: "",
DBName: dbconfig.DBName,
SchemaName: "dvds",
})
assert.NilError(t, err)
assertGeneratedFiles(t)
}
err := os.RemoveAll(genTestDir2)
assert.NilError(t, err)
err = postgres.Generate(genTestDir2, postgres.DBConnection{
Host: dbconfig.Host,
Port: dbconfig.Port,
User: dbconfig.User,
Password: dbconfig.Password,
SslMode: "disable",
Params: "",
DBName: dbconfig.DBName,
SchemaName: "dvds",
})
assert.NilError(t, err)
assertGeneratedFiles(t)
err = os.RemoveAll(genTestDir2)
assert.NilError(t, err)
}
func assertGeneratedFiles(t *testing.T) {