HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix setschema

This commit is contained in:
Lunny Xiao 2020-03-17 10:22:08 +08:00
parent b7b2b21a40
commit 5f616ccc5e
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
2 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dialects/dialect.go
View File

@ -31,7 +31,7 @@ type URI struct {
} }
// SetSchema set schema // SetSchema set schema
func (uri URI) SetSchema(schema string) { func (uri *URI) SetSchema(schema string) {
if uri.DBType == schemas.POSTGRES { if uri.DBType == schemas.POSTGRES {
uri.Schema = schema uri.Schema = schema
} }

13
engine_test.go
View File

@ -12,6 +12,7 @@ import (
"time" "time"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"xorm.io/xorm/schemas"
) )
func TestPingContext(t *testing.T) { func TestPingContext(t *testing.T) {
@ -97,3 +98,15 @@ func TestDump(t *testing.T) {
assert.NoError(t, err) assert.NoError(t, err)
assert.NoError(t, sess.Commit()) assert.NoError(t, sess.Commit())
} }
func TestSetSchema(t *testing.T) {
assert.NoError(t, prepareEngine())
if testEngine.Dialect().URI().DBType == schemas.POSTGRES {
oldSchema := testEngine.Dialect().URI().Schema
testEngine.SetSchema("my_schema")
assert.EqualValues(t, "my_schema", testEngine.Dialect().URI().Schema)
testEngine.SetSchema(oldSchema)
assert.EqualValues(t, oldSchema, testEngine.Dialect().URI().Schema)
}
}