Function TestCreateIndexStatementPool

pkg/sql/ast/pool_ddl_test.go:187–250

============================================================ CreateIndexStatement pool tests ============================================================

(t *testing.T)

// ============================================================

func TestCreateIndexStatementPool(t *testing.T) {
	t.Run("Get returns non-nil", func(t *testing.T) {
		stmt := GetCreateIndexStatement()
		if stmt == nil {
			t.Fatal("GetCreateIndexStatement() returned nil")
		}
		PutCreateIndexStatement(stmt)
	})

	t.Run("Put nil is safe", func(t *testing.T) {
		PutCreateIndexStatement(nil)
	})

	t.Run("Fields zeroed after Put", func(t *testing.T) {
		stmt := GetCreateIndexStatement()
		stmt.Name = "idx_users_email"
		stmt.Table = "users"
		stmt.Unique = true
		stmt.IfNotExists = true
		stmt.Using = "BTREE"
		stmt.Columns = append(stmt.Columns, IndexColumn{Column: "email", Direction: "ASC"})
		stmt.Where = &BinaryExpression{
			Left:     &Identifier{Name: "active"},
			Operator: "=",
			Right:    &LiteralValue{Value: "true"},
		}

		PutCreateIndexStatement(stmt)

		if stmt.Name != "" {
			t.Errorf("Name not cleared, got %q", stmt.Name)
		}
		if stmt.Table != "" {
			t.Errorf("Table not cleared, got %q", stmt.Table)
		}
		if stmt.Unique {
			t.Error("Unique not cleared")
		}
		if stmt.IfNotExists {
			t.Error("IfNotExists not cleared")
		}
		if stmt.Using != "" {
			t.Errorf("Using not cleared, got %q", stmt.Using)
		}
		if len(stmt.Columns) != 0 {
			t.Errorf("Columns not cleared, len=%d", len(stmt.Columns))
		}
		if stmt.Where != nil {
			t.Error("Where not cleared")
		}
	})

	t.Run("Pool roundtrip reuse", func(t *testing.T) {
		stmt1 := GetCreateIndexStatement()
		stmt1.Name = "idx_foo"
		PutCreateIndexStatement(stmt1)

		stmt2 := GetCreateIndexStatement()