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 |