server: enable tenant testing for application_api_test

Informs: cockroachdb#138912
Epic: CRDB-38970
Release note: None

Author: shubhamdhama

pkg/base/test_server_args.go

@@ -465,6 +465,21 @@ func TestIsForStuffThatShouldWorkWithSharedProcessModeButDoesntYet(
 	}
 }
 
+// TestSkippedForExternalModeDueToPerformance can be used to disable selecting
+// the external process virtual cluster due to significant performance
+// degradation compared to other modes. However, the goal is to eventually make
+// it work efficiently in external mode.
+//
+// It should link to a github issue with label C-investigation.
+func TestSkippedForExternalModeDueToPerformance(issueNumber int) DefaultTestTenantOptions {
+	return DefaultTestTenantOptions{
+		testBehavior:           ttSharedProcess,
+		allowAdditionalTenants: true,
+		issueNum:               issueNumber,
+		label:                  "C-bug",
+	}
+}
+
 // InternalNonDefaultDecision builds a sentinel value used inside a
 // mechanism in serverutils. Should not be used by tests directly.
 func InternalNonDefaultDecision(

pkg/server/application_api/BUILD.bazel

@@ -75,8 +75,10 @@ go_test(
         "//pkg/sql/sqlstats",
         "//pkg/sql/sqlstats/persistedsqlstats",
         "//pkg/sql/sqlstats/persistedsqlstats/sqlstatstestutil",
+        "//pkg/storage/fs",
         "//pkg/testutils",
         "//pkg/testutils/diagutils",
+        "//pkg/testutils/listenerutil",
         "//pkg/testutils/serverutils",
         "//pkg/testutils/skip",
         "//pkg/testutils/sqlutils",

pkg/server/application_api/events_test.go

@@ -13,22 +13,51 @@ import (
 	"time"
 
 	"github.com/cockroachdb/cockroach/pkg/base"
+	"github.com/cockroachdb/cockroach/pkg/server"
 	"github.com/cockroachdb/cockroach/pkg/server/serverpb"
 	"github.com/cockroachdb/cockroach/pkg/server/srvtestutils"
-	"github.com/cockroachdb/cockroach/pkg/testutils/serverutils"
+	"github.com/cockroachdb/cockroach/pkg/storage/fs"
+	"github.com/cockroachdb/cockroach/pkg/testutils/listenerutil"
+	"github.com/cockroachdb/cockroach/pkg/testutils/testcluster"
 	"github.com/cockroachdb/cockroach/pkg/util/leaktest"
 	"github.com/cockroachdb/cockroach/pkg/util/log"
+	"github.com/stretchr/testify/require"
 )
 
 func TestAdminAPIEvents(t *testing.T) {
 	defer leaktest.AfterTest(t)()
 	defer log.Scope(t).Close(t)
-	s, db, _ := serverutils.StartServer(t, base.TestServerArgs{
-		// Disable the default test tenant for now as this tests fails
-		// with it enabled. Tracked with #81590.
-		DefaultTestTenant: base.TODOTestTenantDisabled,
+	ctx := context.Background()
+
+	stickyVFSRegistry := fs.NewStickyRegistry()
+	lisReg := listenerutil.NewListenerRegistry()
+	defer lisReg.Close()
+
+	tc := testcluster.StartTestCluster(t, 1, base.TestClusterArgs{
+		ReplicationMode:     base.ReplicationManual,
+		ReusableListenerReg: lisReg,
+		ServerArgs: base.TestServerArgs{
+			// TODO: time taken by first /events API call is unusually high
+			// (>40s) for external mode.
+			DefaultTestTenant: base.TestSkippedForExternalModeDueToPerformance(12345),
+			// Below parameters are required to make tc.Restart work
+			StoreSpecs: []base.StoreSpec{
+				{
+					InMemory:    true,
+					StickyVFSID: "1",
+				},
+			},
+			Knobs: base.TestingKnobs{
+				Server: &server.TestingKnobs{
+					StickyVFSRegistry: stickyVFSRegistry,
+				},
+			},
+		},
 	})
-	defer s.Stopper().Stop(context.Background())
+	defer tc.Stopper().Stop(ctx)
+	require.NoError(t, tc.Restart())
+	s := tc.Server(0)
+	db := tc.ServerConn(0)
 
 	setupQueries := []string{
 		"CREATE DATABASE api_test",
@@ -37,6 +66,7 @@ func TestAdminAPIEvents(t *testing.T) {
 		"CREATE TABLE api_test.tbl3 (a INT)",
 		"DROP TABLE api_test.tbl1",
 		"DROP TABLE api_test.tbl2",
+		"DROP DATABASE api_test",
 		"SET CLUSTER SETTING cluster.label = 'somestring';",
 	}
 	for _, q := range setupQueries {
@@ -54,9 +84,7 @@ func TestAdminAPIEvents(t *testing.T) {
 		expCount   int
 	}
 	testcases := []testcase{
-		{"node_join", false, 0, false, 1},
-		{"node_restart", false, 0, false, 0},
-		{"drop_database", false, 0, false, 0},
+		{"drop_database", false, 0, false, 1},
 		{"create_database", false, 0, false, 3},
 		{"drop_table", false, 0, false, 2},
 		{"create_table", false, 0, false, 3},
@@ -68,6 +96,19 @@ func TestAdminAPIEvents(t *testing.T) {
 		{"create_table", true, -1, false, 3},
 		{"create_table", true, 2, false, 2},
 	}
+
+	if s.StartedDefaultTestTenant() {
+		testcases = append(testcases, []testcase{
+			{"node_join", false, 0, false, 0},
+			{"node_restart", false, 0, false, 0},
+		}...)
+	} else {
+		testcases = append(testcases, []testcase{
+			{"node_join", false, 0, false, 1},
+			{"node_restart", false, 0, false, 1},
+		}...)
+	}
+
 	minTotalEvents := 0
 	for _, tc := range testcases {
 		if !tc.hasLimit {

pkg/server/application_api/insights_test.go

@@ -74,9 +74,6 @@ func TestDatabaseAndTableIndexRecommendations(t *testing.T) {
 	stubDropUnusedDuration := time.Hour
 
 	s, sqlDB, _ := serverutils.StartServer(t, base.TestServerArgs{
-		// Disable the default test tenant for now as this tests fails
-		// with it enabled. Tracked with #81590.
-		DefaultTestTenant: base.TODOTestTenantDisabled,
 		Knobs: base.TestingKnobs{
 			UnusedIndexRecommendKnobs: &idxusage.UnusedIndexRecommendationTestingKnobs{
 				GetCreatedAt:   stubTime.getCreatedAt,

pkg/server/application_api/jobs_test.go

@@ -75,15 +75,12 @@ func TestAdminAPIJobs(t *testing.T) {
 	defer log.Scope(t).Close(t)
 
 	now := timeutil.Now()
-	retentionTime := 336 * time.Hour
+	retentionDuration := 336 * time.Hour
 	s, conn, _ := serverutils.StartServer(t, base.TestServerArgs{
-		// Disable the default test tenant for now as this tests fails
-		// with it enabled. Tracked with #81590.
-		DefaultTestTenant: base.TODOTestTenantDisabled,
 		Knobs: base.TestingKnobs{
 			JobsTestingKnobs: &jobs.TestingKnobs{
 				IntervalOverrides: jobs.TestingIntervalOverrides{
-					RetentionTime: &retentionTime,
+					RetentionTime: &retentionDuration,
 				},
 			},
 			Server: &server.TestingKnobs{
@@ -249,8 +246,8 @@ func TestAdminAPIJobs(t *testing.T) {
 		},
 	}
 
-	testutils.RunTrueAndFalse(t, "isAdmin", func(t *testing.T, isAdmin bool) {
-		for i, testCase := range testCases {
+	for i, testCase := range testCases {
+		testutils.RunTrueAndFalse(t, fmt.Sprintf("%s-isAdmin", testCase.uri), func(t *testing.T, isAdmin bool) {
 			var res serverpb.JobsResponse
 			if err := srvtestutils.GetAdminJSONProtoWithAdminOption(s, testCase.uri, &res, isAdmin); err != nil {
 				t.Fatal(err)
@@ -275,22 +272,17 @@ func TestAdminAPIJobs(t *testing.T) {
 			if e, a := expected, resIDs; !reflect.DeepEqual(e, a) {
 				t.Errorf("%d - %v: expected job IDs %v, but got %v", i, testCase.uri, e, a)
 			}
-			// We don't use require.Equal() because timestamps don't necessarily
-			// compare == due to only one of them having a monotonic clock reading.
-			require.True(t, now.Add(-retentionTime).Equal(res.EarliestRetainedTime))
-		}
-	})
+			retentionTime := now.Add(-retentionDuration)
+			require.WithinDuration(t, retentionTime, res.EarliestRetainedTime, 5*time.Second)
+		})
+	}
 }
 
 func TestAdminAPIJobsDetails(t *testing.T) {
 	defer leaktest.AfterTest(t)()
 	defer log.Scope(t).Close(t)
 
-	s, conn, _ := serverutils.StartServer(t, base.TestServerArgs{
-		// Disable the default test tenant for now as this tests fails
-		// with it enabled. Tracked with #81590.
-		DefaultTestTenant: base.TODOTestTenantDisabled,
-	})
+	s, conn, _ := serverutils.StartServer(t, base.TestServerArgs{})
 	defer s.Stopper().Stop(context.Background())
 	sqlDB := sqlutils.MakeSQLRunner(conn)
 

pkg/server/application_api/query_plan_test.go

@@ -25,11 +25,7 @@ func TestAdminAPIQueryPlan(t *testing.T) {
 	defer leaktest.AfterTest(t)()
 	defer log.Scope(t).Close(t)
 
-	s, conn, _ := serverutils.StartServer(t, base.TestServerArgs{
-		// Disable the default test tenant for now as this tests fails
-		// with it enabled. Tracked with #81590.
-		DefaultTestTenant: base.TODOTestTenantDisabled,
-	})
+	s, conn, _ := serverutils.StartServer(t, base.TestServerArgs{})
 	defer s.Stopper().Stop(context.Background())
 	sqlDB := sqlutils.MakeSQLRunner(conn)
 

pkg/server/application_api/security_test.go

@@ -22,11 +22,7 @@ import (
 func TestAdminAPIUsers(t *testing.T) {
 	defer leaktest.AfterTest(t)()
 	defer log.Scope(t).Close(t)
-	s, db, _ := serverutils.StartServer(t, base.TestServerArgs{
-		// Disable the default test tenant for now as this tests fails
-		// with it enabled. Tracked with #81590.
-		DefaultTestTenant: base.TODOTestTenantDisabled,
-	})
+	s, db, _ := serverutils.StartServer(t, base.TestServerArgs{})
 	defer s.Stopper().Stop(context.Background())
 
 	// Create sample users.

pkg/server/application_api/zcfg_test.go

@@ -27,11 +27,7 @@ import (
 func TestAdminAPIZoneDetails(t *testing.T) {
 	defer leaktest.AfterTest(t)()
 	defer log.Scope(t).Close(t)
-	s, db, _ := serverutils.StartServer(t, base.TestServerArgs{
-		// Disable the default test tenant for now as this tests fails
-		// with it enabled. Tracked with #81590.
-		DefaultTestTenant: base.TODOTestTenantDisabled,
-	})
+	s, db, _ := serverutils.StartServer(t, base.TestServerArgs{})
 	defer s.Stopper().Stop(context.Background())
 
 	// Create database and table.