michael@0: #!/usr/bin/env python
michael@0: 
michael@0: # This Source Code Form is subject to the terms of the Mozilla Public
michael@0: # License, v. 2.0. If a copy of the MPL was not distributed with this file,
michael@0: # You can obtain one at http://mozilla.org/MPL/2.0/.
michael@0: 
michael@0: import os
michael@0: import tempfile
michael@0: import unittest
michael@0: 
michael@0: import mozfile
michael@0: from mozversion import get_version
michael@0: 
michael@0: 
michael@0: class BinaryTest(unittest.TestCase):
michael@0:     """test getting application version information from a binary path"""
michael@0: 
michael@0:     application_ini = """[App]
michael@0: Name = AppName
michael@0: CodeName = AppCodeName
michael@0: Version = AppVersion
michael@0: BuildID = AppBuildID
michael@0: SourceRepository = AppSourceRepo
michael@0: SourceStamp = AppSourceStamp
michael@0: """
michael@0:     platform_ini = """[Build]
michael@0: BuildID = PlatformBuildID
michael@0: SourceStamp = PlatformSourceStamp
michael@0: SourceRepository = PlatformSourceRepo
michael@0: """
michael@0: 
michael@0:     def setUp(self):
michael@0:         self.cwd = os.getcwd()
michael@0:         self.tempdir = tempfile.mkdtemp()
michael@0: 
michael@0:         self.binary = os.path.join(self.tempdir, 'binary')
michael@0:         with open(self.binary, 'w') as f:
michael@0:             f.write('foobar')
michael@0: 
michael@0:     def tearDown(self):
michael@0:         os.chdir(self.cwd)
michael@0:         mozfile.remove(self.tempdir)
michael@0: 
michael@0:     def test_binary(self):
michael@0:         with open(os.path.join(self.tempdir, 'application.ini'), 'w') as f:
michael@0:             f.writelines(self.application_ini)
michael@0: 
michael@0:         with open(os.path.join(self.tempdir, 'platform.ini'), 'w') as f:
michael@0:             f.writelines(self.platform_ini)
michael@0: 
michael@0:         self._check_version(get_version(self.binary))
michael@0: 
michael@0:     def test_binary_in_current_path(self):
michael@0:         with open(os.path.join(self.tempdir, 'application.ini'), 'w') as f:
michael@0:             f.writelines(self.application_ini)
michael@0: 
michael@0:         with open(os.path.join(self.tempdir, 'platform.ini'), 'w') as f:
michael@0:             f.writelines(self.platform_ini)
michael@0:         os.chdir(self.tempdir)
michael@0:         self._check_version(get_version())
michael@0: 
michael@0:     def test_invalid_binary_path(self):
michael@0:         self.assertRaises(IOError, get_version,
michael@0:                           os.path.join(self.tempdir, 'invalid'))
michael@0: 
michael@0:     def test_missing_ini_files(self):
michael@0:         v = get_version(self.binary)
michael@0:         self.assertEqual(v, {})
michael@0: 
michael@0:     def _check_version(self, version):
michael@0:         self.assertEqual(version.get('application_name'), 'AppName')
michael@0:         self.assertEqual(version.get('application_display_name'), 'AppCodeName')
michael@0:         self.assertEqual(version.get('application_version'), 'AppVersion')
michael@0:         self.assertEqual(version.get('application_buildid'), 'AppBuildID')
michael@0:         self.assertEqual(
michael@0:             version.get('application_repository'), 'AppSourceRepo')
michael@0:         self.assertEqual(
michael@0:             version.get('application_changeset'), 'AppSourceStamp')
michael@0:         self.assertIsNone(version.get('platform_name'))
michael@0:         self.assertIsNone(version.get('platform_version'))
michael@0:         self.assertEqual(version.get('platform_buildid'), 'PlatformBuildID')
michael@0:         self.assertEqual(
michael@0:             version.get('platform_repository'), 'PlatformSourceRepo')
michael@0:         self.assertEqual(
michael@0:             version.get('platform_changeset'), 'PlatformSourceStamp')
michael@0:         self.assertIsNone(version.get('invalid_key'))
michael@0: 
michael@0: 
michael@0: if __name__ == '__main__':
michael@0:     unittest.main()