1 files changed, 139 insertions, 0 deletions
diff --git a/tests/storage_test/udev_test.py b/tests/storage_test/udev_test.py
new file mode 100644
index 000000000..5d48279b4
--- /dev/null
+++ b/tests/storage_test/udev_test.py
@@ -0,0 +1,139 @@
+#!/usr/bin/python
+
+import mock
+import os
+
+class UdevTest(mock.TestCase):
+
+    def setUp(self):
+        self.setupModules(["_isys", "block", "ConfigParser"])
+        self.fs = mock.DiskIO()
+
+        import pyanaconda.storage.udev
+        pyanaconda.storage.udev.os = mock.Mock()
+        pyanaconda.storage.udev.log = mock.Mock()
+        pyanaconda.storage.udev.open = self.fs.open
+
+    def tearDown(self):
+        self.tearDownModules()
+
+    def udev_enumerate_devices_test(self):
+        import pyanaconda.storage.udev
+        ENUMERATE_LIST = [
+            '/sys/devices/pci0000:00/0000:00:1f.2/host0/target0:0:0/0:0:0:0/block/sda',
+            '/sys/devices/virtual/block/loop0',
+            '/sys/devices/virtual/block/loop1',
+            '/sys/devices/virtual/block/ram0',
+            '/sys/devices/virtual/block/ram1',
+            '/sys/devices/virtual/block/dm-0',
+        ]
+
+        pyanaconda.storage.udev.global_udev.enumerate_devices = mock.Mock(return_value=ENUMERATE_LIST)
+        ret = pyanaconda.storage.udev.udev_enumerate_devices()
+        self.assertEqual(set(ret),
+            set(['/devices/pci0000:00/0000:00:1f.2/host0/target0:0:0/0:0:0:0/block/sda',
+            '/devices/virtual/block/loop0', '/devices/virtual/block/loop1',
+            '/devices/virtual/block/ram0', '/devices/virtual/block/ram1',
+            '/devices/virtual/block/dm-0'])
+        )
+
+    def udev_get_device_1_test(self):
+        import pyanaconda.storage.udev
+
+        class Device(object):
+            def __init__(self):
+                self.sysname = 'loop1'
+                self.dict = {'symlinks': ['/dev/block/7:1'],
+                    'SUBSYSTEM': 'block',
+                    'MAJOR': '7',
+                    'DEVPATH': '/devices/virtual/block/loop1',
+                    'UDISKS_PRESENTATION_NOPOLICY': '1',
+                    'UDEV_LOG': '3',
+                    'DEVNAME': '/dev/loop1',
+                    'DEVTYPE': 'disk',
+                    'DEVLINKS': '/dev/block/7:1',
+                    'MINOR': '1'
+                }
+
+            def __getitem__(self, key):
+                return self.dict[key]
+
+            def __setitem__(self, key, value):
+                self.dict[key] = value
+
+        pyanaconda.storage.udev.os.path.exists.return_value = True
+        DEV_PATH = '/devices/virtual/block/loop1'
+        dev = Device()
+        pyanaconda.storage.udev.global_udev = mock.Mock()
+        pyanaconda.storage.udev.global_udev.create_device.return_value = dev
+        pyanaconda.storage.udev.udev_parse_uevent_file = mock.Mock(return_value=dev)
+
+        ret = pyanaconda.storage.udev.udev_get_device(DEV_PATH)
+        self.assertTrue(isinstance(ret, Device))
+        self.assertEqual(ret['name'], ret.sysname)
+        self.assertEqual(ret['sysfs_path'], DEV_PATH)
+        self.assertTrue(pyanaconda.storage.udev.udev_parse_uevent_file.called)
+
+    def udev_get_device_2_test(self):
+        import pyanaconda.storage.udev
+        pyanaconda.storage.udev.os.path.exists.return_value = False
+        ret = pyanaconda.storage.udev.udev_get_device('')
+        self.assertEqual(ret, None)
+
+    def udev_get_device_3_test(self):
+        import pyanaconda.storage.udev
+        pyanaconda.storage.udev.os.path.exists.return_value = True
+        pyanaconda.storage.udev.global_udev = mock.Mock()
+        pyanaconda.storage.udev.global_udev.create_device.return_value = None
+        ret = pyanaconda.storage.udev.udev_get_device('')
+        self.assertEqual(ret, None)
+
+    def udev_get_devices_test(self):
+        import pyanaconda.storage.udev
+        pyanaconda.storage.udev.udev_settle = mock.Mock()
+        DEVS = \
+            ['/devices/pci0000:00/0000:00:1f.2/host0/target0:0:0/0:0:0:0/block/sda',
+            '/devices/virtual/block/loop0', '/devices/virtual/block/loop1',
+            '/devices/virtual/block/ram0', '/devices/virtual/block/ram1',
+            '/devices/virtual/block/dm-0']
+        pyanaconda.storage.udev.udev_enumerate_devices = mock.Mock(return_value=DEVS)
+        pyanaconda.storage.udev.udev_get_device = lambda x: x
+        ret = pyanaconda.storage.udev.udev_get_devices()
+        self.assertEqual(ret, DEVS)
+
+    def udev_parse_uevent_file_1_test(self):
+        import pyanaconda.storage.udev
+        pyanaconda.storage.udev.os.path.normpath = os.path.normpath
+        pyanaconda.storage.udev.os.access.return_value = True
+
+        FILE_CONTENT = "MAJOR=7\nMINOR=1\nDEVNAME=loop1\nDEVTYPE=disk\n"
+        self.fs.open('/sys/devices/virtual/block/loop1/uevent', 'w').write(FILE_CONTENT)
+        dev = {'sysfs_path': '/devices/virtual/block/loop1'}
+        ret = pyanaconda.storage.udev.udev_parse_uevent_file(dev)
+        self.assertEqual(ret,
+            {'sysfs_path': '/devices/virtual/block/loop1',
+            'DEVNAME': 'loop1',
+            'DEVTYPE': 'disk',
+            'MAJOR': '7',
+            'MINOR': '1'})
+
+    def udev_parse_uevent_file_2_test(self):
+        import pyanaconda.storage.udev
+        pyanaconda.storage.udev.os.path.normpath = os.path.normpath
+        pyanaconda.storage.udev.os.access.return_value = False
+
+        dev = {'sysfs_path': '/devices/virtual/block/loop1'}
+        ret = pyanaconda.storage.udev.udev_parse_uevent_file(dev)
+        self.assertEqual(ret, {'sysfs_path': '/devices/virtual/block/loop1'})
+
+    def udev_settle_test(self):
+        import pyanaconda.storage.udev
+        pyanaconda.storage.udev.util = mock.Mock()
+        pyanaconda.storage.udev.udev_settle()
+        self.assertTrue(pyanaconda.storage.udev.util.run_program.called)
+
+    def udev_trigger_test(self):
+        import pyanaconda.storage.udev
+        pyanaconda.storage.udev.util = mock.Mock()
+        pyanaconda.storage.udev.udev_trigger()
+        self.assertTrue(pyanaconda.storage.udev.util.run_program.called)