ceph/src/ceph-volume/ceph_volume/tests/util/test_disk.py

   1 from ceph_volume.util import disk
   2
   3
   4 class TestLsblkParser(object):
   5
   6     def test_parses_whitespace_values(self):
   7         output = 'NAME="sdaa5" PARTLABEL="ceph data" RM="0" SIZE="10M" RO="0" TYPE="part"'
   8         result = disk._lsblk_parser(output)
   9         assert result['PARTLABEL'] == 'ceph data'
  10
  11     def test_ignores_bogus_pairs(self):
  12         output = 'NAME="sdaa5" PARTLABEL RM="0" SIZE="10M" RO="0" TYPE="part" MOUNTPOINT=""'
  13         result = disk._lsblk_parser(output)
  14         assert result['SIZE'] == '10M'
  15
  16
  17 class TestDeviceFamily(object):
  18
  19     def test_groups_multiple_devices(self, stub_call):
  20         out = [
  21             'NAME="sdaa5" PARLABEL="ceph lockbox"',
  22             'NAME="sdaa" RO="0"',
  23             'NAME="sdaa1" PARLABEL="ceph data"',
  24             'NAME="sdaa2" PARLABEL="ceph journal"',
  25         ]
  26         stub_call((out, '', 0))
  27         result = disk.device_family('sdaa5')
  28         assert len(result) == 4
  29
  30     def test_parses_output_correctly(self, stub_call):
  31         names = ['sdaa', 'sdaa5', 'sdaa1', 'sdaa2']
  32         out = [
  33             'NAME="sdaa5" PARLABEL="ceph lockbox"',
  34             'NAME="sdaa" RO="0"',
  35             'NAME="sdaa1" PARLABEL="ceph data"',
  36             'NAME="sdaa2" PARLABEL="ceph journal"',
  37         ]
  38         stub_call((out, '', 0))
  39         result = disk.device_family('sdaa5')
  40         for parsed in result:
  41             assert parsed['NAME'] in names