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
|
// Licensed under the MIT license, see LICENSE 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)
})
}
// Unsetenv unsets an environment variable for the duration of a test.
func (c *C) Unsetenv(name string) {
c.Setenv(name, "")
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)
}
}
|
