You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
278 lines
12 KiB
278 lines
12 KiB
// Copyright 2019 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 builder
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestBuilder_Union(t *testing.T) {
|
|
sql, args, err := Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Union("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Union("distinct", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
Union("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) UNION ALL (SELECT * FROM t2 WHERE status=?) UNION DISTINCT (SELECT * FROM t2 WHERE status=?) UNION (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3", "3"}, args)
|
|
|
|
// sub-query will inherit dialect from the main one
|
|
sql, args, err = MySQL().Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Union("all", Select("*").From("t2").Where(Eq{"status": "2"}).Limit(10)).
|
|
Union("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) UNION ALL (SELECT * FROM t2 WHERE status=? LIMIT 10) UNION (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3"}, args)
|
|
|
|
// will raise error
|
|
_, _, err = MySQL().Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Union("all", Oracle().Select("*").From("t2").Where(Eq{"status": "2"}).Limit(10)).
|
|
ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrInconsistentDialect, err)
|
|
|
|
// will raise error
|
|
_, _, err = Select("*").From("table1").Where(Eq{"a": "1"}).
|
|
Union("all", Select("*").From("table2").Where(Eq{"a": "2"})).
|
|
Where(Eq{"a": 2}).Limit(5, 10).
|
|
ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrNotUnexpectedUnionConditions, err)
|
|
|
|
// will raise error
|
|
_, _, err = Delete(Eq{"a": 1}).From("t1").
|
|
Union("all", Select("*").From("t2").Where(Eq{"status": "2"})).ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrUnsupportedUnionMembers, err)
|
|
|
|
// will be overwrote by SELECT op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Union("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Select("*").From("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
|
|
// will be overwrote by DELETE op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Union("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Delete(Eq{"status": "1"}).From("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
|
|
// will be overwrote by INSERT op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Union("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Insert(Eq{"status": "1"}).Into("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
}
|
|
|
|
func TestBuilder_Intersect(t *testing.T) {
|
|
sql, args, err := Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Intersect("distinct", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
Intersect("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) INTERSECT ALL (SELECT * FROM t2 WHERE status=?) INTERSECT DISTINCT (SELECT * FROM t2 WHERE status=?) INTERSECT (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3", "3"}, args)
|
|
|
|
// sub-query will inherit dialect from the main one
|
|
sql, args, err = MySQL().Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"}).Limit(10)).
|
|
Intersect("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) INTERSECT ALL (SELECT * FROM t2 WHERE status=? LIMIT 10) INTERSECT (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3"}, args)
|
|
|
|
// will raise error
|
|
_, _, err = MySQL().Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Oracle().Select("*").From("t2").Where(Eq{"status": "2"}).Limit(10)).
|
|
ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrInconsistentDialect, err)
|
|
|
|
// will raise error
|
|
_, _, err = Select("*").From("table1").Where(Eq{"a": "1"}).
|
|
Intersect("all", Select("*").From("table2").Where(Eq{"a": "2"})).
|
|
Where(Eq{"a": 2}).Limit(5, 10).
|
|
ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrNotUnexpectedUnionConditions, err)
|
|
|
|
// will raise error
|
|
_, _, err = Delete(Eq{"a": 1}).From("t1").
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrUnsupportedUnionMembers, err)
|
|
|
|
// will be overwrote by SELECT op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Select("*").From("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
|
|
// will be overwrote by DELETE op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Delete(Eq{"status": "1"}).From("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
|
|
// will be overwrote by INSERT op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Insert(Eq{"status": "1"}).Into("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
}
|
|
|
|
func TestBuilder_Except(t *testing.T) {
|
|
sql, args, err := Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Except("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Except("distinct", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
Except("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) EXCEPT ALL (SELECT * FROM t2 WHERE status=?) EXCEPT DISTINCT (SELECT * FROM t2 WHERE status=?) EXCEPT (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3", "3"}, args)
|
|
|
|
// sub-query will inherit dialect from the main one
|
|
sql, args, err = MySQL().Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Except("all", Select("*").From("t2").Where(Eq{"status": "2"}).Limit(10)).
|
|
Except("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) EXCEPT ALL (SELECT * FROM t2 WHERE status=? LIMIT 10) EXCEPT (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3"}, args)
|
|
|
|
// will raise error
|
|
_, _, err = MySQL().Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Except("all", Oracle().Select("*").From("t2").Where(Eq{"status": "2"}).Limit(10)).
|
|
ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrInconsistentDialect, err)
|
|
|
|
// will raise error
|
|
_, _, err = Select("*").From("table1").Where(Eq{"a": "1"}).
|
|
Except("all", Select("*").From("table2").Where(Eq{"a": "2"})).
|
|
Where(Eq{"a": 2}).Limit(5, 10).
|
|
ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrNotUnexpectedUnionConditions, err)
|
|
|
|
// will raise error
|
|
_, _, err = Delete(Eq{"a": 1}).From("t1").
|
|
Except("all", Select("*").From("t2").Where(Eq{"status": "2"})).ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrUnsupportedUnionMembers, err)
|
|
|
|
// will be overwrote by SELECT op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Except("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Select("*").From("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
|
|
// will be overwrote by DELETE op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Except("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Delete(Eq{"status": "1"}).From("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
|
|
// will be overwrote by INSERT op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Except("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Insert(Eq{"status": "1"}).Into("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
}
|
|
|
|
func TestBuilder_SetOperations(t *testing.T) {
|
|
sql, args, err := Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Union("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Intersect("distinct", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
Except("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) UNION ALL (SELECT * FROM t2 WHERE status=?) INTERSECT DISTINCT (SELECT * FROM t2 WHERE status=?) EXCEPT (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3", "3"}, args)
|
|
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Union("distinct", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
Except("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) INTERSECT ALL (SELECT * FROM t2 WHERE status=?) UNION DISTINCT (SELECT * FROM t2 WHERE status=?) EXCEPT (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3", "3"}, args)
|
|
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Except("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Intersect("distinct", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
Union("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) EXCEPT ALL (SELECT * FROM t2 WHERE status=?) INTERSECT DISTINCT (SELECT * FROM t2 WHERE status=?) UNION (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3", "3"}, args)
|
|
|
|
// sub-query will inherit dialect from the main one
|
|
sql, args, err = MySQL().Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"}).Limit(10)).
|
|
Intersect("", Select("*").From("t2").Where(Eq{"status": "3"})).
|
|
ToSQL()
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, "(SELECT * FROM t1 WHERE status=?) INTERSECT ALL (SELECT * FROM t2 WHERE status=? LIMIT 10) INTERSECT (SELECT * FROM t2 WHERE status=?)", sql)
|
|
assert.EqualValues(t, []interface{}{"1", "2", "3"}, args)
|
|
|
|
// will raise error
|
|
_, _, err = MySQL().Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Oracle().Select("*").From("t2").Where(Eq{"status": "2"}).Limit(10)).
|
|
ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrInconsistentDialect, err)
|
|
|
|
// will raise error
|
|
_, _, err = Select("*").From("table1").Where(Eq{"a": "1"}).
|
|
Intersect("all", Select("*").From("table2").Where(Eq{"a": "2"})).
|
|
Where(Eq{"a": 2}).Limit(5, 10).
|
|
ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrNotUnexpectedUnionConditions, err)
|
|
|
|
// will raise error
|
|
_, _, err = Delete(Eq{"a": 1}).From("t1").
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).ToSQL()
|
|
assert.Error(t, err)
|
|
assert.EqualValues(t, ErrUnsupportedUnionMembers, err)
|
|
|
|
// will be overwrote by SELECT op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Select("*").From("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
|
|
// will be overwrote by DELETE op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Delete(Eq{"status": "1"}).From("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
|
|
// will be overwrote by INSERT op
|
|
sql, args, err = Select("*").From("t1").Where(Eq{"status": "1"}).
|
|
Intersect("all", Select("*").From("t2").Where(Eq{"status": "2"})).
|
|
Insert(Eq{"status": "1"}).Into("t2").ToSQL()
|
|
assert.NoError(t, err)
|
|
fmt.Println(sql, args)
|
|
}
|