-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathpermission_control_test.go
146 lines (119 loc) · 3.85 KB
/
permission_control_test.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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
package aferocopy_test
import (
"os"
"testing"
"github.com/stretchr/testify/require"
"go.nhat.io/aferomock"
"go.nhat.io/aferocopy/v2"
)
func TestPermissionControl_AddPermission_File(t *testing.T) {
t.Parallel()
// Mocked file info and FS.
srcInfo := aferomock.MockFileInfo(func(fileInfo *aferomock.FileInfo) {
// Original permissions 0111
fileInfo.On("Mode").Return(os.FileMode(0o111))
fileInfo.On("IsDir").Return(false)
})(t)
destFs := aferomock.MockFs(func(fs *aferomock.Fs) {
// Expected original + new permissions.
fs.On("Chmod", "foo.bar", os.FileMode(0o111|0o321)).Return(nil)
})(t)
// Set temporary permissions.
cb, err := aferocopy.AddPermission(0o321)(srcInfo, destFs, "foo.bar")
require.NoError(t, err)
// Set final permissions.
cb(&err)
require.NoError(t, err)
}
func TestPermissionControl_AddPermission_Dir(t *testing.T) {
t.Parallel()
// Mocked file info and FS.
srcInfo := aferomock.MockFileInfo(func(fileInfo *aferomock.FileInfo) {
// Original permissions 0111.
fileInfo.On("Mode").Return(os.FileMode(0o111))
fileInfo.On("IsDir").Return(true)
})(t)
destFs := aferomock.MockFs(func(fs *aferomock.Fs) {
// Expected original + new permissions
fs.On("MkdirAll", "foo", os.FileMode(0o755)).Return(nil)
fs.On("Chmod", "foo", os.FileMode(0o111|0o321)).Return(nil)
})(t)
// Set temporary permissions.
cb, err := aferocopy.AddPermission(0o321)(srcInfo, destFs, "foo")
require.NoError(t, err)
// Set final permissions.
cb(&err)
require.NoError(t, err)
}
func TestPermissionControl_PreservePermission_File(t *testing.T) {
t.Parallel()
// Mocked file info and FS.
srcInfo := aferomock.MockFileInfo(func(fileInfo *aferomock.FileInfo) {
// Original permissions 0123.
fileInfo.On("Mode").Return(os.FileMode(0o123))
fileInfo.On("IsDir").Return(false)
})(t)
destFs := aferomock.MockFs(func(fs *aferomock.Fs) {
// Same permissions as original.
fs.On("Chmod", "foo.bar", os.FileMode(0o123)).Return(nil)
})(t)
// Set temporary permissions.
cb, err := aferocopy.PreservePermission(srcInfo, destFs, "foo.bar")
require.NoError(t, err)
// Set final permissions.
cb(&err)
require.NoError(t, err)
}
func TestPermissionControl_PreservePermission_Dir(t *testing.T) {
t.Parallel()
// Mocked file info and FS.
srcInfo := aferomock.MockFileInfo(func(fileInfo *aferomock.FileInfo) {
// Original permissions 0123.
fileInfo.On("Mode").Return(os.FileMode(0o123))
fileInfo.On("IsDir").Return(true)
})(t)
destFs := aferomock.MockFs(func(fs *aferomock.Fs) {
// Same permissions as original.
fs.On("MkdirAll", "foo", os.FileMode(0o755)).Return(nil)
fs.On("Chmod", "foo", os.FileMode(0o123)).Return(nil)
})(t)
// Set temporary permissions.
cb, err := aferocopy.PreservePermission(srcInfo, destFs, "foo")
require.NoError(t, err)
// Set final permissions.
cb(&err)
require.NoError(t, err)
}
func TestPermissionControl_DoNothing_File(t *testing.T) {
t.Parallel()
// Mocked file info and FS.
srcInfo := aferomock.MockFileInfo(func(fileInfo *aferomock.FileInfo) {
fileInfo.On("IsDir").Return(false)
})(t)
destFs := aferomock.MockFs()(t)
// Set temporary permissions.
cb, err := aferocopy.DoNothing(srcInfo, destFs, "foo.bar")
require.NoError(t, err)
// Set final permissions.
cb(&err)
require.NoError(t, err)
}
func TestPermissionControl_DoNothing_Dir(t *testing.T) {
t.Parallel()
// Mocked file info and FS.
srcInfo := aferomock.MockFileInfo(func(fileInfo *aferomock.FileInfo) {
// Original permissions 0123.
fileInfo.On("Mode").Return(os.FileMode(0o123))
fileInfo.On("IsDir").Return(true)
})(t)
destFs := aferomock.MockFs(func(fs *aferomock.Fs) {
// Same permissions as original.
fs.On("MkdirAll", "foo", os.FileMode(0o123)).Return(nil)
})(t)
// Set temporary permissions.
cb, err := aferocopy.DoNothing(srcInfo, destFs, "foo")
require.NoError(t, err)
// Set final permissions
cb(&err)
require.NoError(t, err)
}