Function TestRefreshMaterializedViewStatementPool

pkg/sql/ast/pool_ddl_test.go:468–513

============================================================ RefreshMaterializedViewStatement pool tests ============================================================

(t *testing.T)

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

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

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

	t.Run("Fields zeroed after Put", func(t *testing.T) {
		withData := true

		stmt := GetRefreshMaterializedViewStatement()
		stmt.Name = "mv_events"
		stmt.Concurrently = true
		stmt.WithData = &withData

		PutRefreshMaterializedViewStatement(stmt)

		if stmt.Name != "" {
			t.Errorf("Name not cleared, got %q", stmt.Name)
		}
		if stmt.Concurrently {
			t.Error("Concurrently not cleared")
		}
		if stmt.WithData != nil {
			t.Error("WithData not cleared")
		}
	})

	t.Run("Pool roundtrip reuse", func(t *testing.T) {
		stmt1 := GetRefreshMaterializedViewStatement()
		stmt1.Name = "my_mv"
		PutRefreshMaterializedViewStatement(stmt1)

		stmt2 := GetRefreshMaterializedViewStatement()
		if stmt2.Name != "" {
			t.Errorf("Reused statement not clean, Name=%q", stmt2.Name)
		}
		PutRefreshMaterializedViewStatement(stmt2)
	})
}

// ============================================================
// DropStatement pool tests