Unit tests for mdraid lib

repos/system_upgrade/common/libraries/tests/test_mdraid.py

@@ -0,0 +1,78 @@
+import os
+
+import pytest
+
+from leapp.libraries.common import mdraid
+from leapp.libraries.common.testutils import logger_mocked
+from leapp.libraries.stdlib import api, CalledProcessError
+
+MD_DEVICE = '/dev/md0'
+NOT_MD_DEVICE = '/dev/sda'
+
+CUR_DIR = os.path.dirname(os.path.abspath(__file__))
+
+
+def raise_call_error(args=None):
+    raise CalledProcessError(
+        message='A Leapp Command Error occurred.',
+        command=args,
+        result={'signal': None, 'exit_code': 1, 'pid': 0, 'stdout': 'fake', 'stderr': 'fake'}
+    )
+
+
+class RunMocked(object):
+
+    def __init__(self, raise_err=False):
+        self.called = 0
+        self.args = None
+        self.raise_err = raise_err
+
+    def __call__(self, args, encoding=None):
+        self.called += 1
+        self.args = args
+        if self.raise_err:
+            raise_call_error(args)
+
+        if self.args == ['mdadm', '--query', MD_DEVICE]:
+            stdout = '/dev/md0: 1022.00MiB raid1 2 devices, 0 spares. Use mdadm --detail for more detail.'
+        elif self.args == ['mdadm', '--query', NOT_MD_DEVICE]:
+            stdout = '/dev/sda: is not an md array'
+
+        elif self.args == ['mdadm', '--detail', '--verbose', '--brief', MD_DEVICE]:
+            stdout = 'ARRAY /dev/md0 level=raid1 num-devices=2 metadata=1.2 name=localhost.localdomain:0 UUID=c4acea6e:d56e1598:91822e3f:fb26832c\n    devices=/dev/sda1,/dev/sdb1'  # noqa: E501; pylint: disable=line-too-long
+        elif self.args == ['mdadm', '--detail', '--verbose', '--brief', NOT_MD_DEVICE]:
+            stdout = 'mdadm: /dev/sda does not appear to be an md device'
+
+        return {'stdout': stdout}
+
+
+@pytest.mark.parametrize('dev,expected', [(MD_DEVICE, True), (NOT_MD_DEVICE, False)])
+def test_is_mdraid_dev(monkeypatch, dev, expected):
+    run_mocked = RunMocked()
+    monkeypatch.setattr(mdraid, 'run', run_mocked)
+    monkeypatch.setattr(api, 'current_logger', logger_mocked())
+
+    result = mdraid.is_mdraid_dev(dev)
+    assert mdraid.run.called == 1
+    assert expected == result
+    assert not api.current_logger.warnmsg
+
+
+def test_get_component_devices_ok(monkeypatch):
+    run_mocked = RunMocked()
+    monkeypatch.setattr(mdraid, 'run', run_mocked)
+    monkeypatch.setattr(api, 'current_logger', logger_mocked())
+
+    result = mdraid.get_component_devices(MD_DEVICE)
+    assert mdraid.run.called == 1
+    assert ['/dev/sda1', '/dev/sdb1'] == result
+    assert not api.current_logger.warnmsg
+
+
+def test_get_component_devices_not_md_device(monkeypatch):
+    run_mocked = RunMocked()
+    monkeypatch.setattr(mdraid, 'run', run_mocked)
+
+    with pytest.raises(ValueError):
+        mdraid.get_component_devices(NOT_MD_DEVICE)
+    assert mdraid.run.called == 1