fork github.com/doug-martin

exp/truncate_clauses_test.go

@@ -0,0 +1,68 @@
+package exp_test
+
+import (
+	"testing"
+
+	"git.fsdpf.net/go/db/v2/exp"
+	"github.com/stretchr/testify/suite"
+)
+
+type truncateClausesSuite struct {
+	suite.Suite
+}
+
+func TestTruncateClausesSuite(t *testing.T) {
+	suite.Run(t, new(truncateClausesSuite))
+}
+
+func (tcs *truncateClausesSuite) TestHasTable() {
+	c := exp.NewTruncateClauses()
+	cle := exp.NewColumnListExpression("test1", "test2")
+	c2 := c.SetTable(cle)
+
+	tcs.False(c.HasTable())
+
+	tcs.True(c2.HasTable())
+}
+
+func (tcs *truncateClausesSuite) TestTable() {
+	c := exp.NewTruncateClauses()
+	cle := exp.NewColumnListExpression("test1", "test2")
+	c2 := c.SetTable(cle)
+
+	tcs.Nil(c.Table())
+
+	tcs.Equal(cle, c2.Table())
+}
+
+func (tcs *truncateClausesSuite) TestSetTable() {
+	cle := exp.NewColumnListExpression("test1", "test2")
+	c := exp.NewTruncateClauses().SetTable(cle)
+	cle2 := exp.NewColumnListExpression("test3", "test4")
+	c2 := c.SetTable(cle2)
+
+	tcs.Equal(cle, c.Table())
+
+	tcs.Equal(cle2, c2.Table())
+}
+
+func (tcs *truncateClausesSuite) TestOptions() {
+	c := exp.NewTruncateClauses()
+	opts := exp.TruncateOptions{Restrict: true, Identity: "RESTART", Cascade: true}
+	c2 := c.SetOptions(opts)
+
+	tcs.Equal(exp.TruncateOptions{}, c.Options())
+
+	tcs.Equal(opts, c2.Options())
+}
+
+func (tcs *truncateClausesSuite) TestSetOptions() {
+	opts := exp.TruncateOptions{Restrict: true, Identity: "RESTART", Cascade: true}
+	c := exp.NewTruncateClauses().SetOptions(opts)
+	opts2 := exp.TruncateOptions{Restrict: false, Identity: "RESTART", Cascade: false}
+	c2 := c.SetOptions(opts2)
+
+	tcs.Equal(opts, c.Options())
+
+	tcs.Equal(opts2, c2.Options())
+}