-
Notifications
You must be signed in to change notification settings - Fork 5.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'same-version' of https://github.com/yujuncen/tidb into …
…same-version
- Loading branch information
Showing
14 changed files
with
674 additions
and
382 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,195 @@ | ||
| // 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, | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| // +build !codes | ||
|
|
||
| package testdata | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "encoding/json" | ||
| "flag" | ||
| "fmt" | ||
| "io" | ||
| "os" | ||
| "path/filepath" | ||
| "reflect" | ||
| "regexp" | ||
| "runtime" | ||
| "strings" | ||
| "testing" | ||
|
|
||
| "github.com/pingcap/errors" | ||
| "github.com/stretchr/testify/require" | ||
| ) | ||
|
|
||
| // record is a flag used for generate test result. | ||
| var record bool | ||
|
|
||
| func init() { | ||
| flag.BoolVar(&record, "record", false, "to generate test result") | ||
| } | ||
|
|
||
| type testCases struct { | ||
| Name string | ||
| Cases *json.RawMessage // For delayed parse. | ||
| decodedOut interface{} // For generate output. | ||
| } | ||
|
|
||
| // TestData stores all the data of a test suite. | ||
| type TestData struct { | ||
| input []testCases | ||
| output []testCases | ||
| filePathPrefix string | ||
| funcMap map[string]int | ||
| } | ||
|
|
||
| // LoadTestSuiteData loads test suite data from file. | ||
| func LoadTestSuiteData(dir, suiteName string) (res TestData, err error) { | ||
| res.filePathPrefix = filepath.Join(dir, suiteName) | ||
| res.input, err = loadTestSuiteCases(fmt.Sprintf("%s_in.json", res.filePathPrefix)) | ||
| if err != nil { | ||
| return res, err | ||
| } | ||
| if record { | ||
| res.output = make([]testCases, len(res.input)) | ||
| for i := range res.input { | ||
| res.output[i].Name = res.input[i].Name | ||
| } | ||
| } else { | ||
| res.output, err = loadTestSuiteCases(fmt.Sprintf("%s_out.json", res.filePathPrefix)) | ||
| if err != nil { | ||
| return res, err | ||
| } | ||
| if len(res.input) != len(res.output) { | ||
| return res, errors.New(fmt.Sprintf("Number of test input cases %d does not match test output cases %d", len(res.input), len(res.output))) | ||
| } | ||
| } | ||
| res.funcMap = make(map[string]int, len(res.input)) | ||
| for i, test := range res.input { | ||
| res.funcMap[test.Name] = i | ||
| if test.Name != res.output[i].Name { | ||
| return res, errors.New(fmt.Sprintf("Input name of the %d-case %s does not match output %s", i, test.Name, res.output[i].Name)) | ||
| } | ||
| } | ||
| return res, nil | ||
| } | ||
|
|
||
| func loadTestSuiteCases(filePath string) (res []testCases, err error) { | ||
| jsonFile, err := os.Open(filePath) | ||
| if err != nil { | ||
| return res, err | ||
| } | ||
| defer func() { | ||
| if err1 := jsonFile.Close(); err == nil && err1 != nil { | ||
| err = err1 | ||
| } | ||
| }() | ||
| byteValue, err := io.ReadAll(jsonFile) | ||
| if err != nil { | ||
| return res, err | ||
| } | ||
| // Remove comments, since they are not allowed in json. | ||
| re := regexp.MustCompile("(?s)//.*?\n") | ||
| err = json.Unmarshal(re.ReplaceAll(byteValue, nil), &res) | ||
| return res, err | ||
| } | ||
|
|
||
| // GetTestCases gets the test cases for a test function. | ||
| func (td *TestData) GetTestCases(t *testing.T, in interface{}, out interface{}) { | ||
| // Extract caller's name. | ||
| pc, _, _, ok := runtime.Caller(1) | ||
| require.True(t, ok) | ||
| details := runtime.FuncForPC(pc) | ||
| funcNameIdx := strings.LastIndex(details.Name(), ".") | ||
| funcName := details.Name()[funcNameIdx+1:] | ||
|
|
||
| casesIdx, ok := td.funcMap[funcName] | ||
| require.Truef(t, ok, "Must get test %s", funcName) | ||
| err := json.Unmarshal(*td.input[casesIdx].Cases, in) | ||
| require.NoError(t, err) | ||
| if !record { | ||
| err = json.Unmarshal(*td.output[casesIdx].Cases, out) | ||
| require.NoError(t, err) | ||
| } else { | ||
| // Init for generate output file. | ||
| inputLen := reflect.ValueOf(in).Elem().Len() | ||
| v := reflect.ValueOf(out).Elem() | ||
| if v.Kind() == reflect.Slice { | ||
| v.Set(reflect.MakeSlice(v.Type(), inputLen, inputLen)) | ||
| } | ||
| } | ||
| td.output[casesIdx].decodedOut = out | ||
| } | ||
|
|
||
| // OnRecord execute the function to update result. | ||
| func (td *TestData) OnRecord(updateFunc func()) { | ||
| if record { | ||
| updateFunc() | ||
| } | ||
| } | ||
|
|
||
| // ConvertRowsToStrings converts [][]interface{} to []string. | ||
| func (td *TestData) ConvertRowsToStrings(rows [][]interface{}) (rs []string) { | ||
| for _, row := range rows { | ||
| s := fmt.Sprintf("%v", row) | ||
| // Trim the leftmost `[` and rightmost `]`. | ||
| s = s[1 : len(s)-1] | ||
| rs = append(rs, s) | ||
| } | ||
| return rs | ||
| } | ||
|
|
||
| // GenerateOutputIfNeeded generate the output file. | ||
| func (td *TestData) GenerateOutputIfNeeded() error { | ||
| if !record { | ||
| return nil | ||
| } | ||
|
|
||
| buf := new(bytes.Buffer) | ||
| enc := json.NewEncoder(buf) | ||
| enc.SetEscapeHTML(false) | ||
| enc.SetIndent("", " ") | ||
| for i, test := range td.output { | ||
| err := enc.Encode(test.decodedOut) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| res := make([]byte, len(buf.Bytes())) | ||
| copy(res, buf.Bytes()) | ||
| buf.Reset() | ||
| rm := json.RawMessage(res) | ||
| td.output[i].Cases = &rm | ||
| } | ||
| err := enc.Encode(td.output) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| file, err := os.Create(fmt.Sprintf("%s_out.json", td.filePathPrefix)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| defer func() { | ||
| if err1 := file.Close(); err == nil && err1 != nil { | ||
| err = err1 | ||
| } | ||
| }() | ||
| _, err = file.Write(buf.Bytes()) | ||
| return err | ||
| } | ||
|
|
||
| // Record is a temporary method for testutil to avoid "flag redefined: record" error, | ||
| // After we migrate all tests based on former testdata, we should remove testutil and this method. | ||
| func Record() bool { | ||
| return record | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,26 @@ | ||
| // 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, | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| package math | ||
|
|
||
| import ( | ||
| "testing" | ||
|
|
||
| "github.com/pingcap/tidb/util/testbridge" | ||
| "go.uber.org/goleak" | ||
| ) | ||
|
|
||
| func TestMain(m *testing.M) { | ||
| testbridge.WorkaroundGoCheckFlags() | ||
| goleak.VerifyTestMain(m) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.