/
patch.go
98 lines (90 loc) · 2.54 KB
/
patch.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
// Licensed under the MIT license, see LICENCE file for details.
package quicktest
import (
"io/ioutil"
"os"
"reflect"
)
// Patch sets a variable to a temporary value for the duration of the test.
//
// It sets the value pointed to by the given destination to the given
// value, which must be assignable to the element type of the destination.
//
// At the end of the test (see "Deferred execution" in the package docs), the
// destination is set back to its original value.
func (c *C) Patch(dest, value interface{}) {
destv := reflect.ValueOf(dest).Elem()
oldv := reflect.New(destv.Type()).Elem()
oldv.Set(destv)
valuev := reflect.ValueOf(value)
if !valuev.IsValid() {
// This isn't quite right when the destination type is not
// nilable, but it's better than the complex alternative.
valuev = reflect.Zero(destv.Type())
}
destv.Set(valuev)
c.cleanup(func() {
destv.Set(oldv)
})
}
// Setenv sets an environment variable to a temporary value for the
// duration of the test.
//
// At the end of the test (see "Deferred execution" in the package docs), the
// environment variable is returned to its original value.
func (c *C) Setenv(name, val string) {
c.setenv(name, val, true)
}
// Unsetenv unsets an environment variable for the duration of a test.
func (c *C) Unsetenv(name string) {
c.setenv(name, "", false)
}
// setenv sets or unsets an environment variable to a temporary value for the
// duration of the test
func (c *C) setenv(name, val string, valOK bool) {
oldVal, oldOK := os.LookupEnv(name)
if valOK {
os.Setenv(name, val)
} else {
os.Unsetenv(name)
}
c.cleanup(func() {
if oldOK {
os.Setenv(name, oldVal)
} else {
os.Unsetenv(name)
}
})
}
// Mkdir makes a temporary directory and returns its name.
//
// At the end of the test (see "Deferred execution" in the package docs), the
// directory and its contents are removed.
//
// Deprecated: in Go >= 1.15 use testing.TB.TempDir instead.
func (c *C) Mkdir() string {
td, ok := c.TB.(interface {
TempDir() string
})
if ok {
return td.TempDir()
}
name, err := ioutil.TempDir("", "quicktest-")
c.Assert(err, Equals, nil)
c.cleanup(func() {
if err := os.RemoveAll(name); err != nil {
// Don't call c.Check because the stack traverse logic won't
// print the source location, so just log instead.
c.Errorf("quicktest cannot remove temporary testing directory: %v", err)
}
})
return name
}
// cleanup uses Cleanup when it can, falling back to using Defer.
func (c *C) cleanup(f func()) {
if tb, ok := c.TB.(cleaner); ok {
tb.Cleanup(f)
} else {
c.Defer(f)
}
}