mirror of
https://github.com/classilla/tenfourfox.git
synced 2025-01-06 09:29:35 +00:00
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
#!/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 = """<?xml version="1.0" ?><manifest>
|
|
<project path="build" revision="build_revision" />
|
|
<project path="gaia" revision="gaia_revision" />
|
|
<project path="gecko" revision="gecko_revision" />
|
|
</manifest>
|
|
"""
|
|
|
|
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()
|