forked from confluentinc/confluent-kafka-go
/
testhelpers.go
104 lines (89 loc) · 2.66 KB
/
testhelpers.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
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
102
103
104
/**
* Copyright 2022 Confluent 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 schemaregistry
import (
"encoding/json"
"fmt"
"os"
"reflect"
"runtime"
"testing"
)
type testConf map[string]interface{}
type failFunc func(string, ...error)
var testconf = make(testConf)
var srClient Client
var maybeFail failFunc
// NewTestConf reads the test suite config file testconf.json which must
// contain at least Brokers and Topic string properties.
// Returns Testconf if the testconf was found and usable,
// error if file can't be read correctly
func testconfRead() bool {
cf, err := os.Open("../kafka/testconf.json")
defer cf.Close()
if err != nil {
fmt.Fprintf(os.Stderr, "%s testconf.json not found - ignoring test\n", err)
return false
}
jp := json.NewDecoder(cf)
err = jp.Decode(&testconf)
if err != nil {
panic(fmt.Sprintf("Failed to parse testconf: %s", err))
}
return true
}
// getObject returns a child object of the root testConf
func (tc testConf) getObject(name string) testConf {
return tc[name].(map[string]interface{})
}
// getString returns a string representation of the value represented by key from the provided namespace
// if the namespace is an empty string the root object will be searched.
func (tc testConf) getString(key string) string {
val, ok := tc[key]
if ok {
return val.(string)
}
return ""
}
// getInt returns an integer representation of the value represented by key from the provided namespace
// If the namespace is an empty string the root object will be searched.
func (tc testConf) getInt(key string) int {
val, ok := tc[key]
if ok {
return val.(int)
}
return 0
}
func initFailFunc(t *testing.T) failFunc {
tester := t
return func(msg string, errors ...error) {
for _, err := range errors {
if err != nil {
pc := make([]uintptr, 1)
runtime.Callers(2, pc)
caller := runtime.FuncForPC(pc[0])
_, line := caller.FileLine(caller.Entry())
tester.Fatalf("%s:%d failed: %s %s", caller.Name(), line, msg, err)
}
}
}
}
func expect(actual, expected interface{}) error {
if !reflect.DeepEqual(actual, expected) {
return fmt.Errorf("expected: %v, Actual: %v", expected, actual)
}
return nil
}