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
|
package forceexport
import (
"fmt"
"testing"
)
func TestTimeNow(t *testing.T) {
var timeNowFunc func() (int64, int32)
GetFunc(&timeNowFunc, "time.now")
sec, nsec := timeNowFunc()
if sec == 0 || nsec == 0 {
t.Error("Expected nonzero result from time.now().")
}
}
// Note that we need to disable inlining here, or else the function won't be
// compiled into the binary. We also need to call it from the test so that the
// compiler doesn't remove it because it's unused.
//go:noinline
func addOne(x int) int {
return x + 1
}
func TestAddOne(t *testing.T) {
if addOne(3) != 4 {
t.Error("addOne should work properly.")
}
var addOneFunc func(x int) int
err := GetFunc(&addOneFunc, "github.com/alangpierce/go-forceexport.addOne")
if err != nil {
t.Error("Expected nil error.")
}
if addOneFunc(3) != 4 {
t.Error("Expected addOneFunc to add one to 3.")
}
}
func TestGetSelf(t *testing.T) {
var getFunc func(interface{}, string) error
err := GetFunc(&getFunc, "github.com/alangpierce/go-forceexport.GetFunc")
if err != nil {
t.Error("Error: %s", err)
}
// The two functions should share the same code pointer, so they should
// have the same string representation.
if fmt.Sprint(getFunc) != fmt.Sprint(GetFunc) {
t.Errorf("Expected ")
}
// Call it again on itself!
err = getFunc(&getFunc, "github.com/alangpierce/go-forceexport.GetFunc")
if err != nil {
t.Error("Error: %s", err)
}
if fmt.Sprint(getFunc) != fmt.Sprint(GetFunc) {
t.Errorf("Expected ")
}
}
func TestInvalidFunc(t *testing.T) {
t.Skip()
var invalidFunc func()
err := GetFunc(&invalidFunc, "invalidpackage.invalidfunction")
if err == nil {
t.Error("Expected an error.")
}
if invalidFunc != nil {
t.Error("Expected a nil function.")
}
}
|
