From 72a4d465bf5fc99ade6296eedab6181a42ed3a62 Mon Sep 17 00:00:00 2001 From: Zhomart Mukhamejanov Date: Thu, 26 Apr 2018 15:49:08 -0700 Subject: updater_sample: fix gen_update_config.py Fix gen_update_config.py. And add tests for gen_update_config.py. Test: python3 updater_sample/tools/gen_update_config_test.py Change-Id: I0d270f1d5d7635855ee9acb0dd094e083cd33e3a Signed-off-by: Zhomart Mukhamejanov --- updater_sample/tests/res/raw/ota_002_package.zip | Bin 748 -> 638 bytes .../tests/res/raw/update_config_stream_002.json | 28 +++++------ updater_sample/tools/gen_update_config.py | 33 ++++++++----- updater_sample/tools/gen_update_config_test.py | 55 +++++++++++++++++++++ 4 files changed, 89 insertions(+), 27 deletions(-) create mode 100755 updater_sample/tools/gen_update_config_test.py diff --git a/updater_sample/tests/res/raw/ota_002_package.zip b/updater_sample/tests/res/raw/ota_002_package.zip index 3bfe37757..145c62e6a 100644 Binary files a/updater_sample/tests/res/raw/ota_002_package.zip and b/updater_sample/tests/res/raw/ota_002_package.zip differ diff --git a/updater_sample/tests/res/raw/update_config_stream_002.json b/updater_sample/tests/res/raw/update_config_stream_002.json index 36c5722f8..f00f19ce6 100644 --- a/updater_sample/tests/res/raw/update_config_stream_002.json +++ b/updater_sample/tests/res/raw/update_config_stream_002.json @@ -1,35 +1,35 @@ { "__": "*** Generated using tools/gen_update_config.py ***", - "name": "S ota_002", - "streaming_metadata": { + "ab_install_type": "STREAMING", + "ab_streaming_metadata": { "property_files": [ { "filename": "payload.bin", - "offset": 195, - "size": 8 + "offset": 41, + "size": 7 }, { "filename": "payload_properties.txt", - "offset": 276, - "size": 19 + "offset": 100, + "size": 18 }, { "filename": "care_map.txt", - "offset": 42, - "size": 9 + "offset": 160, + "size": 8 }, { "filename": "compatibility.zip", - "offset": 119, - "size": 14 + "offset": 215, + "size": 13 }, { "filename": "metadata", - "offset": 375, - "size": 9 + "offset": 287, + "size": 8 } ] }, - "type": "STREAMING", + "name": "S ota_002_package", "url": "file:///data/sample-ota-packages/ota_002_package.zip" -} +} \ No newline at end of file diff --git a/updater_sample/tools/gen_update_config.py b/updater_sample/tools/gen_update_config.py index 44e9ac3cb..cb9bd0119 100755 --- a/updater_sample/tools/gen_update_config.py +++ b/updater_sample/tools/gen_update_config.py @@ -31,7 +31,7 @@ import sys import zipfile -class GenUpdateConfig(object): # pylint: disable=too-few-public-methods +class GenUpdateConfig(object): """ A class that generates update configuration file from an OTA package. @@ -43,9 +43,8 @@ class GenUpdateConfig(object): # pylint: disable=too-few-public-methods AB_INSTALL_TYPE_NON_STREAMING = 'NON_STREAMING' METADATA_NAME = 'META-INF/com/android/metadata' - def __init__(self, package, out, url, ab_install_type): + def __init__(self, package, url, ab_install_type): self.package = package - self.out = out self.url = url self.ab_install_type = ab_install_type self.streaming_required = ( @@ -59,14 +58,20 @@ class GenUpdateConfig(object): # pylint: disable=too-few-public-methods # compatibility.zip is available only if target supports Treble. 'compatibility.zip', ) + self._config = None + + @property + def config(self): + """Returns generated config object.""" + return self._config def run(self): - """generate config""" + """Generates config.""" streaming_metadata = None if self.ab_install_type == GenUpdateConfig.AB_INSTALL_TYPE_STREAMING: streaming_metadata = self._gen_ab_streaming_metadata() - config = { + self._config = { '__': '*** Generated using tools/gen_update_config.py ***', 'name': self.ab_install_type[0] + ' ' + os.path.basename(self.package)[:-4], 'url': self.url, @@ -74,12 +79,8 @@ class GenUpdateConfig(object): # pylint: disable=too-few-public-methods 'ab_install_type': self.ab_install_type, } - with open(self.out, 'w') as out: - json.dump(config, out, indent=4, separators=(',', ': '), sort_keys=True) - print('Config is written to ' + out.name) - def _gen_ab_streaming_metadata(self): - """Open zip file and get metadata for files required for streaming update.""" + """Builds metadata for files required for streaming update.""" with zipfile.ZipFile(self.package, 'r') as package_zip: property_files = self._get_property_files(package_zip) @@ -90,7 +91,7 @@ class GenUpdateConfig(object): # pylint: disable=too-few-public-methods return metadata def _get_property_files(self, zip_file): - """Constructs the property-files list for A/B streaming metadata""" + """Constructs the property-files list for A/B streaming metadata.""" def compute_entry_offset_size(name): """Computes the zip entry offset and size.""" @@ -115,8 +116,13 @@ class GenUpdateConfig(object): # pylint: disable=too-few-public-methods return property_files + def write(self, out): + """Writes config to the output file.""" + with open(out, 'w') as out_file: + json.dump(self.config, out_file, indent=4, separators=(',', ': '), sort_keys=True) + -def main(): # pylint: disable=missing-docstring +def main(): # pylint: disable=missing-docstring ab_install_type_choices = [ GenUpdateConfig.AB_INSTALL_TYPE_STREAMING, GenUpdateConfig.AB_INSTALL_TYPE_NON_STREAMING] @@ -144,10 +150,11 @@ def main(): # pylint: disable=missing-docstring gen = GenUpdateConfig( package=args.package, - out=args.out, url=args.url, ab_install_type=args.ab_install_type) gen.run() + gen.write(args.out) + print('Config is written to ' + args.out) if __name__ == '__main__': diff --git a/updater_sample/tools/gen_update_config_test.py b/updater_sample/tools/gen_update_config_test.py new file mode 100755 index 000000000..951d4c4a7 --- /dev/null +++ b/updater_sample/tools/gen_update_config_test.py @@ -0,0 +1,55 @@ +#!/usr/bin/env python3 +# +# Copyright (C) 2018 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. + +""" +Tests gen_update_config.py +""" + +import os.path +import unittest +from gen_update_config import GenUpdateConfig + + +class GenUpdateConfigTest(unittest.TestCase): # pylint: disable=missing-docstring + + def test_ab_install_type_streaming(self): + """tests if streaming property files' offset and size are generated properly""" + config, package = self._generate_config() + property_files = config['ab_streaming_metadata']['property_files'] + self.assertEqual(len(property_files), 5) + with open(package, 'rb') as pkg_file: + for prop in property_files: + filename, offset, size = prop['filename'], prop['offset'], prop['size'] + pkg_file.seek(offset) + data = pkg_file.read(size).decode('ascii') + # data in the archive are just uppercase filenames without extension + expected_data = filename.split('.')[0].upper() + self.assertEqual(data, expected_data) + + @staticmethod + def _generate_config(): + """Generates JSON config from ota_002_package.zip.""" + ota_package = os.path.join(os.path.dirname(__file__), + '../tests/res/raw/ota_002_package.zip') + gen = GenUpdateConfig(ota_package, + 'file:///foo.bar', + GenUpdateConfig.AB_INSTALL_TYPE_STREAMING) + gen.run() + return gen.config, ota_package + + +if __name__ == '__main__': + unittest.main() -- cgit v1.2.3