Skip to content

Commit 07cc7d9

Browse files
committed
Split tests so we check both the deprecated and new options
1 parent b10d828 commit 07cc7d9

3 files changed

Lines changed: 265 additions & 114 deletions

File tree

disk/tests/test_filter.py

Lines changed: 43 additions & 108 deletions
Original file line numberDiff line numberDiff line change
@@ -25,40 +25,40 @@ def test_default_mock():
2525

2626
def test_bad_config_string_regex():
2727
instance = {
28-
'file_system_whitelist': 'test',
29-
'file_system_blacklist': 'test',
30-
'device_whitelist': 'test',
31-
'device_blacklist': 'test',
32-
'mount_point_whitelist': 'test',
33-
'mount_point_blacklist': 'test',
28+
'file_system_include': 'test',
29+
'file_system_exclude': 'test',
30+
'device_include': 'test',
31+
'device_exclude': 'test',
32+
'mount_point_include': 'test',
33+
'mount_point_exclude': 'test',
3434
}
3535
c = Disk('disk', {}, [instance])
3636

37-
assert_regex_equal(c._file_system_whitelist, re.compile('test', re.I))
38-
assert_regex_equal(c._file_system_blacklist, re.compile('test|iso9660$', re.I))
39-
assert_regex_equal(c._device_whitelist, re.compile('test', IGNORE_CASE))
40-
assert_regex_equal(c._device_blacklist, re.compile('test', IGNORE_CASE))
41-
assert_regex_equal(c._mount_point_whitelist, re.compile('test', IGNORE_CASE))
42-
assert_regex_equal(c._mount_point_blacklist, re.compile('test', IGNORE_CASE))
37+
assert_regex_equal(c._file_system_include, re.compile('test', re.I))
38+
assert_regex_equal(c._file_system_exclude, re.compile('test|iso9660$', re.I))
39+
assert_regex_equal(c._device_include, re.compile('test', IGNORE_CASE))
40+
assert_regex_equal(c._device_exclude, re.compile('test', IGNORE_CASE))
41+
assert_regex_equal(c._mount_point_include, re.compile('test', IGNORE_CASE))
42+
assert_regex_equal(c._mount_point_exclude, re.compile('test', IGNORE_CASE))
4343

4444

4545
def test_ignore_empty_regex():
4646
instance = {
47-
'file_system_whitelist': ['test', ''],
48-
'file_system_blacklist': ['test', ''],
49-
'device_whitelist': ['test', ''],
50-
'device_blacklist': ['test', ''],
51-
'mount_point_whitelist': ['test', ''],
52-
'mount_point_blacklist': ['test', ''],
47+
'file_system_include': ['test', ''],
48+
'file_system_exclude': ['test', ''],
49+
'device_include': ['test', ''],
50+
'device_exclude': ['test', ''],
51+
'mount_point_include': ['test', ''],
52+
'mount_point_exclude': ['test', ''],
5353
}
5454
c = Disk('disk', {}, [instance])
5555

56-
assert_regex_equal(c._file_system_whitelist, re.compile('test', re.I))
57-
assert_regex_equal(c._file_system_blacklist, re.compile('test|iso9660$', re.I))
58-
assert_regex_equal(c._device_whitelist, re.compile('test', IGNORE_CASE))
59-
assert_regex_equal(c._device_blacklist, re.compile('test', IGNORE_CASE))
60-
assert_regex_equal(c._mount_point_whitelist, re.compile('test', IGNORE_CASE))
61-
assert_regex_equal(c._mount_point_blacklist, re.compile('test', IGNORE_CASE))
56+
assert_regex_equal(c._file_system_include, re.compile('test', re.I))
57+
assert_regex_equal(c._file_system_exclude, re.compile('test|iso9660$', re.I))
58+
assert_regex_equal(c._device_include, re.compile('test', IGNORE_CASE))
59+
assert_regex_equal(c._device_exclude, re.compile('test', IGNORE_CASE))
60+
assert_regex_equal(c._mount_point_include, re.compile('test', IGNORE_CASE))
61+
assert_regex_equal(c._mount_point_exclude, re.compile('test', IGNORE_CASE))
6262

6363

6464
def test_exclude_bad_devices():
@@ -76,8 +76,8 @@ def test_exclude_cdrom():
7676
assert c.exclude_disk(MockPart(opts='rw,cdrom')) is True
7777

7878

79-
def test_file_system_whitelist():
80-
instance = {'file_system_whitelist': ['ext[34]', 'ntfs']}
79+
def test_file_system_include():
80+
instance = {'file_system_include': ['ext[34]', 'ntfs']}
8181
c = Disk('disk', {}, [instance])
8282

8383
assert c.exclude_disk(MockPart(fstype='ext3')) is False
@@ -86,16 +86,16 @@ def test_file_system_whitelist():
8686
assert c.exclude_disk(MockPart(fstype='apfs')) is True
8787

8888

89-
def test_file_system_blacklist():
90-
instance = {'file_system_blacklist': ['fat']}
89+
def test_file_system_exclude():
90+
instance = {'file_system_exclude': ['fat']}
9191
c = Disk('disk', {}, [instance])
9292

9393
assert c.exclude_disk(MockPart(fstype='FAT32')) is True
9494
assert c.exclude_disk(MockPart(fstype='zfs')) is False
9595

9696

97-
def test_file_system_whitelist_blacklist():
98-
instance = {'file_system_whitelist': ['ext[2-4]'], 'file_system_blacklist': ['ext2']}
97+
def test_file_system_include_exclude():
98+
instance = {'file_system_include': ['ext[2-4]'], 'file_system_exclude': ['ext2']}
9999
c = Disk('disk', {}, [instance])
100100

101101
assert c.exclude_disk(MockPart(fstype='ext2')) is True
@@ -104,8 +104,8 @@ def test_file_system_whitelist_blacklist():
104104
assert c.exclude_disk(MockPart(fstype='NTFS')) is True
105105

106106

107-
def test_device_whitelist():
108-
instance = {'device_whitelist': ['/dev/sda[1-3]', 'c:']}
107+
def test_device_include():
108+
instance = {'device_include': ['/dev/sda[1-3]', 'c:']}
109109
c = Disk('disk', {}, [instance])
110110

111111
assert c.exclude_disk(MockPart(device='/dev/sda1')) is False
@@ -118,8 +118,8 @@ def test_device_whitelist():
118118
assert c.exclude_disk(MockPart(device='path\\c:\\')) is True
119119

120120

121-
def test_device_blacklist():
122-
instance = {'device_blacklist': ['/dev/sda[1-3]']}
121+
def test_device_exclude():
122+
instance = {'device_exclude': ['/dev/sda[1-3]']}
123123
c = Disk('disk', {}, [instance])
124124

125125
assert c.exclude_disk(MockPart(device='/dev/sda1')) is True
@@ -128,8 +128,8 @@ def test_device_blacklist():
128128
assert c.exclude_disk(MockPart(device='/dev/sda4')) is False
129129

130130

131-
def test_device_whitelist_blacklist():
132-
instance = {'device_whitelist': ['/dev/sda[1-3]'], 'device_blacklist': ['/dev/sda3']}
131+
def test_device_include_exclude():
132+
instance = {'device_include': ['/dev/sda[1-3]'], 'device_exclude': ['/dev/sda3']}
133133
c = Disk('disk', {}, [instance])
134134

135135
assert c.exclude_disk(MockPart(device='/dev/sda1')) is False
@@ -138,8 +138,8 @@ def test_device_whitelist_blacklist():
138138
assert c.exclude_disk(MockPart(device='/dev/sda4')) is True
139139

140140

141-
def test_mount_point_whitelist():
142-
instance = {'mount_point_whitelist': ['/dev/sda[1-3]', 'c:']}
141+
def test_mount_point_include():
142+
instance = {'mount_point_include': ['/dev/sda[1-3]', 'c:']}
143143
c = Disk('disk', {}, [instance])
144144

145145
assert c.exclude_disk(MockPart(mountpoint='/dev/sda1')) is False
@@ -152,8 +152,8 @@ def test_mount_point_whitelist():
152152
assert c.exclude_disk(MockPart(mountpoint='path\\c:\\')) is True
153153

154154

155-
def test_mount_point_blacklist():
156-
instance = {'mount_point_blacklist': ['/dev/sda[1-3]']}
155+
def test_mount_point_exclude():
156+
instance = {'mount_point_exclude': ['/dev/sda[1-3]']}
157157
c = Disk('disk', {}, [instance])
158158

159159
assert c.exclude_disk(MockPart(mountpoint='/dev/sda1')) is True
@@ -162,8 +162,8 @@ def test_mount_point_blacklist():
162162
assert c.exclude_disk(MockPart(mountpoint='/dev/sda4')) is False
163163

164164

165-
def test_mount_point_whitelist_blacklist():
166-
instance = {'mount_point_whitelist': ['/dev/sda[1-3]'], 'mount_point_blacklist': ['/dev/sda3']}
165+
def test_mount_point_include_exclude():
166+
instance = {'mount_point_include': ['/dev/sda[1-3]'], 'mount_point_exclude': ['/dev/sda3']}
167167
c = Disk('disk', {}, [instance])
168168

169169
assert c.exclude_disk(MockPart(mountpoint='/dev/sda1')) is False
@@ -173,72 +173,7 @@ def test_mount_point_whitelist_blacklist():
173173

174174

175175
def test_all_partitions_allow_no_device():
176-
instance = {'all_partitions': 'true', 'mount_point_blacklist': ['/run$']}
177-
c = Disk('disk', {}, [instance])
178-
179-
assert c.exclude_disk(MockPart(device='', mountpoint='/run')) is True
180-
assert c.exclude_disk(MockPart(device='', mountpoint='/run/shm')) is False
181-
182-
183-
def test_legacy_config():
184-
instance = {
185-
'excluded_filesystems': ['test', ''],
186-
'excluded_disks': ['test1', ''],
187-
'excluded_disk_re': 'test2',
188-
'excluded_mountpoint_re': 'test',
189-
}
190-
c = Disk('disk', {}, [instance])
191-
192-
assert_regex_equal(c._file_system_blacklist, re.compile('iso9660$|test$', re.I))
193-
assert_regex_equal(c._device_blacklist, re.compile('test1$|test2', IGNORE_CASE))
194-
assert_regex_equal(c._mount_point_blacklist, re.compile('test', IGNORE_CASE))
195-
196-
197-
def test_legacy_exclude_disk():
198-
"""
199-
Test legacy exclusion logic config
200-
"""
201-
instance = {
202-
'use_mount': 'no',
203-
'excluded_filesystems': ['aaaaaa'],
204-
'excluded_mountpoint_re': '^/run$',
205-
'excluded_disks': ['bbbbbb'],
206-
'excluded_disk_re': '^tev+$',
207-
}
208-
c = Disk('disk', {}, [instance])
209-
210-
# should pass, default mock is a normal disk
211-
assert c.exclude_disk(MockPart()) is False
212-
213-
# standard fake devices
214-
assert c.exclude_disk(MockPart(device='')) is True
215-
assert c.exclude_disk(MockPart(device='none')) is True
216-
assert c.exclude_disk(MockPart(device='udev')) is False
217-
218-
# excluded filesystems list
219-
assert c.exclude_disk(MockPart(fstype='aaaaaa')) is True
220-
assert c.exclude_disk(MockPart(fstype='a')) is False
221-
222-
# excluded devices list
223-
assert c.exclude_disk(MockPart(device='bbbbbb')) is True
224-
assert c.exclude_disk(MockPart(device='b')) is False
225-
226-
# excluded devices regex
227-
assert c.exclude_disk(MockPart(device='tevvv')) is True
228-
assert c.exclude_disk(MockPart(device='tevvs')) is False
229-
230-
# and now with all_partitions
231-
c._all_partitions = True
232-
assert c.exclude_disk(MockPart(device='')) is False
233-
assert c.exclude_disk(MockPart(device='none')) is False
234-
assert c.exclude_disk(MockPart(device='udev')) is False
235-
# excluded mountpoint regex
236-
assert c.exclude_disk(MockPart(device='sdz', mountpoint='/run')) is True
237-
assert c.exclude_disk(MockPart(device='sdz', mountpoint='/run/shm')) is False
238-
239-
240-
def test_legacy_device_exclusion_logic_no_name():
241-
instance = {'use_mount': 'yes', 'excluded_mountpoint_re': '^/run$', 'all_partitions': 'yes'}
176+
instance = {'all_partitions': 'true', 'mount_point_exclude': ['/run$']}
242177
c = Disk('disk', {}, [instance])
243178

244179
assert c.exclude_disk(MockPart(device='', mountpoint='/run')) is True

0 commit comments

Comments
 (0)