Lunny Xiao
109cb1a7d0
* add support group engine * revert code * add NewGroup function * add engine group policy * rename file name * modify policy interface * remove Init function from policy interface * refactor Group Policy * rename and comments * rename and bug fix for WeightRoundRobinPolicy * modify Slave function * modify Slave function and add LeastConnPolicy * use original Engine and Session * remove unused count variables * fix bug on NewEngineGroup * remove unused method * improve range and refactor * add some comments and refactor * implement GroupPolicy of GroupPolicyHandler * refactor * simple code * add tests support for EngineGroup & fix some bugs * improve the NewEngineGroup interface * change the default policy of engine group * fix some tests
70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
// Copyright 2017 The Xorm Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package xorm
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/go-xorm/core"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSetExpr(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type UserExpr struct {
|
|
Id int64
|
|
Show bool
|
|
}
|
|
|
|
assert.NoError(t, testEngine.Sync2(new(UserExpr)))
|
|
|
|
cnt, err := testEngine.Insert(&UserExpr{
|
|
Show: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
|
|
var not = "NOT"
|
|
if testEngine.Dialect().DBType() == core.MSSQL {
|
|
not = "~"
|
|
}
|
|
cnt, err = testEngine.SetExpr("show", not+" `show`").ID(1).Update(new(UserExpr))
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
}
|
|
|
|
func TestCols(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type ColsTable struct {
|
|
Id int64
|
|
Col1 string
|
|
Col2 string
|
|
}
|
|
|
|
assertSync(t, new(ColsTable))
|
|
|
|
_, err := testEngine.Insert(&ColsTable{
|
|
Col1: "1",
|
|
Col2: "2",
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
sess := testEngine.ID(1)
|
|
_, err = sess.Cols("col1").Cols("col2").Update(&ColsTable{
|
|
Col1: "",
|
|
Col2: "",
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
var tb ColsTable
|
|
has, err := testEngine.ID(1).Get(&tb)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, "", tb.Col1)
|
|
assert.EqualValues(t, "", tb.Col2)
|
|
}
|