def test_compare_raise(self):
matcher = drive_selection.SizeMatcher('size', 'None')
disk_dict = Device(path='/dev/vdb', sys_api=dict(size='20.00 GB'))
with pytest.raises(Exception):
matcher.compare(disk_dict)
pytest.fail("Couldn't parse size")
def test_parse_filter_min_low(self):
""" Testing low notation with 20G: """
matcher = drive_selection.SizeMatcher('size', '50G:')
assert isinstance(matcher.exact, tuple)
assert matcher.low[0] == '50'
assert matcher.low[1] == 'GB'
def test_parse_filter_max_high(self):
""" Testing high notation with :50G """
matcher = drive_selection.SizeMatcher('size', ':50G')
assert isinstance(matcher.exact, tuple)
assert matcher.high[0] == '50'
assert matcher.high[1] == 'GB'
def test_parse_filter_high_low(self):
""" Testing high-low notation with 20G:50G """
matcher = drive_selection.SizeMatcher('size', '20G:50G')
assert isinstance(matcher.exact, tuple)
assert matcher.low[0] == '20'
assert matcher.high[0] == '50'
assert matcher.low[1] == 'GB'
assert matcher.high[1] == 'GB'
def test_parse_filter_exact_GB_G(self):
""" Testing exact notation with 20G """
matcher = drive_selection.SizeMatcher('size', '20GB')
assert isinstance(matcher.exact, tuple)
assert matcher.exact[0] == '20'
assert matcher.exact[1] == 'GB'
def test_normalize_suffix_raises(self):
with pytest.raises(ValueError):
drive_selection.SizeMatcher('10P', 'size')._normalize_suffix("P")
pytest.fail("Unit 'P' not supported")
def test_normalize_suffix(self, test_input, expected):
assert drive_selection.SizeMatcher(
'10G', 'size')._normalize_suffix(test_input) == expected
def test_parse_suffix(self, test_input, expected):
assert drive_selection.SizeMatcher(
'size', '10G')._parse_suffix(test_input) == expected
def test_get_k_v(self, test_input, expected):
assert drive_selection.SizeMatcher(
'size', '10G')._get_k_v(test_input) == expected
def test_compare_at_least_1TB(self, test_input, expected):
matcher = drive_selection.SizeMatcher('size', '1TB:')
disk_dict = Device(path='/dev/sdz', sys_api=dict(size=test_input))
ret = matcher.compare(disk_dict)
assert ret is expected
def test_compare_high(self, test_input, expected):
matcher = drive_selection.SizeMatcher('size', ':50GB')
disk_dict = Device(path='/dev/vdb', sys_api=dict(size=test_input))
ret = matcher.compare(disk_dict)
assert ret is expected
def test_compare_exact(self):
matcher = drive_selection.SizeMatcher('size', '20GB')
disk_dict = Device(path='/dev/vdb', sys_api=dict(size='20.00 GB'))
ret = matcher.compare(disk_dict)
assert ret is True
def test_to_byte_PB(self):
""" Expect to raise """
with pytest.raises(ValueError):
drive_selection.SizeMatcher('size', '10P').to_byte(('10', 'PB'))
assert 'Unit \'P\' is not supported'
def test_to_byte_TB(self):
""" Pretty nonesense test.."""
ret = drive_selection.SizeMatcher('size', '10T').to_byte(('10', 'TB'))
assert ret == 10 * 1e+12
