/
test_export.go
49 lines (43 loc) · 1.32 KB
/
test_export.go
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
/**
* Tencent is pleased to support the open source community by making Polaris available.
*
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
*
* Licensed under the BSD 3-Clause License (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://opensource.org/licenses/BSD-3-Clause
*
* 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 store
import (
"errors"
"fmt"
)
// TestGetStore 获取Store
func TestGetStore() (Store, error) {
name := config.Name
if name == "" {
return nil, errors.New("store name is empty")
}
s, ok := StoreSlots[name]
if !ok {
return nil, fmt.Errorf("store `%s` not found", name)
}
_ = s.Destroy()
fmt.Printf("[Store][Info] current use store plugin : %s\n", s.Name())
if err := s.Initialize(config); err != nil {
fmt.Printf("[ERROR] initialize store `%s` fail: %v", s.Name(), err)
panic(err)
}
return s, nil
}
// TestInjectConfig just for test
func TestInjectConfig(conf Config) {
config = &conf
}