forked from yochow/autotest
/
deb_kernel_unittest.py
executable file
·78 lines (59 loc) · 2.77 KB
/
deb_kernel_unittest.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#!/usr/bin/python
import unittest, os
import common
from autotest_lib.client.common_lib.test_utils import mock
from autotest_lib.client.common_lib import utils as common_utils
from autotest_lib.server import deb_kernel, utils, hosts
from autotest_lib.server.hosts import bootloader
class TestDebKernel(unittest.TestCase):
def setUp(self):
self.god = mock.mock_god()
self.kernel = deb_kernel.DEBKernel()
self.host = self.god.create_mock_class(hosts.RemoteHost, "host")
self.host.bootloader = self.god.create_mock_class(
bootloader.Bootloader, "bootloader")
self.god.stub_function(utils, "run")
def tearDown(self):
self.god.unstub_all()
def common_code(self):
self.kernel.source_material = "source.rpm"
basename = os.path.basename(self.kernel.source_material)
self.remote_tmpdir = "remote/tmp/dir"
self.remote_filename = os.path.join(self.remote_tmpdir, basename)
self.host.get_tmp_dir.expect_call().and_return(self.remote_tmpdir)
self.host.send_file.expect_call(self.kernel.source_material,
self.remote_filename)
def test_install(self):
self.common_code()
# record
self.host.run.expect_call('dpkg -i "%s"' %
(utils.sh_escape(self.remote_filename)))
result = common_utils.CmdResult()
result.stdout = "1"
utils.run.expect_call('dpkg-deb -f "%s" version' %
utils.sh_escape(self.kernel.source_material)).and_return(result)
utils.run.expect_call('dpkg-deb -f "%s" version' %
utils.sh_escape(self.kernel.source_material)).and_return(result)
self.host.run.expect_call('mkinitramfs -o "/boot/initrd.img-1" "1"')
utils.run.expect_call('dpkg-deb -f "%s" version' %
utils.sh_escape(self.kernel.source_material)).and_return(result)
utils.run.expect_call('dpkg-deb -f "%s" version' %
utils.sh_escape(self.kernel.source_material)).and_return(result)
self.host.bootloader.add_kernel.expect_call('/boot/vmlinuz-1',
initrd='/boot/initrd.img-1')
# run and check
self.kernel.install(self.host)
self.god.check_playback()
def test_extract(self):
# setup
self.common_code()
content_dir= os.path.join(self.remote_tmpdir, "contents")
# record
self.host.run.expect_call('dpkg -x "%s" "%s"' %
(utils.sh_escape(self.remote_filename),
utils.sh_escape(content_dir)))
# run and test
self.kernel.extract(self.host)
self.god.check_playback()
if __name__ == "__main__":
unittest.main()