Add test case
This commit is contained in:
parent
cebdacd514
commit
57d7b9b042
|
@ -5,6 +5,7 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/go-xorm/core"
|
"github.com/go-xorm/core"
|
||||||
|
"github.com/jackc/pgx/stdlib"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestParsePostgres(t *testing.T) {
|
func TestParsePostgres(t *testing.T) {
|
||||||
|
@ -38,3 +39,48 @@ func TestParsePostgres(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestParsePgx(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
in string
|
||||||
|
expected string
|
||||||
|
valid bool
|
||||||
|
}{
|
||||||
|
{"postgres://auser:password@localhost:5432/db?sslmode=disable", "db", true},
|
||||||
|
{"postgresql://auser:password@localhost:5432/db?sslmode=disable", "db", true},
|
||||||
|
{"postg://auser:password@localhost:5432/db?sslmode=disable", "db", false},
|
||||||
|
//{"postgres://auser:pass with space@localhost:5432/db?sslmode=disable", "db", true},
|
||||||
|
//{"postgres:// auser : password@localhost:5432/db?sslmode=disable", "db", true},
|
||||||
|
{"postgres://%20auser%20:pass%20with%20space@localhost:5432/db?sslmode=disable", "db", true},
|
||||||
|
//{"postgres://auser:パスワード@localhost:5432/データベース?sslmode=disable", "データベース", true},
|
||||||
|
{"dbname=db sslmode=disable", "db", true},
|
||||||
|
{"user=auser password=password dbname=db sslmode=disable", "db", true},
|
||||||
|
{"", "db", false},
|
||||||
|
{"dbname=db =disable", "db", false},
|
||||||
|
}
|
||||||
|
|
||||||
|
driver := core.QueryDriver("pgx")
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
uri, err := driver.Parse("pgx", test.in)
|
||||||
|
|
||||||
|
if err != nil && test.valid {
|
||||||
|
t.Errorf("%q got unexpected error: %s", test.in, err)
|
||||||
|
} else if err == nil && !reflect.DeepEqual(test.expected, uri.DbName) {
|
||||||
|
t.Errorf("%q got: %#v want: %#v", test.in, uri.DbName, test.expected)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Register DriverConfig
|
||||||
|
drvierConfig := stdlib.DriverConfig{}
|
||||||
|
stdlib.RegisterDriverConfig(drvierConfig)
|
||||||
|
uri, err = driver.Parse("pgx",
|
||||||
|
drvierConfig.ConnectionString(test.in))
|
||||||
|
if err != nil && test.valid {
|
||||||
|
t.Errorf("%q got unexpected error: %s", test.in, err)
|
||||||
|
} else if err == nil && !reflect.DeepEqual(test.expected, uri.DbName) {
|
||||||
|
t.Errorf("%q got: %#v want: %#v", test.in, uri.DbName, test.expected)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue