/
patch.go
61 lines (56 loc) · 1.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
// 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 (until c.Done is called).
//
// 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.
//
// When c.Done is called, the destination is set 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.Defer(func() {
destv.Set(oldv)
})
}
// Setenv sets an environment variable to a temporary value for the
// duration of the test (until c.Done is called).
//
// When c.Done is called, the environment variable will be returned
// to its original value.
func (c *C) Setenv(name, val string) {
oldVal := os.Getenv(name)
os.Setenv(name, val)
c.Defer(func() {
os.Setenv(name, oldVal)
})
}
// Mkdir makes a temporary directory and returns its name.
//
// The directory and its contents will be removed when
// c.Done is called.
func (c *C) Mkdir() string {
name, err := ioutil.TempDir("", "quicktest-")
c.Assert(err, Equals, nil)
c.Defer(func() {
err := os.RemoveAll(name)
c.Check(err, Equals, nil)
})
return name
}