forked from spacemeshos/go-spacemesh
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dirs_test_helpers.go
61 lines (54 loc) · 1.24 KB
/
dirs_test_helpers.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
package filesystem
import (
"os"
"os/user"
"testing"
)
// DeleteSpacemeshDataFolders deletes all sub directories and files in the Spacemesh root data folder.
func DeleteSpacemeshDataFolders(t *testing.T) {
aPath, err := GetSpacemeshDataDirectoryPath()
if err != nil {
t.Fatalf("Failed to get spacemesh data dir: %s", err)
}
// remove
err = os.RemoveAll(aPath)
if err != nil {
t.Fatalf("Failed to delete spacemesh data dir: %s", err)
}
}
// TestUsers returns a map of users for testing
func TestUsers() map[string]*user.User {
return map[string]*user.User{
"alice": {
Uid: "100",
Gid: "500",
Username: "alice",
Name: "Alice Smith",
HomeDir: "/home/alice",
},
"bob": {
Uid: "200",
Gid: "500",
Username: "bob",
Name: "Bob Jones",
HomeDir: "/home/bob",
},
"michael": {
Uid: "300",
Gid: "500",
Username: "michael",
Name: "Michael Smith",
HomeDir: "",
},
}
}
// SetupTestHooks sets current user to mock user to test
func SetupTestHooks(users map[string]*user.User) {
currentUser = func() (*user.User, error) {
return users["michael"], nil
}
}
// TearDownTestHooks sets current user back
func TearDownTestHooks() {
currentUser = user.Current
}