Add more unit tests.
This commit is contained in:
parent
74b94f0d8b
commit
65e05021bc
3 changed files with 43 additions and 32 deletions
|
|
@ -25,7 +25,7 @@ func TestToGoIdentifier(t *testing.T) {
|
||||||
assert.Equal(t, ToGoIdentifier("My-Table"), "MyTable")
|
assert.Equal(t, ToGoIdentifier("My-Table"), "MyTable")
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestErrorCatch(t *testing.T) {
|
func TestErrorCatchErr(t *testing.T) {
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
func() {
|
func() {
|
||||||
|
|
@ -36,3 +36,15 @@ func TestErrorCatch(t *testing.T) {
|
||||||
|
|
||||||
assert.Error(t, err, "newError")
|
assert.Error(t, err, "newError")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestErrorCatchNonErr(t *testing.T) {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
func() {
|
||||||
|
defer ErrorCatch(&err)
|
||||||
|
|
||||||
|
panic(11)
|
||||||
|
}()
|
||||||
|
|
||||||
|
assert.Error(t, err, "11")
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -15,22 +15,22 @@ const genTestDirRoot = "./.gentestdata3"
|
||||||
const genTestDir3 = "./.gentestdata3/mysql"
|
const genTestDir3 = "./.gentestdata3/mysql"
|
||||||
|
|
||||||
func TestGenerator(t *testing.T) {
|
func TestGenerator(t *testing.T) {
|
||||||
err := os.RemoveAll(genTestDir3)
|
|
||||||
assert.NilError(t, err)
|
|
||||||
|
|
||||||
err = mysql.Generate(genTestDir3, mysql.DBConnection{
|
for i := 0; i < 3; i++ {
|
||||||
Host: dbconfig.MySqLHost,
|
err := mysql.Generate(genTestDir3, mysql.DBConnection{
|
||||||
Port: dbconfig.MySQLPort,
|
Host: dbconfig.MySqLHost,
|
||||||
User: dbconfig.MySQLUser,
|
Port: dbconfig.MySQLPort,
|
||||||
Password: dbconfig.MySQLPassword,
|
User: dbconfig.MySQLUser,
|
||||||
DBName: "dvds",
|
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)
|
assert.NilError(t, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -71,27 +71,26 @@ func TestCmdGenerator(t *testing.T) {
|
||||||
|
|
||||||
func TestGenerator(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)
|
err := os.RemoveAll(genTestDir2)
|
||||||
assert.NilError(t, err)
|
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) {
|
func assertGeneratedFiles(t *testing.T) {
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue