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
|
package zfs_test
import (
"fmt"
"io/ioutil"
"math"
"os"
"path/filepath"
"reflect"
"runtime"
"testing"
"time"
"github.com/mistifyio/go-zfs/v3"
)
func sleep(delay int) {
time.Sleep(time.Duration(delay) * time.Second)
}
func pow2(x int) int64 {
return int64(math.Pow(2, float64(x)))
}
// https://github.com/benbjohnson/testing
// assert fails the test if the condition is false.
func _assert(t *testing.T, condition bool, msg string, v ...interface{}) {
t.Helper()
if !condition {
_, file, line, _ := runtime.Caller(2)
t.Logf("\033[31m%s:%d: "+msg+"\033[39m\n\n", append([]interface{}{filepath.Base(file), line}, v...)...)
t.FailNow()
}
}
func assert(t *testing.T, condition bool, msg string, v ...interface{}) {
t.Helper()
_assert(t, condition, msg, v...)
}
// ok fails the test if an err is not nil.
func ok(t *testing.T, err error) {
t.Helper()
_assert(t, err == nil, "unexpected error: %v", err)
}
// nok fails the test if an err is nil.
func nok(t *testing.T, err error) {
t.Helper()
_assert(t, err != nil, "expected error, got nil")
}
// equals fails the test if exp is not equal to act.
func equals(t *testing.T, exp, act interface{}) {
t.Helper()
_assert(t, reflect.DeepEqual(exp, act), "exp: %#v\n\ngot: %#v", exp, act)
}
type cleanUpFunc func()
func (f cleanUpFunc) cleanUp() {
f()
}
// do something like Restorer in github.com/packethost/pkg/internal/testenv/clearer.go
func setupZPool(t *testing.T) cleanUpFunc {
t.Helper()
d, err := ioutil.TempDir("/tmp/", "zfs-test-*")
ok(t, err)
var skipRemoveAll bool
defer func() {
if !skipRemoveAll {
t.Logf("cleaning up")
os.RemoveAll(d)
}
}()
tempfiles := make([]string, 3)
for i := range tempfiles {
f, err := ioutil.TempFile(d, fmt.Sprintf("loop%d", i))
ok(t, err)
ok(t, f.Truncate(pow2(30)))
f.Close()
tempfiles[i] = f.Name()
}
pool, err := zfs.CreateZpool("test", nil, tempfiles...)
ok(t, err)
skipRemoveAll = true
return func() {
ok(t, pool.Destroy())
os.RemoveAll(d)
}
}
|
