Fixing unit tests for mdadm

This commit is contained in:
Nitin Madhok 2014-12-17 13:46:34 -05:00
parent 48c5b32b81
commit 502b340553

View File

@ -28,34 +28,34 @@ class MdadmTestCase(TestCase):
with patch.dict(mdadm.__salt__, {'cmd.run': mock}):
ret = mdadm.create(
'/dev/md0', 5,
['/dev/sdb1',
devices="['/dev/sdb1',
'/dev/sdc1',
'/dev/sdd1'],
'/dev/sdd1']",
test_mode=False,
force=True,
chunk=256
)
self.assertEqual('salt', ret)
mock.assert_called_once_with(
['mdadm', '-C', '/dev/md0', '-v', '--chunk', 256, '--force',
'-l', 5, '-e', 'default', '-n', 3, '/dev/sdb1', '/dev/sdc1',
'/dev/sdd1'], python_shell=False)
'mdadm -C /dev/md0 -v --chunk 256 --force
-l 5 -e default -n 3 /dev/sdb1 /dev/sdc1
/dev/sdd1', python_shell=False)
def test_create_test_mode(self):
mock = MagicMock()
with patch.dict(mdadm.__salt__, {'cmd.run': mock}):
ret = mdadm.create(
'/dev/md0', 5,
['/dev/sdb1',
devices="['/dev/sdb1',
'/dev/sdc1',
'/dev/sdd1'],
'/dev/sdd1']",
force=True,
chunk=256,
test_mode=True
)
self.assertEqual(['mdadm', '-C', '/dev/md0', '-v', '--chunk', 256,
'--force', '-l', 5, '-e', 'default', '-n', 3,
'/dev/sdb1', '/dev/sdc1', '/dev/sdd1'], ret)
self.assertEqual(['mdadm -C /dev/md0 -v --chunk 256 '
'--force -l 5 -e default -n 3 '
'/dev/sdb1 /dev/sdc1 /dev/sdd1'], ret)
assert not mock.called, 'test mode failed, cmd.run called'
if __name__ == '__main__':