From 5f8de423f190bbb79a62f804151bc24824fa32d8 Mon Sep 17 00:00:00 2001 From: "Matt A. Tobin" Date: Fri, 2 Feb 2018 04:16:08 -0500 Subject: Add m-esr52 at 52.6.0 --- testing/mozbase/mozversion/mozversion/__init__.py | 7 + testing/mozbase/mozversion/mozversion/errors.py | 30 ++ .../mozbase/mozversion/mozversion/mozversion.py | 340 +++++++++++++++++++++ testing/mozbase/mozversion/setup.py | 29 ++ testing/mozbase/mozversion/tests/manifest.ini | 4 + testing/mozbase/mozversion/tests/test_apk.py | 43 +++ testing/mozbase/mozversion/tests/test_b2g.py | 75 +++++ testing/mozbase/mozversion/tests/test_binary.py | 177 +++++++++++ testing/mozbase/mozversion/tests/test_sources.py | 85 ++++++ 9 files changed, 790 insertions(+) create mode 100644 testing/mozbase/mozversion/mozversion/__init__.py create mode 100644 testing/mozbase/mozversion/mozversion/errors.py create mode 100644 testing/mozbase/mozversion/mozversion/mozversion.py create mode 100644 testing/mozbase/mozversion/setup.py create mode 100644 testing/mozbase/mozversion/tests/manifest.ini create mode 100644 testing/mozbase/mozversion/tests/test_apk.py create mode 100644 testing/mozbase/mozversion/tests/test_b2g.py create mode 100644 testing/mozbase/mozversion/tests/test_binary.py create mode 100644 testing/mozbase/mozversion/tests/test_sources.py (limited to 'testing/mozbase/mozversion') diff --git a/testing/mozbase/mozversion/mozversion/__init__.py b/testing/mozbase/mozversion/mozversion/__init__.py new file mode 100644 index 000000000..7894bcb9c --- /dev/null +++ b/testing/mozbase/mozversion/mozversion/__init__.py @@ -0,0 +1,7 @@ +# flake8: noqa +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this file, +# You can obtain one at http://mozilla.org/MPL/2.0/. + +from .errors import * +from .mozversion import cli, get_version diff --git a/testing/mozbase/mozversion/mozversion/errors.py b/testing/mozbase/mozversion/mozversion/errors.py new file mode 100644 index 000000000..756e772d6 --- /dev/null +++ b/testing/mozbase/mozversion/mozversion/errors.py @@ -0,0 +1,30 @@ +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this file, +# You can obtain one at http://mozilla.org/MPL/2.0/. + + +class VersionError(Exception): + + def __init__(self, message): + Exception.__init__(self, message) + + +class AppNotFoundError(VersionError): + """Exception for the application not found""" + + def __init__(self, message): + VersionError.__init__(self, message) + + +class LocalAppNotFoundError(AppNotFoundError): + """Exception for local application not found""" + + def __init__(self, path): + AppNotFoundError.__init__(self, 'Application not found at: %s' % path) + + +class RemoteAppNotFoundError(AppNotFoundError): + """Exception for remote application not found""" + + def __init__(self, message): + AppNotFoundError.__init__(self, message) diff --git a/testing/mozbase/mozversion/mozversion/mozversion.py b/testing/mozbase/mozversion/mozversion/mozversion.py new file mode 100644 index 000000000..5dfcd306a --- /dev/null +++ b/testing/mozbase/mozversion/mozversion/mozversion.py @@ -0,0 +1,340 @@ +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this file, +# You can obtain one at http://mozilla.org/MPL/2.0/. + +import argparse +import ConfigParser +from StringIO import StringIO +import os +import re +import sys +import tempfile +import xml.dom.minidom +import zipfile + +import mozfile +import mozlog + +import errors + + +INI_DATA_MAPPING = (('application', 'App'), ('platform', 'Build')) + + +class Version(object): + + def __init__(self): + self._info = {} + self._logger = mozlog.get_default_logger(component='mozversion') + if not self._logger: + self._logger = mozlog.unstructured.getLogger('mozversion') + + def get_gecko_info(self, path): + for type, section in INI_DATA_MAPPING: + config_file = os.path.join(path, "%s.ini" % type) + if os.path.exists(config_file): + self._parse_ini_file(open(config_file), type, section) + else: + self._logger.warning('Unable to find %s' % config_file) + + def _parse_ini_file(self, fp, type, section): + config = ConfigParser.RawConfigParser() + config.readfp(fp) + name_map = {'codename': 'display_name', + 'milestone': 'version', + 'sourcerepository': 'repository', + 'sourcestamp': 'changeset'} + for key, value in config.items(section): + name = name_map.get(key, key).lower() + self._info['%s_%s' % (type, name)] = config.has_option( + section, key) and config.get(section, key) or None + + if not self._info.get('application_display_name'): + self._info['application_display_name'] = \ + self._info.get('application_name') + + +class LocalFennecVersion(Version): + + def __init__(self, path, **kwargs): + Version.__init__(self, **kwargs) + self.get_gecko_info(path) + + def get_gecko_info(self, path): + archive = zipfile.ZipFile(path, 'r') + archive_list = archive.namelist() + for type, section in INI_DATA_MAPPING: + filename = "%s.ini" % type + if filename in archive_list: + self._parse_ini_file(archive.open(filename), type, + section) + else: + self._logger.warning('Unable to find %s' % filename) + + if "package-name.txt" in archive_list: + self._info["package_name"] = \ + archive.open("package-name.txt").readlines()[0].strip() + + +class LocalVersion(Version): + + def __init__(self, binary, **kwargs): + Version.__init__(self, **kwargs) + + if binary: + # on Windows, the binary may be specified with or without the + # .exe extension + if not os.path.exists(binary) and not os.path.exists(binary + + '.exe'): + raise IOError('Binary path does not exist: %s' % binary) + path = os.path.dirname(os.path.realpath(binary)) + else: + path = os.getcwd() + + if not self.check_location(path): + if sys.platform == 'darwin': + resources_path = os.path.join(os.path.dirname(path), + 'Resources') + if self.check_location(resources_path): + path = resources_path + else: + raise errors.LocalAppNotFoundError(path) + else: + raise errors.LocalAppNotFoundError(path) + + self.get_gecko_info(path) + + def check_location(self, path): + return (os.path.exists(os.path.join(path, 'application.ini')) + and os.path.exists(os.path.join(path, 'platform.ini'))) + + +class B2GVersion(Version): + + def __init__(self, sources=None, **kwargs): + Version.__init__(self, **kwargs) + + sources = sources or \ + os.path.exists(os.path.join(os.getcwd(), 'sources.xml')) and \ + os.path.join(os.getcwd(), 'sources.xml') + + if sources and os.path.exists(sources): + sources_xml = xml.dom.minidom.parse(sources) + for element in sources_xml.getElementsByTagName('project'): + path = element.getAttribute('path') + changeset = element.getAttribute('revision') + if path in ['gaia', 'gecko', 'build']: + if path == 'gaia' and self._info.get('gaia_changeset'): + break + self._info['_'.join([path, 'changeset'])] = changeset + + def get_gaia_info(self, app_zip): + tempdir = tempfile.mkdtemp() + try: + gaia_commit = os.path.join(tempdir, 'gaia_commit.txt') + try: + zip_file = zipfile.ZipFile(app_zip.name) + with open(gaia_commit, 'w') as f: + f.write(zip_file.read('resources/gaia_commit.txt')) + except zipfile.BadZipfile: + self._logger.info('Unable to unzip application.zip, falling ' + 'back to system unzip') + from subprocess import call + call(['unzip', '-j', app_zip.name, 'resources/gaia_commit.txt', + '-d', tempdir]) + + with open(gaia_commit) as f: + changeset, date = f.read().splitlines() + self._info['gaia_changeset'] = re.match( + '^\w{40}$', changeset) and changeset or None + self._info['gaia_date'] = date + except KeyError: + self._logger.warning( + 'Unable to find resources/gaia_commit.txt in ' + 'application.zip') + finally: + mozfile.remove(tempdir) + + +class LocalB2GVersion(B2GVersion): + + def __init__(self, binary, sources=None, **kwargs): + B2GVersion.__init__(self, sources, **kwargs) + + if binary: + if not os.path.exists(binary): + raise IOError('Binary path does not exist: %s' % binary) + path = os.path.dirname(binary) + else: + if os.path.exists(os.path.join(os.getcwd(), 'application.ini')): + path = os.getcwd() + + self.get_gecko_info(path) + + zip_path = os.path.join( + path, 'gaia', 'profile', 'webapps', + 'settings.gaiamobile.org', 'application.zip') + if os.path.exists(zip_path): + with open(zip_path, 'rb') as zip_file: + self.get_gaia_info(zip_file) + else: + self._logger.warning('Error pulling gaia file') + + +class RemoteB2GVersion(B2GVersion): + + def __init__(self, sources=None, dm_type='adb', host=None, + device_serial=None, adb_host=None, adb_port=None, + **kwargs): + B2GVersion.__init__(self, sources, **kwargs) + + try: + import mozdevice + except ImportError: + self._logger.critical("mozdevice is required to get the version" + " of a remote device") + raise + + if dm_type == 'adb': + dm = mozdevice.DeviceManagerADB(deviceSerial=device_serial, + serverHost=adb_host, + serverPort=adb_port) + elif dm_type == 'sut': + if not host: + raise errors.RemoteAppNotFoundError( + 'A host for SUT must be supplied.') + dm = mozdevice.DeviceManagerSUT(host=host) + else: + raise errors.RemoteAppNotFoundError( + 'Unknown device manager type: %s' % dm_type) + + if not sources: + path = 'system/sources.xml' + if dm.fileExists(path): + sources = StringIO(dm.pullFile(path)) + else: + self._logger.info('Unable to find %s' % path) + + tempdir = tempfile.mkdtemp() + for ini in ('application', 'platform'): + with open(os.path.join(tempdir, '%s.ini' % ini), 'w') as f: + f.write(dm.pullFile('/system/b2g/%s.ini' % ini)) + f.flush() + self.get_gecko_info(tempdir) + mozfile.remove(tempdir) + + for path in ['/system/b2g', '/data/local']: + path += '/webapps/settings.gaiamobile.org/application.zip' + if dm.fileExists(path): + with tempfile.NamedTemporaryFile() as f: + dm.getFile(path, f.name) + self.get_gaia_info(f) + break + else: + self._logger.warning('Error pulling gaia file') + + build_props = dm.pullFile('/system/build.prop') + desired_props = { + 'ro.build.version.incremental': 'device_firmware_version_incremental', + 'ro.build.version.release': 'device_firmware_version_release', + 'ro.build.date.utc': 'device_firmware_date', + 'ro.product.device': 'device_id'} + for line in build_props.split('\n'): + if not line.strip().startswith('#') and '=' in line: + key, value = [s.strip() for s in line.split('=', 1)] + if key in desired_props.keys(): + self._info[desired_props[key]] = value + + if self._info.get('device_id', '').lower() == 'flame': + for prop in ['ro.boot.bootloader', 't2m.sw.version']: + value = dm.shellCheckOutput(['getprop', prop]) + if value: + self._info['device_firmware_version_base'] = value + break + + +def get_version(binary=None, sources=None, dm_type=None, host=None, + device_serial=None, adb_host=None, adb_port=None): + """ + Returns the application version information as a dict. You can specify + a path to the binary of the application or an Android APK file (to get + version information for Firefox for Android). If this is omitted then the + current directory is checked for the existance of an application.ini + file. If not found and that the binary path was not specified, then it is + assumed the target application is a remote Firefox OS instance. + + :param binary: Path to the binary for the application or Android APK file + :param sources: Path to the sources.xml file (Firefox OS) + :param dm_type: Device manager type. Must be 'adb' or 'sut' (Firefox OS) + :param host: Host address of remote Firefox OS instance (SUT) + :param device_serial: Serial identifier of Firefox OS device (ADB) + :param adb_host: Host address of ADB server + :param adb_port: Port of ADB server + """ + try: + if binary and zipfile.is_zipfile(binary) and 'AndroidManifest.xml' in \ + zipfile.ZipFile(binary, 'r').namelist(): + version = LocalFennecVersion(binary) + else: + version = LocalVersion(binary) + if version._info.get('application_name') == 'B2G': + version = LocalB2GVersion(binary, sources=sources) + except errors.LocalAppNotFoundError: + if binary: + # we had a binary argument, do not search for remote B2G + raise + version = RemoteB2GVersion(sources=sources, + dm_type=dm_type, + host=host, + adb_host=adb_host, + adb_port=adb_port, + device_serial=device_serial) + + for (key, value) in sorted(version._info.items()): + if value: + version._logger.info('%s: %s' % (key, value)) + + return version._info + + +def cli(args=sys.argv[1:]): + parser = argparse.ArgumentParser( + description='Display version information for Mozilla applications') + parser.add_argument( + '--binary', + help='path to application binary or apk') + fxos = parser.add_argument_group('Firefox OS') + fxos.add_argument( + '--sources', + help='path to sources.xml') + fxos.add_argument( + '--device', + help='serial identifier of device to target') + fxos.add_argument( + '--adb-host', + help='host running adb') + fxos.add_argument( + '--adb-port', + help='port running adb') + mozlog.commandline.add_logging_group( + parser, + include_formatters=mozlog.commandline.TEXT_FORMATTERS + ) + + args = parser.parse_args() + dm_type = os.environ.get('DM_TRANS', 'adb') + host = os.environ.get('TEST_DEVICE') + + mozlog.commandline.setup_logging( + 'mozversion', args, {'mach': sys.stdout}) + + get_version(binary=args.binary, + sources=args.sources, + dm_type=dm_type, + host=host, + device_serial=args.device, + adb_host=args.adb_host, + adb_port=args.adb_port) + +if __name__ == '__main__': + cli() diff --git a/testing/mozbase/mozversion/setup.py b/testing/mozbase/mozversion/setup.py new file mode 100644 index 000000000..09b027925 --- /dev/null +++ b/testing/mozbase/mozversion/setup.py @@ -0,0 +1,29 @@ +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this file, +# You can obtain one at http://mozilla.org/MPL/2.0/. + +from setuptools import setup + +PACKAGE_VERSION = '1.4' + + +setup(name='mozversion', + version=PACKAGE_VERSION, + description='Library to get version information for applications', + long_description='See http://mozbase.readthedocs.org', + classifiers=[], + keywords='mozilla', + author='Mozilla Automation and Testing Team', + author_email='tools@lists.mozilla.org', + url='https://wiki.mozilla.org/Auto-tools/Projects/Mozbase', + license='MPL', + packages=['mozversion'], + include_package_data=True, + zip_safe=False, + install_requires=['mozfile >= 1.0', 'mozlog >= 3.0'], + extras_require={'device': ['mozdevice >= 0.44']}, + entry_points=""" + # -*- Entry points: -*- + [console_scripts] + mozversion = mozversion:cli + """) diff --git a/testing/mozbase/mozversion/tests/manifest.ini b/testing/mozbase/mozversion/tests/manifest.ini new file mode 100644 index 000000000..3c034ea78 --- /dev/null +++ b/testing/mozbase/mozversion/tests/manifest.ini @@ -0,0 +1,4 @@ +[test_binary.py] +[test_sources.py] +[test_b2g.py] +[test_apk.py] \ No newline at end of file diff --git a/testing/mozbase/mozversion/tests/test_apk.py b/testing/mozbase/mozversion/tests/test_apk.py new file mode 100644 index 000000000..37fcb0bcf --- /dev/null +++ b/testing/mozbase/mozversion/tests/test_apk.py @@ -0,0 +1,43 @@ +#!/usr/bin/env python + +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this file, +# You can obtain one at http://mozilla.org/MPL/2.0/. + +import mozfile +import unittest +import zipfile +from mozversion import get_version + + +class ApkTest(unittest.TestCase): + """test getting version information from an android .apk""" + + application_changeset = 'a' * 40 + platform_changeset = 'b' * 40 + + def create_apk_zipfiles(self, zfile): + zfile.writestr('application.ini', + """[App]\nSourceStamp=%s\n""" % self.application_changeset) + zfile.writestr('platform.ini', + """[Build]\nSourceStamp=%s\n""" % self.platform_changeset) + zfile.writestr('AndroidManifest.xml', '') + + def test_basic(self): + with mozfile.NamedTemporaryFile() as f: + with zipfile.ZipFile(f.name, 'w') as z: + self.create_apk_zipfiles(z) + v = get_version(f.name) + self.assertEqual(v.get('application_changeset'), self.application_changeset) + self.assertEqual(v.get('platform_changeset'), self.platform_changeset) + + def test_with_package_name(self): + with mozfile.NamedTemporaryFile() as f: + with zipfile.ZipFile(f.name, 'w') as z: + self.create_apk_zipfiles(z) + z.writestr('package-name.txt', "org.mozilla.fennec") + v = get_version(f.name) + self.assertEqual(v.get('package_name'), "org.mozilla.fennec") + +if __name__ == '__main__': + unittest.main() diff --git a/testing/mozbase/mozversion/tests/test_b2g.py b/testing/mozbase/mozversion/tests/test_b2g.py new file mode 100644 index 000000000..09e0eb09e --- /dev/null +++ b/testing/mozbase/mozversion/tests/test_b2g.py @@ -0,0 +1,75 @@ +#!/usr/bin/env python + +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this file, +# You can obtain one at http://mozilla.org/MPL/2.0/. + +import os +import tempfile +import unittest +import zipfile + +import mozfile +from mozversion import get_version, errors + + +class SourcesTest(unittest.TestCase): + """test getting version information from a sources xml""" + + def setUp(self): + self.tempdir = tempfile.mkdtemp() + + self.binary = os.path.join(self.tempdir, 'binary') + with open(self.binary, 'w') as f: + f.write('foobar') + + with open(os.path.join(self.tempdir, 'application.ini'), 'w') as f: + f.writelines("""[App]\nName = B2G\n""") + + with open(os.path.join(self.tempdir, 'platform.ini'), 'w') as f: + f.write('[Build]\nBuildID = PlatformBuildID\n') + + def tearDown(self): + mozfile.remove(self.tempdir) + + def _create_zip(self, revision=None, date=None): + zip_path = os.path.join( + self.tempdir, 'gaia', 'profile', 'webapps', + 'settings.gaiamobile.org', 'application.zip') + os.makedirs(os.path.dirname(zip_path)) + app_zip = zipfile.ZipFile(zip_path, 'w') + if revision or date: + app_zip.writestr('resources/gaia_commit.txt', + '%s\n%s' % (revision, date)) + app_zip.close() + + def test_gaia_commit(self): + revision, date = ('a' * 40, 'date') + self._create_zip(revision, date) + v = get_version(self.binary) + self.assertEqual(v.get('gaia_changeset'), revision) + self.assertEqual(v.get('gaia_date'), date) + + def test_invalid_gaia_commit(self): + revision, date = ('a' * 41, 'date') + self._create_zip(revision, date) + v = get_version(self.binary) + self.assertIsNone(v.get('gaia_changeset')) + self.assertEqual(v.get('gaia_date'), date) + + def test_missing_zip_file(self): + v = get_version(self.binary) + self.assertIsNone(v.get('gaia_changeset')) + self.assertIsNone(v.get('gaia_date')) + + def test_missing_gaia_commit(self): + self._create_zip() + v = get_version(self.binary) + self.assertIsNone(v.get('gaia_changeset')) + self.assertIsNone(v.get('gaia_date')) + + def test_b2g_fallback_when_no_binary(self): + self.assertRaises(errors.RemoteAppNotFoundError, get_version) + +if __name__ == '__main__': + unittest.main() diff --git a/testing/mozbase/mozversion/tests/test_binary.py b/testing/mozbase/mozversion/tests/test_binary.py new file mode 100644 index 000000000..54665974f --- /dev/null +++ b/testing/mozbase/mozversion/tests/test_binary.py @@ -0,0 +1,177 @@ +#!/usr/bin/env python + +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this file, +# You can obtain one at http://mozilla.org/MPL/2.0/. + +import os +import sys +import tempfile +import shutil +import unittest + +import mozfile + +from mozversion import errors, get_version + + +class BinaryTest(unittest.TestCase): + """test getting application version information from a binary path""" + + application_ini = """[App] +ID = AppID +Name = AppName +CodeName = AppCodeName +Version = AppVersion +BuildID = AppBuildID +SourceRepository = AppSourceRepo +SourceStamp = AppSourceStamp +Vendor = AppVendor +""" + platform_ini = """[Build] +BuildID = PlatformBuildID +Milestone = PlatformMilestone +SourceStamp = PlatformSourceStamp +SourceRepository = PlatformSourceRepo +""" + + def setUp(self): + self.cwd = os.getcwd() + self.tempdir = tempfile.mkdtemp() + + self.binary = os.path.join(self.tempdir, 'binary') + with open(self.binary, 'w') as f: + f.write('foobar') + + def tearDown(self): + os.chdir(self.cwd) + mozfile.remove(self.tempdir) + + @unittest.skipIf(not os.environ.get('BROWSER_PATH'), + 'No binary has been specified.') + def test_real_binary(self): + v = get_version(os.environ.get('BROWSER_PATH')) + self.assertTrue(isinstance(v, dict)) + + def test_binary(self): + self._write_ini_files() + + self._check_version(get_version(self.binary)) + + @unittest.skipIf(not hasattr(os, 'symlink'), + 'os.symlink not supported on this platform') + def test_symlinked_binary(self): + self._write_ini_files() + + # create a symlink of the binary in another directory and check + # version against this symlink + tempdir = tempfile.mkdtemp() + try: + browser_link = os.path.join(tempdir, + os.path.basename(self.binary)) + os.symlink(self.binary, browser_link) + + self._check_version(get_version(browser_link)) + finally: + mozfile.remove(tempdir) + + def test_binary_in_current_path(self): + self._write_ini_files() + + os.chdir(self.tempdir) + self._check_version(get_version()) + + def test_with_ini_files_on_osx(self): + self._write_ini_files() + + platform = sys.platform + sys.platform = 'darwin' + try: + # get_version is working with ini files next to the binary + self._check_version(get_version(binary=self.binary)) + + # or if they are in the Resources dir + # in this case the binary must be in a Contents dir, next + # to the Resources dir + contents_dir = os.path.join(self.tempdir, 'Contents') + os.mkdir(contents_dir) + moved_binary = os.path.join(contents_dir, + os.path.basename(self.binary)) + shutil.move(self.binary, moved_binary) + + resources_dir = os.path.join(self.tempdir, 'Resources') + os.mkdir(resources_dir) + for ini_file in ('application.ini', 'platform.ini'): + shutil.move(os.path.join(self.tempdir, ini_file), resources_dir) + + self._check_version(get_version(binary=moved_binary)) + finally: + sys.platform = platform + + def test_invalid_binary_path(self): + self.assertRaises(IOError, get_version, + os.path.join(self.tempdir, 'invalid')) + + def test_without_ini_files(self): + """With missing ini files an exception should be thrown""" + self.assertRaises(errors.AppNotFoundError, get_version, + self.binary) + + def test_without_platform_ini_file(self): + """With a missing platform.ini file an exception should be thrown""" + self._write_ini_files(platform=False) + self.assertRaises(errors.AppNotFoundError, get_version, + self.binary) + + def test_without_application_ini_file(self): + """With a missing application.ini file an exception should be thrown""" + self._write_ini_files(application=False) + self.assertRaises(errors.AppNotFoundError, get_version, + self.binary) + + def test_with_exe(self): + """Test that we can resolve .exe files""" + self._write_ini_files() + + exe_name_unprefixed = self.binary + '1' + exe_name = exe_name_unprefixed + '.exe' + with open(exe_name, 'w') as f: + f.write('foobar') + self._check_version(get_version(exe_name_unprefixed)) + + def test_not_found_with_binary_specified(self): + self.assertRaises(errors.LocalAppNotFoundError, get_version, self.binary) + + def _write_ini_files(self, application=True, platform=True): + if application: + with open(os.path.join(self.tempdir, 'application.ini'), 'w') as f: + f.writelines(self.application_ini) + if platform: + with open(os.path.join(self.tempdir, 'platform.ini'), 'w') as f: + f.writelines(self.platform_ini) + + def _check_version(self, version): + self.assertEqual(version.get('application_id'), 'AppID') + self.assertEqual(version.get('application_name'), 'AppName') + self.assertEqual( + version.get('application_display_name'), 'AppCodeName') + self.assertEqual(version.get('application_version'), 'AppVersion') + self.assertEqual(version.get('application_buildid'), 'AppBuildID') + self.assertEqual( + version.get('application_repository'), 'AppSourceRepo') + self.assertEqual( + version.get('application_changeset'), 'AppSourceStamp') + self.assertEqual(version.get('application_vendor'), 'AppVendor') + self.assertIsNone(version.get('platform_name')) + self.assertEqual(version.get('platform_buildid'), 'PlatformBuildID') + self.assertEqual( + version.get('platform_repository'), 'PlatformSourceRepo') + self.assertEqual( + version.get('platform_changeset'), 'PlatformSourceStamp') + self.assertIsNone(version.get('invalid_key')) + self.assertEqual( + version.get('platform_version'), 'PlatformMilestone') + + +if __name__ == '__main__': + unittest.main() diff --git a/testing/mozbase/mozversion/tests/test_sources.py b/testing/mozbase/mozversion/tests/test_sources.py new file mode 100644 index 000000000..6c663edd6 --- /dev/null +++ b/testing/mozbase/mozversion/tests/test_sources.py @@ -0,0 +1,85 @@ +#!/usr/bin/env python + +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this file, +# You can obtain one at http://mozilla.org/MPL/2.0/. + +import os +import tempfile +import unittest + +import mozfile + +from mozversion import errors, get_version + + +class SourcesTest(unittest.TestCase): + """test getting version information from a sources xml""" + + application_ini = """[App]\nName = B2G\n""" + platform_ini = """[Build] +BuildID = PlatformBuildID +SourceStamp = PlatformSourceStamp +SourceRepository = PlatformSourceRepo +""" + sources_xml = """ + + + + +""" + + def setUp(self): + self.cwd = os.getcwd() + self.tempdir = tempfile.mkdtemp() + + self.binary = os.path.join(self.tempdir, 'binary') + with open(self.binary, 'w') as f: + f.write('foobar') + + def tearDown(self): + os.chdir(self.cwd) + mozfile.remove(self.tempdir) + + def _write_conf_files(self, sources=True): + with open(os.path.join(self.tempdir, 'application.ini'), 'w') as f: + f.writelines(self.application_ini) + with open(os.path.join(self.tempdir, 'platform.ini'), 'w') as f: + f.writelines(self.platform_ini) + if sources: + with open(os.path.join(self.tempdir, 'sources.xml'), 'w') as f: + f.writelines(self.sources_xml) + + def test_sources(self): + self._write_conf_files() + + os.chdir(self.tempdir) + self._check_version(get_version(sources=os.path.join(self.tempdir, + 'sources.xml'))) + + def test_sources_in_current_directory(self): + self._write_conf_files() + + os.chdir(self.tempdir) + self._check_version(get_version()) + + def test_invalid_sources_path(self): + """An invalid source path should cause an exception""" + self.assertRaises(errors.AppNotFoundError, get_version, + self.binary, os.path.join(self.tempdir, 'invalid')) + + def test_without_sources_file(self): + """With a missing sources file no exception should be thrown""" + self._write_conf_files(sources=False) + + get_version(self.binary) + + def _check_version(self, version): + self.assertEqual(version.get('build_changeset'), 'build_revision') + self.assertEqual(version.get('gaia_changeset'), 'gaia_revision') + self.assertEqual(version.get('gecko_changeset'), 'gecko_revision') + self.assertIsNone(version.get('invalid_key')) + + +if __name__ == '__main__': + unittest.main() -- cgit v1.2.3