Function assertNumSGIndexPartitions

rest/indextest/index_init_api_test.go:195–219

requireNumSGIndexPartitions ensures that the number of partitions for SG indexes is as expected. Some indexes aren't partitioned.

(t testing.TB, database *db.DatabaseContext)

// requireNumSGIndexPartitions ensures that the number of partitions for SG indexes is as expected. Some indexes aren't partitioned.
func assertNumSGIndexPartitions(t testing.TB, database *db.DatabaseContext) {
	gocbBucket, err := base.AsGocbV2Bucket(database.Bucket)
	require.NoError(t, err)
	re := regexp.MustCompile(`sg_(?:allDocs|channels)_x1(?:_p(\d+))?$`)
	for _, dsName := range []sgbucket.DataStoreName{db.GetSingleDatabaseCollection(t, database).GetCollectionDatastore(), database.MetadataStore} {
		allIndexes, err := gocbBucket.GetCluster().Bucket(gocbBucket.BucketName()).Scope(dsName.ScopeName()).Collection(dsName.CollectionName()).QueryIndexes().GetAllIndexes(nil)
		require.NoError(t, err)
		require.Greaterf(t, len(allIndexes), 0, "expected at least one index for datastore %s", dsName)
		for _, index := range allIndexes {
			// only two SG indexes are partitioned currently
			partitionsFromNameStrs := re.FindStringSubmatch(index.Name)
			if len(partitionsFromNameStrs) > 1 && partitionsFromNameStrs[1] != "" {
				expectedPartitionsFromName, err := strconv.ParseInt(partitionsFromNameStrs[1], 10, 64)
				require.NoError(t, err)
				require.Greaterf(t, int(expectedPartitionsFromName), 1, "expected at least one partition for %s", index.Name)
				assert.NotEqualf(t, "", index.Partition, "expected partition clause for %s", index.Name)
				assert.Equal(t, int(expectedPartitionsFromName), int(db.GetIndexPartitionCount(t, gocbBucket, dsName, index.Name)))
			} else {
				assert.Equal(t, "", index.Partition)
				assert.True(t, strings.HasSuffix(index.Name, "_x1"), "expected no partitions for %+v", index)
				assert.Equal(t, 1, int(db.GetIndexPartitionCount(t, gocbBucket, dsName, index.Name)))
			}
		}
	}
}

func TestChangeIndexPartitionsErrors(t *testing.T) {
	tests := []struct {