63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
package mysql
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func TestUpdateWithOneValue(t *testing.T) {
|
|
expectedSQL := `
|
|
UPDATE db.table1
|
|
SET col_int = ?
|
|
WHERE table1.col_int >= ?;
|
|
`
|
|
stmt := table1.UPDATE(table1ColInt).
|
|
SET(1).
|
|
WHERE(table1ColInt.GT_EQ(Int(33)))
|
|
|
|
fmt.Println(stmt.Sql())
|
|
|
|
assertStatement(t, stmt, expectedSQL, 1, int64(33))
|
|
}
|
|
|
|
func TestUpdateWithValues(t *testing.T) {
|
|
expectedSQL := `
|
|
UPDATE db.table1
|
|
SET col_int = ?,
|
|
col_float = ?
|
|
WHERE table1.col_int >= ?;
|
|
`
|
|
stmt := table1.UPDATE(table1ColInt, table1ColFloat).
|
|
SET(1, 22.2).
|
|
WHERE(table1ColInt.GT_EQ(Int(33)))
|
|
|
|
fmt.Println(stmt.Sql())
|
|
|
|
assertStatement(t, stmt, expectedSQL, 1, 22.2, int64(33))
|
|
}
|
|
|
|
func TestUpdateOneColumnWithSelect(t *testing.T) {
|
|
expectedSQL := `
|
|
UPDATE db.table1
|
|
SET col_float = (
|
|
SELECT table1.col_float AS "table1.col_float"
|
|
FROM db.table1
|
|
)
|
|
WHERE table1.col1 = ?;
|
|
`
|
|
stmt := table1.
|
|
UPDATE(table1ColFloat).
|
|
SET(
|
|
table1.SELECT(table1ColFloat),
|
|
).
|
|
WHERE(table1Col1.EQ(Int(2)))
|
|
|
|
//fmt.Println(stmt.Sql())
|
|
|
|
assertStatement(t, stmt, expectedSQL, int64(2))
|
|
}
|
|
|
|
func TestInvalidInputs(t *testing.T) {
|
|
assertStatementErr(t, table1.UPDATE(table1ColInt).SET(1), "jet: WHERE clause not set")
|
|
assertStatementErr(t, table1.UPDATE(nil).SET(1), "jet: nil column in columns list")
|
|
}
|