fork github.com/doug-martin

exp/insert_clauses_test.go

@@ -0,0 +1,242 @@
+package exp_test
+
+import (
+	"testing"
+
+	"git.fsdpf.net/go/db/v2/exp"
+	"github.com/stretchr/testify/suite"
+)
+
+type insertClausesSuite struct {
+	suite.Suite
+}
+
+func TestInsertClausesSuite(t *testing.T) {
+	suite.Run(t, new(insertClausesSuite))
+}
+
+func (ics *insertClausesSuite) TestInto() {
+	c := exp.NewInsertClauses()
+	ti := exp.NewIdentifierExpression("", "test", "")
+	c2 := c.SetInto(ti)
+
+	ics.Nil(c.Into())
+
+	ics.Equal(ti, c2.Into())
+}
+
+func (ics *insertClausesSuite) TestHasInto() {
+	c := exp.NewInsertClauses()
+	ti := exp.NewIdentifierExpression("", "test", "")
+	c2 := c.SetInto(ti)
+
+	ics.False(c.HasInto())
+
+	ics.True(c2.HasInto())
+}
+
+func (ics *insertClausesSuite) TestFrom() {
+	c := exp.NewInsertClauses()
+	ae := newTestAppendableExpression("select * from test", nil)
+	c2 := c.SetFrom(ae)
+
+	ics.Nil(c.From())
+
+	ics.Equal(ae, c2.From())
+}
+
+func (ics *insertClausesSuite) TestHasFrom() {
+	c := exp.NewInsertClauses()
+	ae := newTestAppendableExpression("select * from test", nil)
+	c2 := c.SetFrom(ae)
+
+	ics.False(c.HasFrom())
+
+	ics.True(c2.HasFrom())
+}
+
+func (ics *insertClausesSuite) TestSetFrom() {
+	c := exp.NewInsertClauses()
+	ae := newTestAppendableExpression("select * from test", nil)
+	c2 := c.SetFrom(ae)
+
+	ics.Nil(c.From())
+
+	ics.Equal(ae, c2.From())
+}
+
+func (ics *insertClausesSuite) TestCols() {
+	c := exp.NewInsertClauses()
+	cle := exp.NewColumnListExpression("a", "b")
+	c2 := c.SetCols(cle)
+
+	ics.Nil(c.Cols())
+
+	ics.Equal(cle, c2.Cols())
+}
+
+func (ics *insertClausesSuite) TestHasCols() {
+	c := exp.NewInsertClauses()
+	cle := exp.NewColumnListExpression("a", "b")
+	c2 := c.SetCols(cle)
+
+	ics.False(c.HasCols())
+
+	ics.True(c2.HasCols())
+}
+
+func (ics *insertClausesSuite) TestColsAppend() {
+	cle := exp.NewColumnListExpression("a")
+	cle2 := exp.NewColumnListExpression("b")
+	c := exp.NewInsertClauses().SetCols(cle)
+	c2 := c.ColsAppend(cle2)
+
+	ics.Equal(cle, c.Cols())
+
+	ics.Equal(exp.NewColumnListExpression("a", "b"), c2.Cols())
+}
+
+func (ics *insertClausesSuite) TestVals() {
+	c := exp.NewInsertClauses()
+	vals := [][]interface{}{{"a", "b"}}
+	c2 := c.SetVals(vals)
+
+	ics.Nil(c.Vals())
+
+	ics.Equal(vals, c2.Vals())
+}
+
+func (ics *insertClausesSuite) TestHasVals() {
+	c := exp.NewInsertClauses()
+	vals := [][]interface{}{{"a", "b"}}
+	c2 := c.SetVals(vals)
+
+	ics.False(c.HasVals())
+
+	ics.True(c2.HasVals())
+}
+
+func (ics *insertClausesSuite) TestValsAppend() {
+	vals := [][]interface{}{{"a", "b"}}
+	vals2 := [][]interface{}{{"c", "d"}}
+	c := exp.NewInsertClauses().SetVals(vals)
+	c2 := c.ValsAppend(vals2)
+
+	ics.Equal(vals, c.Vals())
+
+	ics.Equal([][]interface{}{
+		{"a", "b"},
+		{"c", "d"},
+	}, c2.Vals())
+}
+
+func (ics *insertClausesSuite) TestRows() {
+	c := exp.NewInsertClauses()
+	rs := []interface{}{exp.Record{"a": "a1", "b": "b1"}}
+	c2 := c.SetRows(rs)
+
+	ics.Nil(c.Rows())
+
+	ics.Equal(rs, c2.Rows())
+}
+
+func (ics *insertClausesSuite) TestHasRows() {
+	c := exp.NewInsertClauses()
+	rs := []interface{}{exp.Record{"a": "a1", "b": "b1"}}
+	c2 := c.SetRows(rs)
+
+	ics.False(c.HasRows())
+
+	ics.True(c2.HasRows())
+}
+
+func (ics *insertClausesSuite) TestSetRows() {
+	rs := []interface{}{exp.Record{"a": "a1", "b": "b1"}}
+	c := exp.NewInsertClauses().SetRows(rs)
+	rs2 := []interface{}{exp.Record{"a": "a2", "b": "b2"}}
+	c2 := c.SetRows(rs2)
+
+	ics.Equal(rs, c.Rows())
+
+	ics.Equal(rs2, c2.Rows())
+}
+
+func (ics *insertClausesSuite) TestCommonTables() {
+	cte := exp.NewCommonTableExpression(true, "test", newTestAppendableExpression(`SELECT * FROM "foo"`, []interface{}{}))
+
+	c := exp.NewInsertClauses()
+	c2 := c.CommonTablesAppend(cte)
+
+	ics.Nil(c.CommonTables())
+
+	ics.Equal([]exp.CommonTableExpression{cte}, c2.CommonTables())
+}
+
+func (ics *insertClausesSuite) TestAddCommonTablesAppend() {
+	cte := exp.NewCommonTableExpression(true, "test", testSQLExpression("test_cte"))
+	cte2 := exp.NewCommonTableExpression(true, "test", testSQLExpression("test_cte2"))
+
+	c := exp.NewInsertClauses().CommonTablesAppend(cte)
+	c2 := c.CommonTablesAppend(cte2)
+
+	ics.Equal([]exp.CommonTableExpression{cte}, c.CommonTables())
+
+	ics.Equal([]exp.CommonTableExpression{cte, cte2}, c2.CommonTables())
+}
+
+func (ics *insertClausesSuite) TestOnConflict() {
+	ce := exp.NewDoNothingConflictExpression()
+
+	c := exp.NewInsertClauses()
+	c2 := c.SetOnConflict(ce)
+
+	ics.Nil(c.OnConflict())
+
+	ics.Equal(ce, c2.OnConflict())
+}
+
+func (ics *insertClausesSuite) TestSetOnConflict() {
+	ce := exp.NewDoNothingConflictExpression()
+
+	c := exp.NewInsertClauses().SetOnConflict(ce)
+	ce2 := exp.NewDoUpdateConflictExpression("test", exp.Record{"a": "a1"})
+	c2 := c.SetOnConflict(ce2)
+
+	ics.Equal(ce, c.OnConflict())
+
+	ics.Equal(ce2, c2.OnConflict())
+}
+
+func (ics *insertClausesSuite) TestReturning() {
+	cl := exp.NewColumnListExpression(exp.NewIdentifierExpression("", "", "col"))
+
+	c := exp.NewInsertClauses()
+	c2 := c.SetReturning(cl)
+
+	ics.Nil(c.Returning())
+
+	ics.Equal(cl, c2.Returning())
+}
+
+func (ics *insertClausesSuite) TestHasReturning() {
+	cl := exp.NewColumnListExpression(exp.NewIdentifierExpression("", "", "col"))
+
+	c := exp.NewInsertClauses()
+	c2 := c.SetReturning(cl)
+
+	ics.False(c.HasReturning())
+
+	ics.True(c2.HasReturning())
+}
+
+func (ics *insertClausesSuite) TestSetReturning() {
+	cl := exp.NewColumnListExpression(exp.NewIdentifierExpression("", "", "col"))
+	cl2 := exp.NewColumnListExpression(exp.NewIdentifierExpression("", "", "col2"))
+
+	c := exp.NewInsertClauses().SetReturning(cl)
+	c2 := c.SetReturning(cl2)
+
+	ics.Equal(cl, c.Returning())
+
+	ics.Equal(cl2, c2.Returning())
+}