backup/queries_shared_test.go

package backup_test

import (
	"database/sql"
	"database/sql/driver"

	"github.com/DATA-DOG/go-sqlmock"
	"github.com/apache/cloudberry-backup/backup"
	"github.com/apache/cloudberry-backup/testutils"
	"github.com/apache/cloudberry-go-libs/structmatcher"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

var _ = Describe("backup/queries_shared tests", func() {
	Describe("GetConstraints", func() {
		It("GetConstraints properly handles NULL constraint definitions", func() {
			if (connectionPool.Version.IsGPDB() && connectionPool.Version.AtLeast("6")) || connectionPool.Version.IsCBDB() {
				Skip("Test does not apply for GPDB versions after 5")
			}

			header := []string{"oid", "schema", "name", "contype", "def", "conislocal", "owningobject", "isdomainconstraint", "ispartitionparent"}
			rowOne := []driver.Value{"1", "mock_schema", "mock_table", "mock_contype", "mock_condef", false, "mock_owningobject", false, false}
			rowTwo := []driver.Value{"2", "mock_schema2", "mock_table2", "mock_contype2", nil, false, "mock_owningobject2", false, false}
			fakeRows := sqlmock.NewRows(header).AddRow(rowOne...).AddRow(rowTwo...)
			mock.ExpectQuery(`SELECT (.*)`).WillReturnRows(fakeRows)
			result := backup.GetConstraints(connectionPool)

			// Expect the GetConstraints function to return only the 1st row since the 2nd row has a NULL constraint definition
			expectedResult := []backup.Constraint{{Oid: 1, Schema: "mock_schema", Name: "mock_table", ConType: "mock_contype",
				Def: sql.NullString{String: "mock_condef", Valid: true}, ConIsLocal: false, OwningObject: "mock_owningobject",
				IsDomainConstraint: false, IsPartitionParent: false}}
			Expect(result).To(HaveLen(1))
			structmatcher.ExpectStructsToMatch(&expectedResult[0], &result[0])
		})
	})
	Describe("RenameExchangedPartitionConstraints", func() {
		It("RenameExchangedPartitionConstraints properly renames constraints and their definitions", func() {
			testutils.SkipIfBefore7(connectionPool)

			constraints := []backup.Constraint{
				{Oid: 1, Schema: "mock_schema", Name: "mock_constraint", ConType: "p", Def: sql.NullString{String: "PRIMARY KEY (a, b)", Valid: true},
					ConIsLocal: true, OwningObject: "mock_table", IsDomainConstraint: false, IsPartitionParent: true},
				{Oid: 2, Schema: "mock_schema", Name: "like_table2_pkey", ConType: "p", Def: sql.NullString{String: "PRIMARY KEY (a, b)", Valid: true},
					ConIsLocal: true, OwningObject: "part_table_for_upgrade2", IsDomainConstraint: false, IsPartitionParent: false},
				{Oid: 3, Schema: "mock_schema", Name: "part_table_for_upgrade2_pkey", ConType: "p", Def: sql.NullString{String: "PRIMARY KEY (a, b)", Valid: true},
					ConIsLocal: true, OwningObject: "like_table2", IsDomainConstraint: false, IsPartitionParent: false}}
			header := []string{"indexname", "tablename", "relispartition"}
			rowOne := []driver.Value{"mock_constraint", "mock_table", "false"}
			rowTwo := []driver.Value{"like_table2_pkey", "part_table_for_upgrade2", "true"}
			rowThree := []driver.Value{"part_table_for_upgrade2_pkey", "like_table2", "false"}
			fakeRows := sqlmock.NewRows(header).AddRow(rowOne...).AddRow(rowTwo...).AddRow(rowThree...)
			mock.ExpectQuery(`SELECT (.*)`).WillReturnRows(fakeRows)
			backup.RenameExchangedPartitionConstraints(connectionPool, &constraints)

			Expect(constraints).To(HaveLen(3))
			for _, idx := range constraints {
				switch idx.Oid {
				case 1:
					Expect(idx.Name).To(Equal("mock_constraint"))
					Expect(idx.Def.String).To(Equal("PRIMARY KEY (a, b)"))
				case 2:
					// We do not rename the partition constraint, as it will populate down the
					// hierarchy when applied to the root. Instead rename the one that got
					// exchanged out.
					Expect(idx.Name).To(Equal("like_table2_pkey"))
					Expect(idx.Def.String).To(Equal("PRIMARY KEY (a, b)"))
				case 3:
					Expect(idx.Name).To(Equal("like_table2_pkey"))
					Expect(idx.Def.String).To(Equal("PRIMARY KEY (a, b)"))
				default:
					Fail("Unexpected index OID found")
				}
			}

		})
	})
})