vendor dependencies with dep

vendor/github.com/go-pg/pg/orm/insert_test.go

@@ -0,0 +1,128 @@
+package orm
+
+import (
+	"github.com/go-pg/pg/types"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+type InsertTest struct {
+	Id    int
+	Value string
+}
+
+type EmbeddingTest struct {
+	tableName struct{} `sql:"name"`
+
+	Id    int
+	Field int
+}
+
+type EmbeddedInsertTest struct {
+	tableName struct{} `sql:"my_name"`
+	EmbeddingTest
+	Field2 int
+}
+
+type OverrideInsertTest struct {
+	EmbeddingTest `pg:",override"`
+	Field2        int
+}
+
+type InsertNullTest struct {
+	F1 int
+	F2 int `sql:",notnull"`
+	F3 int `sql:",pk"`
+	F4 int `sql:",pk,notnull"`
+}
+
+type InsertQTest struct {
+	Geo  types.Q
+	Func types.ValueAppender
+}
+
+var _ = Describe("Insert", func() {
+	It("multi inserts", func() {
+		q := NewQuery(nil, &InsertTest{
+			Id:    1,
+			Value: "hello",
+		}, &InsertTest{
+			Id: 2,
+		})
+
+		b, err := insertQuery{q: q}.AppendQuery(nil)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(b)).To(Equal(`INSERT INTO "insert_tests" ("id", "value") VALUES (1, 'hello'), (2, DEFAULT) RETURNING "value"`))
+	})
+
+	It("supports ON CONFLICT DO UPDATE", func() {
+		q := NewQuery(nil, &InsertTest{}).
+			Where("1 = 1").
+			OnConflict("(unq1) DO UPDATE").
+			Set("count1 = count1 + 1").
+			Where("2 = 2")
+
+		b, err := insertQuery{q: q}.AppendQuery(nil)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(b)).To(Equal(`INSERT INTO "insert_tests" AS "insert_test" ("id", "value") VALUES (DEFAULT, DEFAULT) ON CONFLICT (unq1) DO UPDATE SET count1 = count1 + 1 WHERE (2 = 2) RETURNING "id", "value"`))
+	})
+
+	It("supports ON CONFLICT DO NOTHING", func() {
+		q := NewQuery(nil, &InsertTest{}).
+			OnConflict("(unq1) DO NOTHING").
+			Set("count1 = count1 + 1").
+			Where("cond1 IS TRUE")
+
+		b, err := insertQuery{q: q}.AppendQuery(nil)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(b)).To(Equal(`INSERT INTO "insert_tests" AS "insert_test" ("id", "value") VALUES (DEFAULT, DEFAULT) ON CONFLICT (unq1) DO NOTHING RETURNING "id", "value"`))
+	})
+
+	It("supports custom table name on embedded struct", func() {
+		q := NewQuery(nil, &EmbeddedInsertTest{})
+
+		b, err := insertQuery{q: q}.AppendQuery(nil)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(b)).To(Equal(`INSERT INTO my_name ("id", "field", "field2") VALUES (DEFAULT, DEFAULT, DEFAULT) RETURNING "id", "field", "field2"`))
+	})
+
+	It("overrides table name with embedded struct", func() {
+		q := NewQuery(nil, &OverrideInsertTest{})
+
+		b, err := insertQuery{q: q}.AppendQuery(nil)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(b)).To(Equal(`INSERT INTO name ("id", "field", "field2") VALUES (DEFAULT, DEFAULT, DEFAULT) RETURNING "id", "field", "field2"`))
+	})
+
+	It("supports notnull", func() {
+		q := NewQuery(nil, &InsertNullTest{})
+
+		b, err := insertQuery{q: q}.AppendQuery(nil)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(b)).To(Equal(`INSERT INTO "insert_null_tests" ("f1", "f2", "f3", "f4") VALUES (DEFAULT, 0, DEFAULT, 0) RETURNING "f1", "f3"`))
+	})
+
+	It("inserts types.Q", func() {
+		q := NewQuery(nil, &InsertQTest{
+			Geo:  types.Q("ST_GeomFromText('POLYGON((75.150000 29.530000, 77.000000 29.000000, 77.600000 29.500000, 75.150000 29.530000))')"),
+			Func: Q("my_func(?)", "param"),
+		})
+
+		b, err := insertQuery{q: q}.AppendQuery(nil)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(b)).To(Equal(`INSERT INTO "insert_q_tests" ("geo", "func") VALUES (ST_GeomFromText('POLYGON((75.150000 29.530000, 77.000000 29.000000, 77.600000 29.500000, 75.150000 29.530000))'), my_func('param'))`))
+	})
+
+	It("supports FROM", func() {
+		q := NewQuery(nil, &InsertTest{})
+		q = q.WrapWith("data").
+			TableExpr("dst").
+			ColumnExpr("dst_col1, dst_col2").
+			TableExpr("data")
+
+		b, err := insertQuery{q: q}.AppendQuery(nil)
+		Expect(err).NotTo(HaveOccurred())
+		Expect(string(b)).To(Equal(`WITH "data" AS (SELECT "insert_test"."id", "insert_test"."value" FROM "insert_tests" AS "insert_test") INSERT INTO dst (dst_col1, dst_col2) SELECT * FROM data`))
+	})
+})