[ssm-devel] [PATCH 1/2] New mock functions for unittests

[Jan T. <jt...@re...>]

Add new mock functions to unittests, that were omitted before and where
the tests could get real, physical data instead of the mock ones.

Signed-off-by: Jan Tulak <jt...@re...>
---
 tests/unittests/common.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/tests/unittests/common.py b/tests/unittests/common.py
index 16cdb5a..c33bfc3 100644
--- a/tests/unittests/common.py
+++ b/tests/unittests/common.py
@@ -74,6 +74,10 @@ class MockSystemDataSource(unittest.TestCase):
         misc.run = self.mock_run
         self.get_partitions_orig = misc.get_partitions
         misc.get_partitions = self.mock_get_partitions
+        self.get_real_device_orig = misc.get_real_device
+        misc.get_real_device = self.mock_get_real_device
+        self.get_device_size_orig = misc.get_device_size
+        misc.get_device_size = self.mock_get_device_size
         self.is_bdevice_orig = misc.is_bdevice
         misc.is_bdevice = self.mock_is_bdevice
         self.is_directory_orig = main.is_directory
@@ -94,6 +98,7 @@ class MockSystemDataSource(unittest.TestCase):
         self.vol_data = {}
         self.pool_data = {}
         self.mount_data = {}
+        self.links = {}
         self._mpoint = False
         main.SSM_NONINTERACTIVE = True
 
@@ -106,6 +111,8 @@ class MockSystemDataSource(unittest.TestCase):
         self.mount_data = {}
         misc.run = self.run_orig
         misc.get_partitions = self.get_partitions_orig
+        misc.get_real_device = self.get_real_device_orig
+        misc.get_device_size = self.get_device_size_orig
         misc.is_bdevice = self.is_bdevice_orig
         main.is_directory = self.is_directory_orig
         misc.get_mounts = self.get_mounts_orig
@@ -154,6 +161,18 @@ class MockSystemDataSource(unittest.TestCase):
                               data['dev_name']])
         return partitions
 
+    def mock_get_real_device(self, devname):
+        for dev, data in self.dev_data.items():
+            if dev == devname:
+                return dev
+        for name, target in self.links.items():
+            if name == devname:
+                return self.mock_get_real_device(target)
+        return False
+
+    def mock_get_device_size(self, device):
+        return self.dev_data[device]['dev_size']
+
     def mock_is_directory(self, string):
         if string in self.directories:
             return string
@@ -201,6 +220,9 @@ class MockSystemDataSource(unittest.TestCase):
     def _addDir(self, dirname):
         self.directories.append(dirname)
 
+    def _addLink(self, target, name):
+        self.links[name] = target
+
     def _addDevice(self, dev_name, dev_size, minor=0):
         self.dev_data[dev_name] = {'dev_name': dev_name, 'dev_size': dev_size,
                 'major': '8', 'minor': str(minor)}
-- 
2.10.0