Function TestImportPartitionsServerless

rest/serverless_test.go:684–744

(t *testing.T)

}

func TestImportPartitionsServerless(t *testing.T) {
	RequireBucketSpecificCredentials(t)
	if base.UnitTestUrlIsWalrus() {
		t.Skip("This test requires cbgt")
	}
	if !base.TestUseXattrs() {
		t.Skip("tests import which is not available without xattrs")
	}
	tests := []struct {
		name               string
		importPartition    *uint16
		expectedPartitions *uint16
		serverless         bool
	}{
		{
			name:               "serverless partitions",
			expectedPartitions: base.Ptr(uint16(6)),
			serverless:         true,
		},
		{
			name:               "serverless partitions with import_partition specified",
			importPartition:    base.Ptr(uint16(8)),
			expectedPartitions: base.Ptr(uint16(8)),
			serverless:         true,
		},
		{
			name:               "non serverless partitions",
			expectedPartitions: base.Ptr(uint16(16)),
			serverless:         false,
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			expectedPartitions := test.expectedPartitions
			if !base.IsEnterpriseEdition() {
				t.Logf("Import partitions setting is only supported in EE")
				expectedPartitions = nil
			}

			ctx := base.TestCtx(t)
			tb := base.GetTestBucket(t)
			defer tb.Close(ctx)
			rt := NewRestTester(t, &RestTesterConfig{CustomTestBucket: tb.NoCloseClone(), PersistentConfig: true, serverless: test.serverless})
			defer rt.Close()
			sc := rt.ServerContext()

			var dbconf *DbConfig
			if test.name == "serverless partitions with import_partition specified" {
				resp := rt.SendAdminRequest(http.MethodPut, "/db/", fmt.Sprintf(`{"bucket": "%s", "use_views": %t, "num_index_replicas": 0, "import_partitions": 8}`,
					tb.GetName(), base.TestsDisableGSI()))
				RequireStatus(t, resp, http.StatusCreated)
				dbconf = sc.GetDbConfig("db")
			} else {
				dbconf = DefaultDbConfig(sc.Config, base.TestUseXattrs())
			}

			assert.Equal(t, expectedPartitions, dbconf.ImportPartitions)