| #!/usr/bin/env python3 |
| # Copyright 2022 The Fuchsia Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import json |
| import os |
| import shutil |
| import tempfile |
| import unittest |
| import update_platform_version |
| |
| FAKE_VERSION_HISTORY_FILE_CONTENT = """{ |
| \"data\": { |
| "name": "Platform version map", |
| "type": "version_history", |
| "versions": [ |
| { |
| "api_level": "1", |
| "abi_revision": "0x201665C5B012BA43" |
| } |
| ] |
| }, |
| "schema_id": "https://fuchsia.dev/schema/version_history-ef02ef45.json" |
| } |
| """ |
| |
| NEW_API_LEVEL = 2 |
| |
| |
| class TestUpdatePlatformVersionMethods(unittest.TestCase): |
| |
| def setUp(self): |
| self.test_dir = tempfile.mkdtemp() |
| self.fake_version_history_file = os.path.join( |
| self.test_dir, 'version_history.json') |
| with open(self.fake_version_history_file, 'w') as f: |
| f.write(FAKE_VERSION_HISTORY_FILE_CONTENT) |
| update_platform_version.VERSION_HISTORY_PATH = self.fake_version_history_file |
| self.fake_milestone_version_file = os.path.join( |
| self.test_dir, 'platform_version.json') |
| with open(self.fake_milestone_version_file, 'w') as f: |
| f.write('1') |
| update_platform_version.PLATFORM_VERSION_PATH = self.fake_milestone_version_file |
| |
| def tearDown(self): |
| shutil.rmtree(self.test_dir) |
| |
| def _version_history_contains_entry_for_api_level(self, api_level): |
| with open(self.fake_version_history_file, "r") as f: |
| version_history = json.load(f) |
| versions = version_history['data']['versions'] |
| return any( |
| version['api_level'] == str(api_level) for version in versions) |
| |
| def test_update_version_history(self): |
| self.assertFalse( |
| self._version_history_contains_entry_for_api_level(NEW_API_LEVEL)) |
| self.assertTrue( |
| update_platform_version.update_version_history(NEW_API_LEVEL)) |
| self.assertTrue( |
| self._version_history_contains_entry_for_api_level(NEW_API_LEVEL)) |
| self.assertFalse( |
| update_platform_version.update_version_history(NEW_API_LEVEL)) |
| |
| def _get_current_platform_version(self): |
| with open(self.fake_milestone_version_file) as f: |
| version = int(f.read()) |
| return version |
| |
| def test_update_platform_version(self): |
| self.assertNotEqual(NEW_API_LEVEL, self._get_current_platform_version()) |
| self.assertTrue( |
| update_platform_version.update_platform_version(NEW_API_LEVEL)) |
| self.assertEqual(NEW_API_LEVEL, self._get_current_platform_version()) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
