You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

141 lines
4.3 KiB

// Copyright 2017 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,
// See the License for the specific language governing permissions and
// limitations under the License.
package expression
import (
"fmt"
. "github.com/pingcap/check"
)
// generateKeys4Schema will generate keys for a given schema. Used only in this file.
func generateKeys4Schema(schema *Schema) {
keyCount := len(schema.Columns) - 1
keys := make([]KeyInfo, 0, keyCount)
for i := 0; i < keyCount; i++ {
keys = append(keys, []*Column{schema.Columns[i]})
}
schema.Keys = keys
}
// generateSchema will generate a schema for test. Used only in this file.
func (s *testEvalSuite) generateSchema(colCount int) *Schema {
cols := make([]*Column, 0, colCount)
for i := 0; i < colCount; i++ {
cols = append(cols, &Column{
UniqueID: s.allocColID(),
})
}
return NewSchema(cols...)
}
func (s *testEvalSuite) TestSchemaString(c *C) {
schema := s.generateSchema(5)
c.Assert(schema.String(), Equals, "Column: [Column#1,Column#2,Column#3,Column#4,Column#5] Unique key: []")
generateKeys4Schema(schema)
c.Assert(schema.String(), Equals, "Column: [Column#1,Column#2,Column#3,Column#4,Column#5] Unique key: [[Column#1],[Column#2],[Column#3],[Column#4]]")
}
func (s *testEvalSuite) TestSchemaRetrieveColumn(c *C) {
schema := s.generateSchema(5)
colOutSchema := &Column{
UniqueID: 100,
}
for _, col := range schema.Columns {
c.Assert(schema.RetrieveColumn(col), Equals, col)
}
c.Assert(schema.RetrieveColumn(colOutSchema), IsNil)
}
func (s *testEvalSuite) TestSchemaIsUniqueKey(c *C) {
schema := s.generateSchema(5)
generateKeys4Schema(schema)
colOutSchema := &Column{
UniqueID: 100,
}
for i, col := range schema.Columns {
if i < len(schema.Columns)-1 {
c.Assert(schema.IsUniqueKey(col), Equals, true)
} else {
c.Assert(schema.IsUniqueKey(col), Equals, false)
}
}
c.Assert(schema.IsUniqueKey(colOutSchema), Equals, false)
}
func (s *testEvalSuite) TestSchemaContains(c *C) {
schema := s.generateSchema(5)
colOutSchema := &Column{
UniqueID: 100,
}
for _, col := range schema.Columns {
c.Assert(schema.Contains(col), Equals, true)
}
c.Assert(schema.Contains(colOutSchema), Equals, false)
}
func (s *testEvalSuite) TestSchemaColumnsIndices(c *C) {
schema := s.generateSchema(5)
colOutSchema := &Column{
UniqueID: 100,
}
for i := 0; i < len(schema.Columns)-1; i++ {
colIndices := schema.ColumnsIndices([]*Column{schema.Columns[i], schema.Columns[i+1]})
for j, res := range colIndices {
c.Assert(res, Equals, i+j)
}
}
c.Assert(schema.ColumnsIndices([]*Column{schema.Columns[0], schema.Columns[1], colOutSchema, schema.Columns[2]}), IsNil)
}
func (s *testEvalSuite) TestSchemaColumnsByIndices(c *C) {
schema := s.generateSchema(5)
indices := []int{0, 1, 2, 3}
retCols := schema.ColumnsByIndices(indices)
for i, ret := range retCols {
c.Assert(fmt.Sprintf("%p", schema.Columns[i]), Equals, fmt.Sprintf("%p", ret))
}
}
func (s *testEvalSuite) TestSchemaMergeSchema(c *C) {
lSchema := s.generateSchema(5)
generateKeys4Schema(lSchema)
rSchema := s.generateSchema(5)
generateKeys4Schema(rSchema)
c.Assert(MergeSchema(nil, nil), IsNil)
c.Assert(MergeSchema(lSchema, nil).String(), Equals, lSchema.String())
c.Assert(MergeSchema(nil, rSchema).String(), Equals, rSchema.String())
schema := MergeSchema(lSchema, rSchema)
for i := 0; i < len(lSchema.Columns); i++ {
c.Assert(schema.Columns[i].UniqueID, Equals, lSchema.Columns[i].UniqueID)
}
for i := 0; i < len(rSchema.Columns); i++ {
c.Assert(schema.Columns[i+len(lSchema.Columns)].UniqueID, Equals, rSchema.Columns[i].UniqueID)
}
}
func (s *testEvalSuite) TestGetUsedList(c *C) {
schema := s.generateSchema(5)
var usedCols []*Column
usedCols = append(usedCols, schema.Columns[3])
usedCols = append(usedCols, s.generateSchema(2).Columns...)
usedCols = append(usedCols, schema.Columns[1])
usedCols = append(usedCols, schema.Columns[3])
used := GetUsedList(usedCols, schema)
c.Assert(used, DeepEquals, []bool{false, true, false, true, false})
}