#!/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"
}
"""

OLD_API_LEVEL = 1
OLD_SUPPORTED_API_LEVELS = [1]

NEW_API_LEVEL = 2
NEW_SUPPORTED_API_LEVELS = [1, 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:
            pv = {
                'current_fuchsia_api_level': OLD_API_LEVEL,
                'supported_fuchsia_api_levels': OLD_SUPPORTED_API_LEVELS,
            }
            json.dump(pv, f)
        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_platform_version(self):
        with open(self.fake_milestone_version_file) as f:
            return json.load(f)

    def test_update_platform_version(self):
        pv = self._get_platform_version()
        self.assertNotEqual(NEW_API_LEVEL, pv['current_fuchsia_api_level'])
        self.assertNotEqual(
            NEW_SUPPORTED_API_LEVELS, pv['supported_fuchsia_api_levels'])

        self.assertTrue(
            update_platform_version.update_platform_version(NEW_API_LEVEL))

        pv = self._get_platform_version()
        self.assertEqual(NEW_API_LEVEL, pv['current_fuchsia_api_level'])
        self.assertEqual(
            NEW_SUPPORTED_API_LEVELS, pv['supported_fuchsia_api_levels'])


if __name__ == '__main__':
    unittest.main()