Adding upstream version 1.11.0.

Signed-off-by: Daniel Baumann <daniel@debian.org>

select_test.go

@@ -0,0 +1,179 @@
+// Copyright 2016 Qiang Xue. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package dbx
+
+import (
+	"testing"
+
+	"database/sql"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSelectQuery(t *testing.T) {
+	db := getDB()
+
+	// minimal select query
+	q := db.Select().From("users").Build()
+	expected := "SELECT * FROM `users`"
+	assert.Equal(t, q.SQL(), expected, "t1")
+	assert.Equal(t, len(q.Params()), 0, "t2")
+
+	// a full select query
+	q = db.Select("id", "name").
+		PreFragment("pre").
+		PostFragment("post").
+		AndSelect("age").
+		Distinct(true).
+		SelectOption("CALC").
+		From("users").
+		Where(NewExp("age>30")).
+		AndWhere(NewExp("status=1")).
+		OrWhere(NewExp("type=2")).
+		InnerJoin("profile", NewExp("user.id=profile.id")).
+		LeftJoin("team", nil).
+		RightJoin("dept", nil).
+		OrderBy("age DESC", "type").
+		AndOrderBy("id").
+		GroupBy("id").
+		AndGroupBy("age").
+		Having(NewExp("id>10")).
+		AndHaving(NewExp("id<20")).
+		OrHaving(NewExp("type=3")).
+		Limit(10).
+		Offset(20).
+		Bind(Params{"id": 1}).
+		AndBind(Params{"age": 30}).
+		Build()
+
+	expected = "pre SELECT DISTINCT CALC `id`, `name`, `age` FROM `users` INNER JOIN `profile` ON user.id=profile.id LEFT JOIN `team` RIGHT JOIN `dept` WHERE ((age>30) AND (status=1)) OR (type=2) GROUP BY `id`, `age` HAVING ((id>10) AND (id<20)) OR (type=3) ORDER BY `age` DESC, `type`, `id` LIMIT 10 OFFSET 20 post"
+	assert.Equal(t, q.SQL(), expected, "t3")
+	assert.Equal(t, len(q.Params()), 2, "t4")
+
+	q3 := db.Select().AndBind(Params{"id": 1}).Build()
+	assert.Equal(t, len(q3.Params()), 1)
+
+	// union
+	q1 := db.Select().From("users").PreFragment("pre_q1").Build()
+	q2 := db.Select().From("posts").PostFragment("post_q2").Build()
+	q = db.Select().From("profiles").Union(q1).UnionAll(q2).PreFragment("pre").PostFragment("post").Build()
+	expected = "(pre SELECT * FROM `profiles` post) UNION (pre_q1 SELECT * FROM `users`) UNION ALL (SELECT * FROM `posts` post_q2)"
+	assert.Equal(t, q.SQL(), expected, "t5")
+}
+
+func TestSelectQuery_Data(t *testing.T) {
+	db := getPreparedDB()
+	defer db.Close()
+
+	q := db.Select("id", "email").From("customer").OrderBy("id")
+
+	var customer Customer
+	q.One(&customer)
+	assert.Equal(t, customer.Email, "user1@example.com", "customer.Email")
+
+	var customers []Customer
+	q.All(&customers)
+	assert.Equal(t, len(customers), 3, "len(customers)")
+
+	rows, _ := q.Rows()
+	customer.Email = ""
+	rows.one(&customer)
+	assert.Equal(t, customer.Email, "user1@example.com", "customer.Email")
+
+	var id, email string
+	q.Row(&id, &email)
+	assert.Equal(t, id, "1", "id")
+	assert.Equal(t, email, "user1@example.com", "email")
+
+	var emails []string
+	err := db.Select("email").From("customer").Column(&emails)
+	if assert.Nil(t, err) {
+		assert.Equal(t, 3, len(emails))
+	}
+
+	var e int
+	err = db.Select().From("customer").One(&e)
+	assert.NotNil(t, err)
+	err = db.Select().From("customer").All(&e)
+	assert.NotNil(t, err)
+}
+
+func TestSelectQuery_Model(t *testing.T) {
+	db := getPreparedDB()
+	defer db.Close()
+
+	{
+		// One without specifying FROM
+		var customer CustomerPtr
+		err := db.Select().OrderBy("id").One(&customer)
+		if assert.Nil(t, err) {
+			assert.Equal(t, "user1@example.com", *customer.Email)
+		}
+	}
+
+	{
+		// All without specifying FROM
+		var customers []CustomerPtr
+		err := db.Select().OrderBy("id").All(&customers)
+		if assert.Nil(t, err) {
+			assert.Equal(t, 3, len(customers))
+		}
+	}
+
+	{
+		// Model without specifying FROM
+		var customer CustomerPtr
+		err := db.Select().Model(2, &customer)
+		if assert.Nil(t, err) {
+			assert.Equal(t, "user2@example.com", *customer.Email)
+		}
+	}
+
+	{
+		// Model with WHERE
+		var customer CustomerPtr
+		err := db.Select().Where(HashExp{"id": 1}).Model(2, &customer)
+		assert.Equal(t, sql.ErrNoRows, err)
+
+		err = db.Select().Where(HashExp{"id": 2}).Model(2, &customer)
+		assert.Nil(t, err)
+	}
+
+	{
+		// errors
+		var i int
+		err := db.Select().Model(1, &i)
+		assert.Equal(t, VarTypeError("must be a pointer to a struct"), err)
+
+		var a struct {
+			Name string
+		}
+
+		err = db.Select().Model(1, &a)
+		assert.Equal(t, MissingPKError, err)
+		var b struct {
+			ID1 string `db:"pk"`
+			ID2 string `db:"pk"`
+		}
+		err = db.Select().Model(1, &b)
+		assert.Equal(t, CompositePKError, err)
+	}
+}
+
+func TestSelectWithBuildHook(t *testing.T) {
+	db := getPreparedDB()
+	defer db.Close()
+
+	var buildSQL string
+
+	db.Select("id").
+		From("user").
+		WithBuildHook(func(q *Query) {
+			buildSQL = q.SQL()
+		}).
+		Build()
+
+	assert.Equal(t, "SELECT `id` FROM `user`", buildSQL)
+}