28 lines
549 B
Go
28 lines
549 B
Go
package sqlbuilder
|
|
|
|
import (
|
|
"gotest.tools/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestLockSingleTable(t *testing.T) {
|
|
lock := table1.LOCK().IN(LOCK_ROW_SHARE)
|
|
|
|
queryStr, _, err := lock.Sql()
|
|
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, queryStr, `
|
|
LOCK TABLE db.table1 IN ROW SHARE MODE;
|
|
`)
|
|
}
|
|
|
|
func TestLockMultipleTable(t *testing.T) {
|
|
lock := LOCK(table2, table1).IN(LOCK_ACCESS_EXCLUSIVE).NOWAIT()
|
|
|
|
queryStr, _, err := lock.Sql()
|
|
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, queryStr, `
|
|
LOCK TABLE db.table2, db.table1 IN ACCESS EXCLUSIVE MODE NOWAIT;
|
|
`)
|
|
}
|