1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
// Copyright 2020 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 task
import (
"context"
"testing"
"time"
"github.com/pingcap/tiup/pkg/cluster/ctxt"
"github.com/pingcap/tiup/pkg/cluster/spec"
logprinter "github.com/pingcap/tiup/pkg/logger/printer"
"github.com/pingcap/tiup/pkg/meta"
"github.com/pingcap/tiup/pkg/utils/mock"
"github.com/pingcap/errors"
"github.com/stretchr/testify/require"
)
type fakeExecutor struct{}
func (e *fakeExecutor) Execute(ctx context.Context, cmd string, sudo bool, timeout ...time.Duration) (stdout []byte, stderr []byte, err error) {
return []byte{}, []byte{}, nil
}
func (e *fakeExecutor) Transfer(ctx context.Context, src, dst string, download bool, limit int, compress bool) error {
return nil
}
type fakeInstance struct {
hasConfigError bool
*spec.TiDBInstance
}
func (i *fakeInstance) InitConfig(
ctx context.Context,
e ctxt.Executor,
clusterName string,
clusterVersion string,
deployUser string,
paths meta.DirPaths,
) error {
if i.hasConfigError {
return errors.Annotate(spec.ErrorCheckConfig, "test error")
}
return nil
}
func (i *fakeInstance) GetHost() string {
return "1.1.1.1"
}
func (i *fakeInstance) GetPort() int {
return 4000
}
func (i *fakeInstance) GetManageHost() string {
return "1.1.1.1"
}
func TestCheckConfig(t *testing.T) {
ctx := ctxt.New(context.Background(), 0, logprinter.NewLogger(""))
mf := mock.With("FakeExecutor", &fakeExecutor{})
defer mf()
initCfg := &InitConfig{
clusterName: "test-cluster-name",
clusterVersion: "v6.0.0",
paths: meta.DirPaths{
Cache: "/tmp",
},
}
tests := [][]bool{
{false, false, false}, // hasConfigError, ignoreConfigError, expectError
{true, false, true},
{false, true, false},
{true, true, false},
}
for _, test := range tests {
initCfg.instance = &fakeInstance{test[0], nil}
initCfg.ignoreCheck = test[1]
if test[2] {
require.Error(t, initCfg.Execute(ctx))
} else {
require.NoError(t, initCfg.Execute(ctx))
}
}
}
|
