Cloning files from https://github.com/dropbox/godropbox/tree/master/database
This commit is contained in:
sub0Zero
zer0sub
parent
9adb801298
commit
3190d6f933
12 changed files with 4162 additions and 0 deletions
660
sqlbuilder/statement_test.go
Normal file
660
sqlbuilder/statement_test.go
Normal file
|
|
@ -0,0 +1,660 @@
|
|||
package sqlbuilder
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
gc "gopkg.in/check.v1"
|
||||
|
||||
"github.com/dropbox/godropbox/errors"
|
||||
)
|
||||
|
||||
type StmtSuite struct {
|
||||
}
|
||||
|
||||
var _ = gc.Suite(&StmtSuite{})
|
||||
|
||||
// NOTE: tables / columns are defined in test_utils.go
|
||||
|
||||
//
|
||||
// SELECT statement tests
|
||||
//
|
||||
|
||||
func (s *StmtSuite) TestSelectEmptyProjection(c *gc.C) {
|
||||
_, err := table1.Select().String("db")
|
||||
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectSingleColumn(c *gc.C) {
|
||||
sql, err := table1.Select(table1Col1).String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1`")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectMultiColumns(c *gc.C) {
|
||||
sql, err := table1.Select(table1Col1, table1Col2).String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1`,`table1`.`col2` FROM `db`.`table1`")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectWhere(c *gc.C) {
|
||||
q := table1.Select(table1Col1).Where(GtL(table1Col1, 123))
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1` WHERE `table1`.`col1`>123")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectWhereDate(c *gc.C) {
|
||||
date := time.Date(1999, 1, 2, 3, 4, 5, 0, time.UTC)
|
||||
|
||||
q := table1.Select(table1Col1).Where(GtL(table1Col4, date))
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1` "+
|
||||
"WHERE `table1`.`col4`>'1999-01-02 03:04:05.000000'")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectAndWhere(c *gc.C) {
|
||||
q := table1.Select(table1Col1).AndWhere(GtL(table1Col1, 123))
|
||||
q.AndWhere(LtL(table1Col1, 321))
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1` WHERE (`table1`.`col1`>123 AND `table1`.`col1`<321)")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectCopy(c *gc.C) {
|
||||
q := table1.Select(table1Col1).Where(GtL(table1Col1, 123))
|
||||
qq := q.Copy().Where(GtL(table1Col1, 321)).OrderBy(table1Col1)
|
||||
|
||||
// Initial query unchanged
|
||||
sql, err := q.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1` WHERE `table1`.`col1`>123")
|
||||
// New query changed
|
||||
sql, err = qq.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1` WHERE `table1`.`col1`>321 ORDER BY `table1`.`col1`")
|
||||
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectLimitWithoutOffset(c *gc.C) {
|
||||
q := table1.Select(table1Col1).Limit(5)
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1` LIMIT 5")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectLimitWithOffset(c *gc.C) {
|
||||
q := table1.Select(table1Col1).Limit(5).Offset(2)
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1` LIMIT 2, 5")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectGroupBy(c *gc.C) {
|
||||
q := table1.Select(
|
||||
table1Col1,
|
||||
table1Col2,
|
||||
Alias("total", SqlFunc("sum", table1Col3)))
|
||||
q.GroupBy(table1Col1, table1Col2)
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1`,`table1`.`col2`,"+
|
||||
"(sum(`table1`.`col3`)) AS `total` "+
|
||||
"FROM `db`.`table1` GROUP BY `table1`.`col1`,`table1`.`col2`")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectSingleOrderBy(c *gc.C) {
|
||||
q := table1.Select(table1Col1, table1Col2).OrderBy(table1Col2)
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1`,`table1`.`col2` "+
|
||||
"FROM `db`.`table1` ORDER BY `table1`.`col2`")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectOrderByAsc(c *gc.C) {
|
||||
q := table1.Select(table1Col1, table1Col2).OrderBy(Asc(table1Col2))
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1`,`table1`.`col2` "+
|
||||
"FROM `db`.`table1` ORDER BY `table1`.`col2` ASC")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectOrderByDesc(c *gc.C) {
|
||||
q := table1.Select(table1Col1, table1Col2).OrderBy(Desc(table1Col2))
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1`,`table1`.`col2` "+
|
||||
"FROM `db`.`table1` ORDER BY `table1`.`col2` DESC")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectMultiOrderBy(c *gc.C) {
|
||||
q := table1.Select(table1Col1, table1Col2)
|
||||
q.OrderBy(table1Col2, table1Col1)
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1`,`table1`.`col2` "+
|
||||
"FROM `db`.`table1` "+
|
||||
"ORDER BY `table1`.`col2`,`table1`.`col1`")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectOnJoin(c *gc.C) {
|
||||
|
||||
join := table1.InnerJoinOn(table2, Eq(table1Col3, table2Col3))
|
||||
sql, err := join.Select(table1Col1, table2Col4).String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1`,`table2`.`col4` "+
|
||||
"FROM `db`.`table1` JOIN `db`.`table2` "+
|
||||
"ON `table1`.`col3`=`table2`.`col3`")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectWithSharedLock(c *gc.C) {
|
||||
|
||||
q := table1.Select(table1Col1).Where(GtL(table1Col1, 123)).WithSharedLock()
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT `table1`.`col1` FROM `db`.`table1` "+
|
||||
"WHERE `table1`.`col1`>123 LOCK IN SHARE MODE")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestSelectDistinct(c *gc.C) {
|
||||
q := table1.Select(table1Col1).Distinct()
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"SELECT DISTINCT `table1`.`col1` FROM `db`.`table1`")
|
||||
}
|
||||
|
||||
//
|
||||
// INSERT statement tests
|
||||
//
|
||||
|
||||
func (s *StmtSuite) TestInsertNoColumn(c *gc.C) {
|
||||
_, err := table1.Insert().Add().String("db")
|
||||
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertNoRow(c *gc.C) {
|
||||
_, err := table1.Insert(table1Col1).String("db")
|
||||
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertColumnLengthMismatch(c *gc.C) {
|
||||
_, err := table1.Insert(table1Col1, table1Col2).Add(nil).String("db")
|
||||
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertNilValue(c *gc.C) {
|
||||
_, err := table1.Insert(table1Col1).Add(nil).String("db")
|
||||
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertNilColumn(c *gc.C) {
|
||||
_, err := table1.Insert(nil).Add(Literal(1)).String("db")
|
||||
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertSingleValue(c *gc.C) {
|
||||
sql, err := table1.Insert(table1Col1).Add(Literal(1)).String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"INSERT INTO `db`.`table1` (`table1`.`col1`) VALUES (1)")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertDate(c *gc.C) {
|
||||
date := time.Date(1999, 1, 2, 3, 4, 5, 0, time.UTC)
|
||||
|
||||
sql, err := table1.Insert(table1Col4).Add(Literal(date)).String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"INSERT INTO `db`.`table1` (`table1`.`col4`) "+
|
||||
"VALUES ('1999-01-02 03:04:05.000000')")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertIgnore(c *gc.C) {
|
||||
stmt := table1.Insert(table1Col1).Add(Literal(1)).IgnoreDuplicates(true)
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"INSERT IGNORE INTO `db`.`table1` (`table1`.`col1`) VALUES (1)")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertMultipleValues(c *gc.C) {
|
||||
stmt := table1.Insert(table1Col1, table1Col2, table1Col3)
|
||||
stmt.Add(Literal(1), Literal(2), Literal(3))
|
||||
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"INSERT INTO `db`.`table1` "+
|
||||
"(`table1`.`col1`,`table1`.`col2`,`table1`.`col3`) "+
|
||||
"VALUES (1,2,3)")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestInsertMultipleRows(c *gc.C) {
|
||||
stmt := table1.Insert(table1Col1, table1Col2)
|
||||
stmt.Add(Literal(1), Literal(2))
|
||||
stmt.Add(Literal(11), Literal(22))
|
||||
stmt.Add(Literal(111), Literal(222))
|
||||
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"INSERT INTO `db`.`table1` "+
|
||||
"(`table1`.`col1`,`table1`.`col2`) "+
|
||||
"VALUES (1,2), (11,22), (111,222)")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestOnDuplicateKeyUpdateNilCol(c *gc.C) {
|
||||
stmt := table1.Insert(table1Col1, table1Col2)
|
||||
stmt.Add(Literal(1), Literal(2))
|
||||
stmt.AddOnDuplicateKeyUpdate(nil, Literal(3))
|
||||
|
||||
_, err := stmt.String("db")
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestOnDuplicateKeyUpdateNilExpr(c *gc.C) {
|
||||
stmt := table1.Insert(table1Col1, table1Col2)
|
||||
stmt.Add(Literal(1), Literal(2))
|
||||
stmt.AddOnDuplicateKeyUpdate(table1Col1, nil)
|
||||
|
||||
_, err := stmt.String("db")
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestOnDuplicateKeyUpdateSingle(c *gc.C) {
|
||||
stmt := table1.Insert(table1Col1, table1Col2)
|
||||
stmt.Add(Literal(1), Literal(2))
|
||||
stmt.AddOnDuplicateKeyUpdate(table1Col3, Literal(3))
|
||||
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"INSERT INTO `db`.`table1` "+
|
||||
"(`table1`.`col1`,`table1`.`col2`) "+
|
||||
"VALUES (1,2) "+
|
||||
"ON DUPLICATE KEY UPDATE `table1`.`col3`=3")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestOnDuplicateKeyUpdateMulti(c *gc.C) {
|
||||
stmt := table1.Insert(table1Col1, table1Col2)
|
||||
stmt.Add(Literal(1), Literal(2))
|
||||
stmt.AddOnDuplicateKeyUpdate(table1Col3, Literal(3))
|
||||
stmt.AddOnDuplicateKeyUpdate(table1Col2, Literal(4))
|
||||
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"INSERT INTO `db`.`table1` "+
|
||||
"(`table1`.`col1`,`table1`.`col2`) "+
|
||||
"VALUES (1,2) "+
|
||||
"ON DUPLICATE KEY UPDATE `table1`.`col3`=3, `table1`.`col2`=4")
|
||||
}
|
||||
|
||||
//
|
||||
// UPDATE statement tests =====================================================
|
||||
//
|
||||
|
||||
func (s *StmtSuite) TestUpdateNilColumn(c *gc.C) {
|
||||
stmt := table1.Update().Set(nil, Literal(1))
|
||||
_, err := stmt.String("db")
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUpdateNilExpr(c *gc.C) {
|
||||
stmt := table1.Update().Set(table1Col1, nil)
|
||||
_, err := stmt.String("db")
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUpdateUnconditionally(c *gc.C) {
|
||||
stmt := table1.Update().Set(table1Col1, Literal(1))
|
||||
_, err := stmt.String("db")
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUpdateSingleValue(c *gc.C) {
|
||||
stmt := table1.Update().Set(table1Col1, Literal(1))
|
||||
stmt.Where(EqL(table1Col2, 2))
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"UPDATE `db`.`table1` SET `table1`.`col1`=1 WHERE `table1`.`col2`=2")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUpdateUsingDeferredLookupColumns(c *gc.C) {
|
||||
stmt := table1.Update().Set(table1.C("col1"), Literal(1))
|
||||
stmt.Where(EqL(table1Col2, 2))
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"UPDATE `db`.`table1` SET `table1`.`col1`=1 WHERE `table1`.`col2`=2")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUpdateMultiValues(c *gc.C) {
|
||||
stmt := table1.Update()
|
||||
stmt.Set(table1Col1, Literal(1))
|
||||
stmt.Set(table1Col2, Literal(2))
|
||||
stmt.Where(EqL(table1Col2, 3))
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"UPDATE `db`.`table1` "+
|
||||
"SET `table1`.`col1`=1, `table1`.`col2`=2 "+
|
||||
"WHERE `table1`.`col2`=3")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUpdateWithOrderBy(c *gc.C) {
|
||||
stmt := table1.Update().Set(table1Col1, Literal(1))
|
||||
stmt.Where(EqL(table1Col2, 2))
|
||||
stmt.OrderBy(table1Col2)
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"UPDATE `db`.`table1` "+
|
||||
"SET `table1`.`col1`=1 "+
|
||||
"WHERE `table1`.`col2`=2 "+
|
||||
"ORDER BY `table1`.`col2`")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUpdateWithLimit(c *gc.C) {
|
||||
stmt := table1.Update().Set(table1Col1, Literal(1))
|
||||
stmt.Where(EqL(table1Col2, 2))
|
||||
stmt.Limit(5)
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"UPDATE `db`.`table1` "+
|
||||
"SET `table1`.`col1`=1 "+
|
||||
"WHERE `table1`.`col2`=2 "+
|
||||
"LIMIT 5")
|
||||
}
|
||||
|
||||
//
|
||||
// DELETE statement tests =====================================================
|
||||
//
|
||||
|
||||
func (s *StmtSuite) TestDeleteUnconditionally(c *gc.C) {
|
||||
_, err := table1.Delete().String("db")
|
||||
c.Assert(err, gc.NotNil)
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestDeleteWithWhere(c *gc.C) {
|
||||
sql, err := table1.Delete().Where(EqL(table1Col1, 1)).String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"DELETE FROM `db`.`table1` WHERE `table1`.`col1`=1")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestDeleteWithOrderBy(c *gc.C) {
|
||||
stmt := table1.Delete().Where(EqL(table1Col1, 1)).OrderBy(table1Col1)
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"DELETE FROM `db`.`table1` "+
|
||||
"WHERE `table1`.`col1`=1 "+
|
||||
"ORDER BY `table1`.`col1`")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestDeleteWithLimit(c *gc.C) {
|
||||
stmt := table1.Delete().Where(EqL(table1Col1, 1)).Limit(5)
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"DELETE FROM `db`.`table1` WHERE `table1`.`col1`=1 LIMIT 5")
|
||||
}
|
||||
|
||||
//
|
||||
// LOCK/UNLOCK statement tests ================================================
|
||||
//
|
||||
|
||||
func (s *StmtSuite) TestLockStatement(c *gc.C) {
|
||||
stmt := NewLockStatement().AddReadLock(table1).AddWriteLock(table2)
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
|
||||
c.Assert(sql, gc.Equals, "LOCK TABLES `db`.`table1` READ, `db`.`table2` WRITE")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUnlockStatement(c *gc.C) {
|
||||
stmt := NewUnlockStatement()
|
||||
sql, err := stmt.String("db")
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(sql, gc.Equals, "UNLOCK TABLES")
|
||||
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUnionSelectStatement(c *gc.C) {
|
||||
select_queries := make([]SelectStatement, 0, 3)
|
||||
|
||||
select_queries = append(select_queries,
|
||||
table1.Select(table1Col1).Where(GtL(table1Col1, 123)),
|
||||
table1.Select(table1Col1).Where(GtL(table1Col1, 456)),
|
||||
table1.Select(table1Col1).Where(LtL(table1Col1, 23)),
|
||||
)
|
||||
|
||||
q := Union(select_queries...)
|
||||
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"(SELECT `table1`.`col1` FROM `db`.`table1` WHERE `table1`.`col1`>123) "+
|
||||
"UNION (SELECT `table1`.`col1` FROM `db`.`table1` WHERE `table1`.`col1`>456) "+
|
||||
"UNION (SELECT `table1`.`col1` FROM `db`.`table1` WHERE `table1`.`col1`<23)")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUnionLimitWithoutOrderBy(c *gc.C) {
|
||||
select_queries := make([]SelectStatement, 0, 3)
|
||||
|
||||
select_queries = append(select_queries,
|
||||
table1.Select(table1Col1).Where(GtL(table1Col1, 123)).OrderBy(table1Col2),
|
||||
table1.Select(table1Col1).Where(GtL(table1Col1, 456)),
|
||||
table1.Select(table1Col1).Where(LtL(table1Col1, 23)),
|
||||
)
|
||||
|
||||
q := Union(select_queries...)
|
||||
|
||||
_, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.NotNil)
|
||||
c.Assert(
|
||||
errors.GetMessage(err),
|
||||
gc.Equals,
|
||||
"All inner selects in Union statement must have LIMIT if they have ORDER BY")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestUnionSelectWithMismatchedColumns(c *gc.C) {
|
||||
select_queries := make([]SelectStatement, 0, 3)
|
||||
|
||||
select_queries = append(select_queries,
|
||||
|
||||
table1.Select(
|
||||
table1Col1,
|
||||
table1Col2,
|
||||
table1Col3,
|
||||
table1Col4).AndWhere(GtL(table1Col1, 123)).AndWhere(LtL(table1Col1, 321)),
|
||||
table1.Select(table1Col1).Where(And(GtL(table1Col1, 123), LtL(table1Col1, 321))),
|
||||
table1.Select(table1Col1).Where(LtL(table1Col1, 23)).OrderBy(table1Col4).Limit(20),
|
||||
)
|
||||
|
||||
q := Union(select_queries...)
|
||||
q = q.Where(And(LtL(table1Col1, 1000), GtL(table1Col1, 15)))
|
||||
q = q.OrderBy(Desc(table1Col4), Asc(table1Col3))
|
||||
q = q.Limit(5)
|
||||
|
||||
_, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.NotNil)
|
||||
c.Assert(
|
||||
errors.GetMessage(err),
|
||||
gc.Equals,
|
||||
"All inner selects in Union statement must select the "+
|
||||
"same number of columns. For sanity, you probably "+
|
||||
"want to select the same table columns in the same "+
|
||||
"order. If you are selecting on multiple tables, "+
|
||||
"use Null to pad to the right number of fields.")
|
||||
}
|
||||
|
||||
func (s *StmtSuite) TestComplicatedUnionSelectWithWhereStatement(c *gc.C) {
|
||||
|
||||
// tests on outer statement: Group By, Order By, Limit
|
||||
// on inner statement: AndWhere, Where (with And), Order By, Limit
|
||||
select_queries := make([]SelectStatement, 0, 3)
|
||||
|
||||
// We're not trying to write a SQL parser, so we won't warn if you do something silly like
|
||||
// try to apply a where clause on more columns than you've selected in your union select
|
||||
select_queries = append(select_queries,
|
||||
table1.Select(
|
||||
table1Col1,
|
||||
).AndWhere(GtL(table1Col1, 123)).AndWhere(LtL(table1Col1, 321)),
|
||||
table1.Select(
|
||||
table1Col1,
|
||||
).Where(And(GtL(table1Col1, 456), LtL(table1Col1, 654))),
|
||||
table1.Select(
|
||||
table1Col1,
|
||||
).Where(LtL(table1Col1, 23)).OrderBy(table1Col4).Limit(20),
|
||||
)
|
||||
|
||||
q := Union(select_queries...)
|
||||
q = q.Where(And(LtL(table1Col1, 1000), GtL(table1Col1, 15)))
|
||||
|
||||
q = q.OrderBy(Desc(table1Col4), Asc(table1Col3))
|
||||
q = q.Limit(5)
|
||||
q = q.GroupBy(table1Col4)
|
||||
|
||||
sql, err := q.String("db")
|
||||
|
||||
c.Assert(err, gc.IsNil)
|
||||
c.Assert(
|
||||
sql,
|
||||
gc.Equals,
|
||||
"(SELECT `table1`.`col1` FROM `db`.`table1` WHERE "+
|
||||
"(`table1`.`col1`>123 AND `table1`.`col1`<321)) "+
|
||||
"UNION (SELECT `table1`.`col1` FROM `db`.`table1` "+
|
||||
"WHERE (`table1`.`col1`>456 AND `table1`.`col1`<654)) "+
|
||||
"UNION (SELECT `table1`.`col1` FROM `db`.`table1` "+
|
||||
"WHERE `table1`.`col1`<23 ORDER BY `table1`.`col4` LIMIT 20) "+
|
||||
"WHERE (`table1`.`col1`<1000 AND `table1`.`col1`>15) "+
|
||||
"GROUP BY `table1`.`col4` ORDER BY `table1`.`col4` DESC,`table1`.`col3` ASC "+
|
||||
"LIMIT 5")
|
||||
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue