gitee.com/eden-framework/sqlx@v0.0.3/builder/stmt_select_test.go

package builder_test

import (
	"testing"

	. "gitee.com/eden-framework/sqlx/builder"
	. "gitee.com/eden-framework/sqlx/builder/buidertestingutils"
	"github.com/onsi/gomega"
)

func TestSelect(t *testing.T) {
	table := T("T")

	t.Run("select with modifier", func(t *testing.T) {
		gomega.NewWithT(t).Expect(
			Select(nil, "DISTINCT").
				From(
					table,
					Where(
						Col("F_a").Eq(1),
					),
				),
		).To(BeExpr(`
SELECT DISTINCT * FROM T
WHERE f_a = ?`, 1))
	})
	t.Run("select simple", func(t *testing.T) {
		gomega.NewWithT(t).Expect(
			Select(nil).
				From(
					table,
					Where(
						Col("F_a").Eq(1),
					),
					Comment("comment"),
				),
		).To(BeExpr(`
SELECT * FROM T
WHERE f_a = ?
/* comment */
`, 1))
	})
	t.Run("select with target", func(t *testing.T) {
		gomega.NewWithT(t).Expect(
			Select(Col("F_a")).
				From(table,
					Where(
						Col("F_a").Eq(1),
					),
				),
		).To(BeExpr(`
SELECT f_a FROM T
WHERE f_a = ?`, 1))
	})
	t.Run("select for update", func(t *testing.T) {
		gomega.NewWithT(t).Expect(
			Select(nil).From(
				table,
				Where(Col("F_a").Eq(1)),
				ForUpdate(),
			),
		).To(BeExpr(
			`
SELECT * FROM T
WHERE f_a = ?
FOR UPDATE
`,
			1,
		))
	})
}