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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#!/usr/bin/env python
#
# Copyright (C) 2020 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
"""Unit testing paycheck.py."""
# This test requires new (Y) and old (X) images, as well as a full payload
# from image Y and a delta payload from Y to X for each partition.
# Payloads are from sample_images/generate_payloads.
#
# The test performs the following:
#
# - It statically applies the full and delta payloads.
#
# - It applies full_payload to yield a new kernel (kern.part) and rootfs
# (root.part) and compares them to the new image partitions.
#
# - It applies delta_payload to the old image to yield a new kernel and rootfs
# and compares them to the new image partitions.
#
# Previously test_paycheck.sh. Run with update_payload ebuild.
# Disable check for function names to avoid errors based on old code
# pylint: disable-msg=invalid-name
import filecmp
import os
import subprocess
import unittest
class PaycheckTest(unittest.TestCase):
"""Test paycheck functions."""
def setUp(self):
self.tmpdir = os.getenv('T')
self._full_payload = os.path.join(self.tmpdir, 'full_payload.bin')
self._delta_payload = os.path.join(self.tmpdir, 'delta_payload.bin')
self._new_kernel = os.path.join(self.tmpdir, 'disk_ext2_4k.img')
self._new_root = os.path.join(self.tmpdir, 'disk_sqfs_default.img')
self._old_kernel = os.path.join(self.tmpdir,
'disk_ext2_4k_empty.img')
self._old_root = os.path.join(self.tmpdir, 'disk_sqfs_empty.img')
# Temp output files.
self._kernel_part = os.path.join(self.tmpdir, 'kern.part')
self._root_part = os.path.join(self.tmpdir, 'root.part')
def checkPayload(self, type_arg, payload):
"""Checks Payload."""
self.assertEqual(0, subprocess.check_call(['./paycheck.py', '-t',
type_arg, payload]))
def testFullPayload(self):
"""Checks the full payload statically."""
self.checkPayload('full', self._full_payload)
def testDeltaPayload(self):
"""Checks the delta payload statically."""
self.checkPayload('delta', self._delta_payload)
def testApplyFullPayload(self):
"""Applies full payloads and compares results to new sample images."""
self.assertEqual(0, subprocess.check_call(['./paycheck.py',
self._full_payload,
'--part_names', 'kernel', 'root',
'--out_dst_part_paths',
self._kernel_part,
self._root_part]))
# Check if generated full image is equal to sample image.
self.assertTrue(filecmp.cmp(self._kernel_part, self._new_kernel))
self.assertTrue(filecmp.cmp(self._root_part, self._new_root))
def testApplyDeltaPayload(self):
"""Applies delta to old image and checks against new sample images."""
self.assertEqual(0, subprocess.check_call(['./paycheck.py',
self._delta_payload,
'--part_names', 'kernel', 'root',
'--src_part_paths',
self._old_kernel, self._old_root,
'--out_dst_part_paths',
self._kernel_part,
self._root_part]))
self.assertTrue(filecmp.cmp(self._kernel_part, self._new_kernel))
self.assertTrue(filecmp.cmp(self._root_part, self._new_root))
if __name__ == '__main__':
unittest.main()
|
