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
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
package storage
import (
"reflect"
"testing"
"github.com/containers/storage/pkg/idtools"
)
func TestGetAutoUserNSMapping(t *testing.T) {
type args struct {
size int
availableUIDs *idSet
availableGIDs *idSet
usedUIDMappings []idtools.IDMap
usedGIDMappings []idtools.IDMap
additionalUIDMappings []idtools.IDMap
additionalGIDMappings []idtools.IDMap
}
tests := []struct {
name string
args args
wantUIDMappings []idtools.IDMap
wantGIDMappings []idtools.IDMap
wantErr bool
}{
{
name: "Normal",
args: args{
size: 65536,
availableUIDs: newIDSet([]interval{{1000, 100000}}),
availableGIDs: newIDSet([]interval{{1000, 100000}}),
},
wantUIDMappings: []idtools.IDMap{{ContainerID: 0, HostID: 1000, Size: 65536}},
wantGIDMappings: []idtools.IDMap{{ContainerID: 0, HostID: 1000, Size: 65536}},
},
{
name: "NotEnoughAvailableUIDs",
args: args{
size: 65536,
availableUIDs: newIDSet([]interval{{1000, 10000}}),
availableGIDs: newIDSet([]interval{{1000, 100000}}),
},
wantErr: true,
},
{
name: "NotEnoughAvailableGIDs",
args: args{
size: 65536,
availableUIDs: newIDSet([]interval{{1000, 100000}}),
availableGIDs: newIDSet([]interval{{1000, 10000}}),
},
wantErr: true,
},
{
name: "WithUsedIDs",
args: args{
size: 65536,
availableUIDs: newIDSet([]interval{{1000, 100000}}),
availableGIDs: newIDSet([]interval{{1000, 100000}}),
usedUIDMappings: []idtools.IDMap{{ContainerID: 0, HostID: 2000, Size: 10000}},
usedGIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 1000, Size: 10000},
{ContainerID: 10000, HostID: 20000, Size: 5000},
{ContainerID: 15000, HostID: 30000, Size: 5000},
},
},
wantUIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 1000, Size: 1000},
{ContainerID: 1000, HostID: 12000, Size: 65536 - 1000},
},
wantGIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 11000, Size: 9000},
{ContainerID: 9000, HostID: 25000, Size: 5000},
{ContainerID: 14000, HostID: 35000, Size: 65536 - 9000 - 5000},
},
},
{
name: "WithUsedAndAdditionalIDs",
args: args{
size: 65536,
availableUIDs: newIDSet([]interval{{1000, 100000}}),
availableGIDs: newIDSet([]interval{{1000, 100000}}),
usedUIDMappings: []idtools.IDMap{{ContainerID: 0, HostID: 2000, Size: 10000}},
usedGIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 1000, Size: 10000},
{ContainerID: 10000, HostID: 20000, Size: 5000},
{ContainerID: 15000, HostID: 30000, Size: 5000},
},
additionalUIDMappings: []idtools.IDMap{{ContainerID: 0, HostID: 1000, Size: 1}},
additionalGIDMappings: []idtools.IDMap{{ContainerID: 1, HostID: 1001, Size: 1}},
},
wantUIDMappings: []idtools.IDMap{
{ContainerID: 1, HostID: 1001, Size: 999},
{ContainerID: 1000, HostID: 12000, Size: 65535 - 999},
{ContainerID: 0, HostID: 1000, Size: 1},
},
wantGIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 11000, Size: 1},
{ContainerID: 2, HostID: 11001, Size: 8999},
{ContainerID: 9001, HostID: 25000, Size: 5000},
{ContainerID: 14001, HostID: 35000, Size: 65535 - 1 - 8999 - 5000},
{ContainerID: 1, HostID: 1001, Size: 1},
},
},
{
name: "DiscontinuedAvailableIntervals",
args: args{
size: 65536,
availableUIDs: newIDSet([]interval{{1000, 50000}, {80000, 130000}}),
availableGIDs: newIDSet([]interval{{1000, 30000}, {70000, 90000}, {110000, 160000}}),
usedUIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 2000, Size: 10000},
{ContainerID: 0, HostID: 10000, Size: 10000},
{ContainerID: 0, HostID: 40000, Size: 10000},
},
usedGIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 1000, Size: 10000},
{ContainerID: 100, HostID: 20000, Size: 5000},
{ContainerID: 150, HostID: 30000, Size: 5000},
},
additionalUIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 1002, Size: 1},
},
additionalGIDMappings: []idtools.IDMap{
{ContainerID: 0, HostID: 1003, Size: 1},
{ContainerID: 1, HostID: 1001, Size: 1},
{ContainerID: 2, HostID: 1006, Size: 1},
{ContainerID: 100, HostID: 1100, Size: 10},
},
},
wantUIDMappings: []idtools.IDMap{
{ContainerID: 1, HostID: 1000, Size: 2},
{ContainerID: 3, HostID: 1003, Size: 997},
{ContainerID: 1000, HostID: 20000, Size: 20000},
{ContainerID: 21000, HostID: 80000, Size: 65535 - 2 - 997 - 20000},
{ContainerID: 0, HostID: 1002, Size: 1},
},
wantGIDMappings: []idtools.IDMap{
{ContainerID: 3, HostID: 11000, Size: 97},
{ContainerID: 110, HostID: 11000 + 97, Size: /*9000-97=*/ 8903},
{ContainerID: /*110+8903=*/ 9013, HostID: 25000, Size: 5000},
{ContainerID: /*9013+5000=*/ 14013, HostID: 70000, Size: 20000},
{ContainerID: /*14013+20000*/ 34013, HostID: 110000, Size: 65536 - 13 - 97 - 8903 - 5000 - 20000},
{ContainerID: 0, HostID: 1003, Size: 1},
{ContainerID: 1, HostID: 1001, Size: 1},
{ContainerID: 2, HostID: 1006, Size: 1},
{ContainerID: 100, HostID: 1100, Size: 10},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
gotUIDMappings, gotGIDMappings, err := getAutoUserNSIDMappings(
tt.args.size,
tt.args.availableUIDs, tt.args.availableGIDs,
tt.args.usedUIDMappings, tt.args.usedGIDMappings,
tt.args.additionalUIDMappings, tt.args.additionalGIDMappings,
)
if (err != nil) != tt.wantErr {
t.Errorf("getAutoUserNSMapping() error = %v, wantErr %v", err, tt.wantErr)
return
}
if !reflect.DeepEqual(gotUIDMappings, tt.wantUIDMappings) {
t.Errorf("getAutoUserNSMapping() gotUIDMappings = %v, want %v", gotUIDMappings, tt.wantUIDMappings)
}
if !reflect.DeepEqual(gotGIDMappings, tt.wantGIDMappings) {
t.Errorf("getAutoUserNSMapping() gotGIDMappings = %v, want %v", gotGIDMappings, tt.wantGIDMappings)
}
})
}
}
|
