34 lines
904 B
Go
34 lines
904 B
Go
|
package sqlbuilder
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gotest.tools/assert"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCase1(t *testing.T) {
|
||
|
|
query := CASE().
|
||
|
|
WHEN(table3Col1.EqL(1)).THEN(table3Col1.Add(IntLiteral(1))).
|
||
|
|
WHEN(table3Col1.EqL(2)).THEN(table3Col1.Add(IntLiteral(2)))
|
||
|
|
|
||
|
|
queryData := &queryData{}
|
||
|
|
|
||
|
|
err := query.Serialize(queryData)
|
||
|
|
|
||
|
|
assert.NilError(t, err)
|
||
|
|
assert.Equal(t, queryData.buff.String(), `(CASE WHEN table3.col1 = $1 THEN table3.col1 + $2 WHEN table3.col1 = $3 THEN table3.col1 + $4 END)`)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCase2(t *testing.T) {
|
||
|
|
query := CASE(table3Col1).
|
||
|
|
WHEN(IntLiteral(1)).THEN(table3Col1.Add(IntLiteral(1))).
|
||
|
|
WHEN(IntLiteral(2)).THEN(table3Col1.Add(IntLiteral(2))).
|
||
|
|
ELSE(IntLiteral(0))
|
||
|
|
|
||
|
|
queryData := &queryData{}
|
||
|
|
|
||
|
|
err := query.Serialize(queryData)
|
||
|
|
|
||
|
|
assert.NilError(t, err)
|
||
|
|
assert.Equal(t, queryData.buff.String(), `(CASE table3.col1 WHEN $1 THEN table3.col1 + $2 WHEN $3 THEN table3.col1 + $4 ELSE $5 END)`)
|
||
|
|
}
|