Files
tidb/pkg/util/sem/sem_test.go

117 lines
5.0 KiB
Go

// Copyright 2021 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package sem
import (
"testing"
"github.com/pingcap/tidb/pkg/meta/metadef"
"github.com/pingcap/tidb/pkg/parser/mysql"
"github.com/pingcap/tidb/pkg/sessionctx/vardef"
"github.com/stretchr/testify/assert"
)
func TestInvisibleSchema(t *testing.T) {
assert := assert.New(t)
assert.True(IsInvisibleSchema(metadef.MetricSchemaName.L))
assert.True(IsInvisibleSchema("METRICS_ScHEma"))
assert.False(IsInvisibleSchema("mysql"))
assert.False(IsInvisibleSchema(metadef.InformationSchemaName.L))
assert.False(IsInvisibleSchema("Bogusname"))
}
func TestIsInvisibleTable(t *testing.T) {
assert := assert.New(t)
mysqlTbls := []string{exprPushdownBlacklist, gcDeleteRange, gcDeleteRangeDone, optRuleBlacklist, tidb, globalVariables}
infoSchemaTbls := []string{clusterConfig, clusterHardware, clusterLoad, clusterLog, clusterSystemInfo, inspectionResult,
inspectionRules, inspectionSummary, metricsSummary, metricsSummaryByLabel, metricsTables, tidbHotRegions}
perfSChemaTbls := []string{pdProfileAllocs, pdProfileBlock, pdProfileCPU, pdProfileGoroutines, pdProfileMemory,
pdProfileMutex, tidbProfileAllocs, tidbProfileBlock, tidbProfileCPU, tidbProfileGoroutines,
tidbProfileMemory, tidbProfileMutex, tikvProfileCPU}
for _, tbl := range mysqlTbls {
assert.True(IsInvisibleTable(mysql.SystemDB, tbl))
}
for _, tbl := range infoSchemaTbls {
assert.True(IsInvisibleTable(metadef.InformationSchemaName.L, tbl))
}
for _, tbl := range perfSChemaTbls {
assert.True(IsInvisibleTable(metadef.PerformanceSchemaName.L, tbl))
}
assert.True(IsInvisibleTable(metadef.MetricSchemaName.L, "acdc"))
assert.True(IsInvisibleTable(metadef.MetricSchemaName.L, "fdsgfd"))
assert.False(IsInvisibleTable("test", "t1"))
}
func TestIsRestrictedPrivilege(t *testing.T) {
assert := assert.New(t)
assert.True(IsRestrictedPrivilege("RESTRICTED_TABLES_ADMIN"))
assert.True(IsRestrictedPrivilege("RESTRICTED_STATUS_VARIABLES_ADMIN"))
assert.False(IsRestrictedPrivilege("CONNECTION_ADMIN"))
assert.False(IsRestrictedPrivilege("BACKUP_ADMIN"))
assert.False(IsRestrictedPrivilege("AA"))
}
func TestIsInvisibleStatusVar(t *testing.T) {
assert := assert.New(t)
assert.True(IsInvisibleStatusVar(tidbGCLeaderDesc))
assert.False(IsInvisibleStatusVar("server_id"))
assert.False(IsInvisibleStatusVar("ddl_schema_version"))
assert.False(IsInvisibleStatusVar("Ssl_version"))
}
func TestIsInvisibleSysVar(t *testing.T) {
assert := assert.New(t)
assert.False(IsInvisibleSysVar(vardef.Hostname)) // changes the value to default, but is not invisible
assert.False(IsInvisibleSysVar(vardef.TiDBEnableEnhancedSecurity)) // should be able to see the mode is on.
assert.False(IsInvisibleSysVar(vardef.TiDBAllowRemoveAutoInc))
assert.True(IsInvisibleSysVar(vardef.TiDBCheckMb4ValueInUTF8))
assert.True(IsInvisibleSysVar(vardef.TiDBConfig))
assert.True(IsInvisibleSysVar(vardef.TiDBEnableSlowLog))
assert.True(IsInvisibleSysVar(vardef.TiDBExpensiveQueryTimeThreshold))
assert.True(IsInvisibleSysVar(vardef.TiDBForcePriority))
assert.True(IsInvisibleSysVar(vardef.TiDBGeneralLog))
assert.True(IsInvisibleSysVar(vardef.TiDBMetricSchemaRangeDuration))
assert.True(IsInvisibleSysVar(vardef.TiDBMetricSchemaStep))
assert.True(IsInvisibleSysVar(vardef.TiDBOptWriteRowID))
assert.True(IsInvisibleSysVar(vardef.TiDBPProfSQLCPU))
assert.True(IsInvisibleSysVar(vardef.TiDBRecordPlanInSlowLog))
assert.True(IsInvisibleSysVar(vardef.TiDBSlowQueryFile))
assert.True(IsInvisibleSysVar(vardef.TiDBSlowLogThreshold))
assert.True(IsInvisibleSysVar(vardef.TiDBEnableCollectExecutionInfo))
assert.True(IsInvisibleSysVar(vardef.TiDBMemoryUsageAlarmRatio))
assert.True(IsInvisibleSysVar(vardef.TiDBEnableTelemetry))
assert.True(IsInvisibleSysVar(vardef.TiDBRowFormatVersion))
assert.True(IsInvisibleSysVar(vardef.TiDBRedactLog))
assert.True(IsInvisibleSysVar(vardef.TiDBTopSQLMaxTimeSeriesCount))
assert.True(IsInvisibleSysVar(vardef.TiDBServiceScope))
assert.True(IsInvisibleSysVar(vardef.TiDBCloudStorageURI))
assert.True(IsInvisibleSysVar(vardef.TiDBStmtSummaryMaxStmtCount))
assert.True(IsInvisibleSysVar(vardef.TiDBServerMemoryLimit))
assert.True(IsInvisibleSysVar(vardef.TiDBServerMemoryLimitGCTrigger))
assert.True(IsInvisibleSysVar(vardef.TiDBInstancePlanCacheMaxMemSize))
assert.True(IsInvisibleSysVar(vardef.TiDBStatsCacheMemQuota))
assert.True(IsInvisibleSysVar(vardef.TiDBMemQuotaBindingCache))
assert.True(IsInvisibleSysVar(vardef.TiDBSchemaCacheSize))
assert.True(IsInvisibleSysVar(tidbAuditRetractLog))
}