Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(77)

Side by Side Diff: client/bin/kernel_unittest.py

Issue 4823005: Merge remote branch 'cros/upstream' into tempbranch (Closed) Base URL: http://git.chromium.org/git/autotest.git@master
Patch Set: patch Created 10 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « client/bin/kernel.py ('k') | client/bin/partition.py » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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
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
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
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()
OLDNEW
« no previous file with comments | « client/bin/kernel.py ('k') | client/bin/partition.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698