| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 2 | 2 |
| 3 import unittest, os, time, re, glob, logging | 3 import unittest, os, time, re, glob, logging |
| 4 import common | 4 import common |
| 5 from autotest_lib.client.common_lib.test_utils import mock | 5 from autotest_lib.client.common_lib.test_utils import mock |
| 6 from autotest_lib.client.bin import kernel, job, utils, kernelexpand | 6 from autotest_lib.client.bin import kernel, job, utils, kernelexpand |
| 7 from autotest_lib.client.bin import kernel_config, boottool, os_dep | 7 from autotest_lib.client.bin import kernel_config, boottool, os_dep |
| 8 | 8 |
| 9 | 9 |
| 10 class TestAddKernelToBootLoader(unittest.TestCase): | 10 class TestAddKernelToBootLoader(unittest.TestCase): |
| 11 | 11 |
| 12 def add_to_bootloader(self, base_args, args, bootloader_args, | 12 def add_to_bootloader(self, base_args, args, bootloader_args, |
| 13 bootloader_root, tag='image', image='image', | 13 bootloader_root, tag='image', image='image', |
| 14 initrd='initrd'): | 14 initrd='initrd'): |
| 15 god = mock.mock_god() | 15 god = mock.mock_god() |
| 16 bootloader = god.create_mock_class(boottool.boottool, "boottool") | 16 bootloader = god.create_mock_class(boottool.boottool, "boottool") |
| 17 | 17 |
| 18 # record | 18 # record |
| 19 bootloader.remove_kernel.expect_call(tag) | 19 bootloader.remove_kernel.expect_call(tag) |
| 20 bootloader.add_kernel.expect_call(image, tag, initrd=initrd, | 20 bootloader.add_kernel.expect_call(image, tag, initrd=initrd, |
| 21 args=bootloader_args, | 21 args='_dummy_', root=bootloader_root) |
| 22 root=bootloader_root) | 22 |
| 23 for a in bootloader_args.split(): |
| 24 bootloader.add_args.expect_call(kernel=tag, args=a) |
| 25 bootloader.remove_args.expect_call(kernel=tag, args='_dummy_') |
| 23 | 26 |
| 24 # run and check | 27 # run and check |
| 25 kernel._add_kernel_to_bootloader(bootloader, base_args, tag, args, | 28 kernel._add_kernel_to_bootloader(bootloader, base_args, tag, args, |
| 26 image, initrd) | 29 image, initrd) |
| 27 god.check_playback() | 30 god.check_playback() |
| 28 | 31 |
| 29 | 32 |
| 30 def test_add_kernel_to_bootloader(self): | 33 def test_add_kernel_to_bootloader(self): |
| 31 self.add_to_bootloader(base_args='baseargs', args='', | 34 self.add_to_bootloader(base_args='baseargs', args='', |
| 32 bootloader_args='baseargs', bootloader_root=None) | 35 bootloader_args='baseargs', bootloader_root=None) |
| (...skipping 435 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 468 | 471 |
| 469 | 472 |
| 470 def test_build_timed(self): | 473 def test_build_timed(self): |
| 471 self.construct_kernel() | 474 self.construct_kernel() |
| 472 self.god.stub_function(self.kernel, "set_cross_cc") | 475 self.god.stub_function(self.kernel, "set_cross_cc") |
| 473 self.god.stub_function(self.kernel, "clean") | 476 self.god.stub_function(self.kernel, "clean") |
| 474 | 477 |
| 475 # record | 478 # record |
| 476 os.chdir.expect_call(self.build_dir) | 479 os.chdir.expect_call(self.build_dir) |
| 477 self.kernel.set_cross_cc.expect_call() | 480 self.kernel.set_cross_cc.expect_call() |
| 478 self.kernel.clean.expect_call(logged=False) | 481 self.kernel.clean.expect_call() |
| 479 build_string = "/usr/bin/time -o /dev/null make -j 8 vmlinux" | 482 build_string = "/usr/bin/time -o /dev/null make -j 8 vmlinux" |
| 480 build_string += ' > /dev/null 2>&1' | 483 build_string += ' > /dev/null 2>&1' |
| 481 utils.system.expect_call(build_string) | 484 utils.system.expect_call(build_string) |
| 482 os.path.isfile.expect_call('vmlinux').and_return(True) | 485 os.path.isfile.expect_call('vmlinux').and_return(True) |
| 483 | 486 |
| 484 # run and check | 487 # run and check |
| 485 self.kernel.build_timed(threads=8) | 488 self.kernel.build_timed(threads=8) |
| 486 self.god.check_playback() | 489 self.god.check_playback() |
| 487 | 490 |
| 488 | 491 |
| (...skipping 13 matching lines...) Expand all Loading... |
| 502 def test_mkinitrd(self): | 505 def test_mkinitrd(self): |
| 503 self.construct_kernel() | 506 self.construct_kernel() |
| 504 | 507 |
| 505 # record | 508 # record |
| 506 utils.get_os_vendor.expect_call().and_return('Ubuntu') | 509 utils.get_os_vendor.expect_call().and_return('Ubuntu') |
| 507 os.path.isfile.expect_call('initrd').and_return(True) | 510 os.path.isfile.expect_call('initrd').and_return(True) |
| 508 os.remove.expect_call('initrd') | 511 os.remove.expect_call('initrd') |
| 509 self.job.config_get.expect_call( | 512 self.job.config_get.expect_call( |
| 510 'kernel.mkinitrd_extra_args').and_return(None) | 513 'kernel.mkinitrd_extra_args').and_return(None) |
| 511 args = '' | 514 args = '' |
| 515 glob.glob.expect_call('/lib/modules/2.6.24*').and_return(['2.6.24']) |
| 512 os.path.isfile.expect_call('/usr/sbin/mkinitrd').and_return(True) | 516 os.path.isfile.expect_call('/usr/sbin/mkinitrd').and_return(True) |
| 513 cmd = '/usr/sbin/mkinitrd' | 517 cmd = '/usr/sbin/mkinitrd' |
| 514 utils.system.expect_call('%s %s -o initrd 2.6.24' % (cmd, args)) | 518 utils.system.expect_call('%s %s -o initrd 2.6.24' % (cmd, args)) |
| 515 self.job.record.expect_call('GOOD', self.subdir, 'kernel.mkinitrd') | 519 self.job.record.expect_call('GOOD', self.subdir, 'kernel.mkinitrd') |
| 516 | 520 |
| 517 # run and check | 521 # run and check |
| 518 self.kernel.mkinitrd(version="2.6.24", image="image", | 522 self.kernel.mkinitrd(version="2.6.24", image="image", |
| 519 system_map="system_map", initrd="initrd") | 523 system_map="system_map", initrd="initrd") |
| 520 self.god.check_playback() | 524 self.god.check_playback() |
| 521 | 525 |
| (...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 635 args, ident, expected_ident, | 639 args, ident, expected_ident, |
| 636 self.subdir, self.kernel.applied_patches) | 640 self.subdir, self.kernel.applied_patches) |
| 637 | 641 |
| 638 # run and check | 642 # run and check |
| 639 self.kernel.boot(args=args, ident=ident) | 643 self.kernel.boot(args=args, ident=ident) |
| 640 self.god.check_playback() | 644 self.god.check_playback() |
| 641 | 645 |
| 642 | 646 |
| 643 if __name__ == "__main__": | 647 if __name__ == "__main__": |
| 644 unittest.main() | 648 unittest.main() |
| OLD | NEW |