Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions pkg/server/server_sql.go
Original file line number Diff line number Diff line change
Expand Up @@ -1270,7 +1270,6 @@ func newSQLServer(ctx context.Context, cfg sqlServerArgs) (*SQLServer, error) {
var c upgrade.Cluster
var systemDeps upgrade.SystemDeps
keyVisKnobs, _ := cfg.TestingKnobs.KeyVisualizer.(*keyvisualizer.TestingKnobs)
sqlStatsKnobs, _ := cfg.TestingKnobs.SQLStatsKnobs.(*sqlstats.TestingKnobs)
upgradeKnobs, _ := cfg.TestingKnobs.UpgradeManager.(*upgradebase.TestingKnobs)
if codec.ForSystemTenant() {
c = upgradecluster.New(upgradecluster.ClusterConfig{
Expand All @@ -1296,7 +1295,6 @@ func newSQLServer(ctx context.Context, cfg sqlServerArgs) (*SQLServer, error) {
JobRegistry: jobRegistry,
Stopper: cfg.stopper,
KeyVisKnobs: keyVisKnobs,
SQLStatsKnobs: sqlStatsKnobs,
TenantInfoAccessor: cfg.tenantConnect,
TestingKnobs: upgradeKnobs,
}
Expand Down
1 change: 0 additions & 1 deletion pkg/sql/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -730,7 +730,6 @@ go_test(
"schema_changer_test.go",
"schema_resolver_test.go",
"sequence_test.go",
"server_params_test.go",
"session_migration_test.go",
"set_zone_config_test.go",
"show_cluster_setting_test.go",
Expand Down
12 changes: 10 additions & 2 deletions pkg/sql/as_of_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,11 @@ import (
"github.com/cockroachdb/cockroach/pkg/base"
"github.com/cockroachdb/cockroach/pkg/jobs/jobspb"
"github.com/cockroachdb/cockroach/pkg/kv/kvpb"
"github.com/cockroachdb/cockroach/pkg/kv/kvserver"
"github.com/cockroachdb/cockroach/pkg/kv/kvserver/kvserverbase"
"github.com/cockroachdb/cockroach/pkg/sql"
"github.com/cockroachdb/cockroach/pkg/sql/sem/tree"
"github.com/cockroachdb/cockroach/pkg/sql/tests"
"github.com/cockroachdb/cockroach/pkg/testutils"
"github.com/cockroachdb/cockroach/pkg/testutils/serverutils"
"github.com/cockroachdb/cockroach/pkg/testutils/sqlutils"
Expand All @@ -30,7 +32,7 @@ func TestAsOfTime(t *testing.T) {
defer log.Scope(t).Close(t)

ctx, cancel := context.WithCancel(context.Background())
params, _ := createTestServerParamsAllowTenants()
var params base.TestServerArgs
params.Knobs.GCJob = &sql.GCJobTestingKnobs{RunBeforeResume: func(_ jobspb.JobID) error {
<-ctx.Done()
return nil
Expand Down Expand Up @@ -263,7 +265,13 @@ func TestAsOfRetry(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

params, cmdFilters := createTestServerParamsAllowTenants()
var cmdFilters tests.CommandFilters
params := base.TestServerArgs{}
params.Knobs.Store = &kvserver.StoreTestingKnobs{
EvalKnobs: kvserverbase.BatchEvalTestingKnobs{
TestingEvalFilter: cmdFilters.RunFilters,
},
}
s, sqlDB, _ := serverutils.StartServer(t, params)
defer s.Stopper().Stop(context.Background())

Expand Down
4 changes: 2 additions & 2 deletions pkg/sql/comment_on_column_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import (
"fmt"
"testing"

"github.com/cockroachdb/cockroach/pkg/base"
"github.com/cockroachdb/cockroach/pkg/testutils/serverutils"
"github.com/cockroachdb/cockroach/pkg/util/leaktest"
"github.com/cockroachdb/cockroach/pkg/util/log"
Expand Down Expand Up @@ -225,8 +226,7 @@ func runCommentOnTestsDeclarativeOnly(t *testing.T, testFunc func(db *gosql.DB))
}

func runOneCommentOnTest(t *testing.T, setupQuery string, testFunc func(db *gosql.DB)) {
params, _ := createTestServerParamsAllowTenants()
s, db, _ := serverutils.StartServer(t, params)
s, db, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop(context.Background())
_, err := db.Exec(setupQuery)
require.NoError(t, err)
Expand Down
7 changes: 4 additions & 3 deletions pkg/sql/conn_executor_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -75,9 +75,10 @@ import (
func TestSessionFinishRollsBackTxn(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

aborter := NewTxnAborter()
defer aborter.Close(t)
params, _ := createTestServerParamsAllowTenants()
var params base.TestServerArgs
params.Knobs.SQLExecutor = aborter.executorKnobs()
srv, mainDB, _ := serverutils.StartServer(t, params)
defer srv.Stopper().Stop(context.Background())
Expand Down Expand Up @@ -369,7 +370,7 @@ func TestHalloweenProblemAvoidance(t *testing.T) {
defer mutations.ResetMaxBatchSizeForTests()
numRows := smallerKvBatchSize + smallerInsertBatchSize + 10

params, _ := createTestServerParamsAllowTenants()
var params base.TestServerArgs
params.Insecure = true
params.Knobs.DistSQL = &execinfra.TestingKnobs{
TableReaderBatchBytesLimit: 10,
Expand Down Expand Up @@ -440,7 +441,7 @@ func TestAppNameStatisticsInitialization(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

params, _ := createTestServerParamsAllowTenants()
var params base.TestServerArgs
params.Insecure = true

s := serverutils.StartServerOnly(t, params)
Expand Down
12 changes: 4 additions & 8 deletions pkg/sql/crdb_internal_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -83,9 +83,7 @@ func TestGetAllNamesInternal(t *testing.T) {
defer log.Scope(t).Close(t)

ctx := context.Background()
params, _ := createTestServerParamsAllowTenants()

s, _ /* sqlDB */, kvDB := serverutils.StartServer(t, params)
s, _ /* sqlDB */, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop(ctx)

err := kvDB.Txn(ctx, func(ctx context.Context, txn *kv.Txn) error {
Expand Down Expand Up @@ -169,8 +167,7 @@ func TestGossipAlertsTable(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

params, _ := createTestServerParamsAllowTenants()
s := serverutils.StartServerOnly(t, params)
s := serverutils.StartServerOnly(t, base.TestServerArgs{})
defer s.Stop(context.Background())
ctx := context.Background()

Expand Down Expand Up @@ -210,8 +207,7 @@ func TestOldBitColumnMetadata(t *testing.T) {
defer log.Scope(t).Close(t)

ctx := context.Background()
params, _ := createTestServerParamsAllowTenants()
s, sqlDB, kvDB := serverutils.StartServer(t, params)
s, sqlDB, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop(ctx)

// The descriptor changes made must have an immediate effect
Expand Down Expand Up @@ -428,7 +424,7 @@ func TestInvalidObjects(t *testing.T) {
defer log.Scope(t).Close(t)

ctx := context.Background()
params, _ := createTestServerParamsAllowTenants()
var params base.TestServerArgs
params.Knobs = base.TestingKnobs{
Store: &kvserver.StoreTestingKnobs{
DisableMergeQueue: true,
Expand Down
4 changes: 2 additions & 2 deletions pkg/sql/create_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -244,8 +244,8 @@ func TestParallelCreateConflictingTables(t *testing.T) {
func TestTableReadErrorsBeforeTableCreation(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)
params, _ := createTestServerParamsAllowTenants()
s, sqlDB, _ := serverutils.StartServer(t, params)

s, sqlDB, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop(context.Background())

if _, err := sqlDB.Exec(`
Expand Down
13 changes: 5 additions & 8 deletions pkg/sql/delete_preserving_index_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,8 @@ import (
func TestDeletePreservingIndexEncoding(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)
params, _ := createTestServerParamsAllowTenants()

var params base.TestServerArgs
mergeFinished := make(chan struct{})
completeSchemaChange := make(chan struct{})
errorChan := make(chan error, 1)
Expand Down Expand Up @@ -246,8 +247,7 @@ func TestDeletePreservingIndexEncodingUsesNormalDeletesInDeleteOnly(t *testing.T
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

params, _ := createTestServerParamsAllowTenants()
server, sqlDB, kvDB := serverutils.StartServer(t, params)
server, sqlDB, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer server.Stopper().Stop(context.Background())

// The descriptor changes made must have an immediate effect
Expand Down Expand Up @@ -311,8 +311,7 @@ func TestDeletePreservingIndexEncodingWithEmptyValues(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

params, _ := createTestServerParamsAllowTenants()
server, sqlDB, kvDB := serverutils.StartServer(t, params)
server, sqlDB, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer server.Stopper().Stop(context.Background())

// The descriptor changes made must have an immediate effect
Expand Down Expand Up @@ -515,8 +514,6 @@ func TestMergeProcessor(t *testing.T) {
defer log.Scope(t).Close(t)
ctx := context.Background()

params, _ := createTestServerParamsAllowTenants()

type TestCase struct {
name string
setupSQL string
Expand Down Expand Up @@ -611,7 +608,7 @@ func TestMergeProcessor(t *testing.T) {
}

run := func(t *testing.T, test TestCase) {
server, tdb, kvDB := serverutils.StartServer(t, params)
server, tdb, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer server.Stopper().Stop(context.Background())
defer lease.TestingDisableTableLeases()()

Expand Down
23 changes: 9 additions & 14 deletions pkg/sql/descriptor_mutation_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -165,9 +165,7 @@ func TestUpsertWithColumnMutationAndNotNullDefault(t *testing.T) {
// no job associated with the added mutations, those mutations stay on the
// table descriptor but don't do anything, which is what we want.

// Disable external processing of mutations.
params, _ := createTestServerParamsAllowTenants()
server, sqlDB, kvDB := serverutils.StartServer(t, params)
server, sqlDB, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
ctx := context.Background()
defer server.Stopper().Stop(ctx)
// The descriptor changes made must have an immediate effect
Expand Down Expand Up @@ -227,8 +225,7 @@ func TestOperationsWithColumnMutation(t *testing.T) {

// Disable external processing of mutations.
ctx := context.Background()
params, _ := createTestServerParamsAllowTenants()
server, sqlDB, kvDB := serverutils.StartServer(t, params)
server, sqlDB, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer server.Stopper().Stop(ctx)
sqlRunner := sqlutils.MakeSQLRunner(sqlDB)

Expand Down Expand Up @@ -494,9 +491,7 @@ func TestOperationsWithIndexMutation(t *testing.T) {
// no job associated with the added mutations, those mutations stay on the
// table descriptor but don't do anything, which is what we want.

// Disable external processing of mutations.
params, _ := createTestServerParamsAllowTenants()
server, sqlDB, kvDB := serverutils.StartServer(t, params)
server, sqlDB, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer server.Stopper().Stop(context.Background())
sqlRunner := sqlutils.MakeSQLRunner(sqlDB)
// The descriptor changes made must have an immediate effect.
Expand Down Expand Up @@ -661,8 +656,7 @@ func TestOperationsWithColumnAndIndexMutation(t *testing.T) {
// no job associated with the added mutations, those mutations stay on the
// table descriptor but don't do anything, which is what we want.

params, _ := createTestServerParamsAllowTenants()
server, sqlDB, kvDB := serverutils.StartServer(t, params)
server, sqlDB, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
ctx := context.Background()
defer server.Stopper().Stop(ctx)
sqlRunner := sqlutils.MakeSQLRunner(sqlDB)
Expand Down Expand Up @@ -876,8 +870,9 @@ func TestOperationsWithColumnAndIndexMutation(t *testing.T) {
func TestSchemaChangeCommandsWithPendingMutations(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

// Disable external processing of mutations.
params, _ := createTestServerParamsAllowTenants()
var params base.TestServerArgs
params.Knobs = base.TestingKnobs{
SQLSchemaChanger: &sql.SchemaChangerTestingKnobs{
SchemaChangeJobNoOp: func() bool {
Expand Down Expand Up @@ -1099,9 +1094,10 @@ CREATE TABLE t.test (a STRING PRIMARY KEY, b STRING, c STRING, INDEX foo (c));
func TestTableMutationQueue(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

// Disable synchronous and asynchronous schema change processing so that
// the mutations get queued up.
params, _ := createTestServerParamsAllowTenants()
var params base.TestServerArgs
params.Knobs = base.TestingKnobs{
SQLSchemaChanger: &sql.SchemaChangerTestingKnobs{
SchemaChangeJobNoOp: func() bool {
Expand Down Expand Up @@ -1218,8 +1214,7 @@ func TestAddingFKs(t *testing.T) {
defer leaktest.AfterTest(t)()
defer log.Scope(t).Close(t)

params, _ := createTestServerParamsAllowTenants()
s, sqlDB, kvDB := serverutils.StartServer(t, params)
s, sqlDB, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop(context.Background())

if _, err := sqlDB.Exec(`
Expand Down
Loading
Loading