| #!/usr/bin/python2 |
| |
| """Tests for autotest_lib.client.bin.partition.""" |
| |
| from __future__ import absolute_import |
| from __future__ import division |
| from __future__ import print_function |
| |
| __author__ = 'gps@google.com (Gregory P. Smith)' |
| |
| import os, sys, unittest |
| from six import StringIO |
| import common |
| from autotest_lib.client.common_lib.test_utils import mock |
| from autotest_lib.client.bin import partition |
| from six.moves import range |
| |
| |
| class FsOptions_common(object): |
| def test_constructor(self): |
| self.assertRaises(ValueError, partition.FsOptions, '', '', '', '') |
| self.assertRaises(ValueError, partition.FsOptions, 'ext2', '', '', '') |
| obj = partition.FsOptions('ext2', 'ext2_vanilla', '', '') |
| obj = partition.FsOptions(fstype='ext2', fs_tag='ext2_vanilla') |
| obj = partition.FsOptions('fs', 'shortie', 'mkfs opts', 'mount opts') |
| self.assertEqual('fs', obj.fstype) |
| self.assertEqual('shortie', obj.fs_tag) |
| self.assertEqual('mkfs opts', obj.mkfs_flags) |
| self.assertEqual('mount opts', obj.mount_options) |
| |
| |
| def test__str__(self): |
| str_obj = str(partition.FsOptions('abc', 'def', 'ghi', 'jkl')) |
| self.assert_('FsOptions' in str_obj) |
| self.assert_('abc' in str_obj) |
| self.assert_('def' in str_obj) |
| self.assert_('ghi' in str_obj) |
| self.assert_('jkl' in str_obj) |
| |
| |
| # Test data used in GetPartitionTest below. |
| |
| SAMPLE_SWAPS = """ |
| Filename Type Size Used Priority |
| /dev/hdc2 partition 9863868 0 -1 |
| """ |
| |
| SAMPLE_PARTITIONS_HDC_ONLY = """ |
| major minor #blocks name |
| |
| 8 16 390711384 hdc |
| 8 18 530113 hdc2 |
| 8 19 390178687 hdc3 |
| """ |
| |
| # yes I manually added a hda1 line to this output to test parsing when the Boot |
| # flag exists. |
| SAMPLE_FDISK = "/sbin/fdisk -l -u '/dev/hdc'" |
| SAMPLE_FDISK_OUTPUT = """ |
| Disk /dev/hdc: 400.0 GB, 400088457216 bytes |
| 255 heads, 63 sectors/track, 48641 cylinders, total 781422768 sectors |
| Units = sectors of 1 * 512 = 512 bytes |
| |
| Device Boot Start End Blocks Id System |
| /dev/hdc2 63 1060289 530113+ 82 Linux swap / Solaris |
| /dev/hdc3 1060290 781417664 390178687+ 83 Linux |
| /dev/hdc4 * faketest FAKETEST 232323+ 83 Linux |
| """ |
| |
| |
| class get_partition_list_common(object): |
| def setUp(self): |
| self.god = mock.mock_god() |
| self.god.stub_function(os, 'popen') |
| |
| |
| def tearDown(self): |
| self.god.unstub_all() |
| |
| |
| def test_is_linux_fs_type(self): |
| for unused in range(4): |
| os.popen.expect_call(SAMPLE_FDISK).and_return( |
| StringIO(SAMPLE_FDISK_OUTPUT)) |
| self.assertFalse(partition.is_linux_fs_type('/dev/hdc1')) |
| self.assertFalse(partition.is_linux_fs_type('/dev/hdc2')) |
| self.assertTrue(partition.is_linux_fs_type('/dev/hdc3')) |
| self.assertTrue(partition.is_linux_fs_type('/dev/hdc4')) |
| self.god.check_playback() |
| |
| |
| def test_get_partition_list(self): |
| def fake_open(filename): |
| """Fake open() to pass to get_partition_list as __open.""" |
| if filename == '/proc/swaps': |
| return StringIO(SAMPLE_SWAPS) |
| elif filename == '/proc/partitions': |
| return StringIO(SAMPLE_PARTITIONS_HDC_ONLY) |
| else: |
| self.assertFalse("Unexpected open() call: %s" % filename) |
| |
| job = 'FakeJob' |
| |
| # Test a filter func that denies all. |
| parts = partition.get_partition_list(job, filter_func=lambda x: False, |
| open_func=fake_open) |
| self.assertEqual([], parts) |
| self.god.check_playback() |
| |
| # Test normal operation. |
| self.god.stub_function(partition, 'partition') |
| partition.partition.expect_call(job, '/dev/hdc3').and_return('3') |
| parts = partition.get_partition_list(job, open_func=fake_open) |
| self.assertEqual(['3'], parts) |
| self.god.check_playback() |
| |
| # Test exclude_swap can be disabled. |
| partition.partition.expect_call(job, '/dev/hdc2').and_return('2') |
| partition.partition.expect_call(job, '/dev/hdc3').and_return('3') |
| parts = partition.get_partition_list(job, exclude_swap=False, |
| open_func=fake_open) |
| self.assertEqual(['2', '3'], parts) |
| self.god.check_playback() |
| |
| # Test that min_blocks works. |
| partition.partition.expect_call(job, '/dev/hdc3').and_return('3') |
| parts = partition.get_partition_list(job, min_blocks=600000, |
| exclude_swap=False, |
| open_func=fake_open) |
| self.assertEqual(['3'], parts) |
| self.god.check_playback() |
| |
| |
| # we want to run the unit test suite once strictly on the non site specific |
| # version of partition (ie on base_partition.py) and once on the version |
| # that would result after the site specific overrides take place in order |
| # to check that the overrides to not break expected functionality of the |
| # non site specific code |
| class FSOptions_base_test(FsOptions_common, unittest.TestCase): |
| def setUp(self): |
| sys.modules['autotest_lib.client.bin.site_partition'] = None |
| reload(partition) |
| |
| |
| class get_partition_list_base_test(get_partition_list_common, unittest.TestCase): |
| def setUp(self): |
| sys.modules['autotest_lib.client.bin.site_partition'] = None |
| reload(partition) |
| get_partition_list_common.setUp(self) |
| |
| |
| class FSOptions_test(FsOptions_common, unittest.TestCase): |
| def setUp(self): |
| if 'autotest_lib.client.bin.site_partition' in sys.modules: |
| del sys.modules['autotest_lib.client.bin.site_partition'] |
| reload(partition) |
| |
| |
| class get_partition_list_test(get_partition_list_common, unittest.TestCase): |
| def setUp(self): |
| if 'autotest_lib.client.bin.site_partition' in sys.modules: |
| del sys.modules['autotest_lib.client.bin.site_partition'] |
| reload(partition) |
| get_partition_list_common.setUp(self) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
