# Copyright 2024 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os
import sys
import tempfile
import unittest

# Allow us to run even if not at the `tools` directory.
root_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
sys.path.insert(0, root_dir)

from app_header import PebbleAppHeader
from uuid import UUID

V1_APP_HEADER = "\x50\x42\x4C\x41\x50\x50\x00\x00\x08\x01\x03\x01\x03\x00" \
                "\xD8\x1A\x34\x0A\x00\x00\xC6\xF1\x2E\x8B\x57\x46\x47\x20" \
                "\x44\x65\x6D\x6F\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00" \
                "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00" \
                "\x57\x61\x74\x63\x68\x66\x61\x63\x65\x20\x47\x65\x6E\x65" \
                "\x72\x61\x74\x6F\x72\x00\x00\x00\x00\x00\x00\x00\x00\x00" \
                "\x00\x00\x00\x00\x01\x00\x00\x00\x00\x13\x00\x00\x01\x00" \
                "\x00\x00\xD8\x1A\x00\x00\x22\x00\x00\x00\xC9\x5A\x9A\x75" \
                "\x6E\x8C\x01\x59\xD3\xE0\x2F\x94\x1F\xA6\xB9\x75"

V2_APP_HEADER = "\x50\x42\x4C\x41\x50\x50\x00\x00\x10\x00\x05\x00\x01\x00" \
                "\xA1\x0C\x08\x05\x00\x00\x06\x3E\x92\x94\x57\x46\x47\x20" \
                "\x44\x65\x6D\x6F\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00" \
                "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00" \
                "\x57\x61\x74\x63\x68\x66\x61\x63\x65\x2D\x47\x65\x6E\x65" \
                "\x72\x61\x74\x6F\x72\x2E\x64\x65\x00\x00\x00\x00\x00\x00" \
                "\x00\x00\x00\x00\x01\x00\x00\x00\x94\x00\x00\x00\x01\x00" \
                "\x00\x00\x04\x00\x00\x00\x13\x37\x13\x37\xD7\xAA\x1F\xEB" \
                "\xB8\x78\x99\x91\x62\x89\xCD\x1E\x07\x60\x88\xCF\xCE\x4A" \
                "\xD0\x52\xC8\x0D"


class TestAppHeader(unittest.TestCase):
    def test_deserialize_v1_header(self):
        h = PebbleAppHeader(V1_APP_HEADER)
        self.assertEquals(h.sentinel, "PBLAPP\x00\x00")
        self.assertEquals(h.struct_version_major,
                          PebbleAppHeader.V1_STRUCT_VERSION[0])
        self.assertEquals(h.struct_version_minor,
                          PebbleAppHeader.V1_STRUCT_VERSION[1])
        self.assertEquals(h.sdk_version_major, 0x03)
        self.assertEquals(h.sdk_version_minor, 0x01)
        self.assertEquals(h.app_version_major, 0x03)
        self.assertEquals(h.app_version_minor, 0x00)
        self.assertEquals(h.app_size, 6872)
        self.assertEquals(h.offset, 2612)
        self.assertEquals(h.crc, 0x8b2ef1c6)
        self.assertEquals(h.app_name, "WFG Demo")
        self.assertEquals(h.company_name, "Watchface Generator")
        self.assertEquals(h.icon_resource_id, 1)
        self.assertEquals(h.symbol_table_addr, 4864)
        self.assertEquals(h.flags, 1)
        self.assertEquals(h.relocation_list_index, 6872)
        self.assertEquals(h.num_relocation_entries, 34)
        self.assertEquals(h.uuid, UUID('c95a9a75-6e8c-0159-d3e0-2f941fa6b975'))

    def test_deserialize_v2_header(self):
        h = PebbleAppHeader(V2_APP_HEADER)
        self.assertEquals(h.sentinel, "PBLAPP\x00\x00")
        self.assertEquals(h.struct_version_major,
                          PebbleAppHeader.V2_STRUCT_VERSION[0])
        self.assertEquals(h.struct_version_minor,
                          PebbleAppHeader.V2_STRUCT_VERSION[1])
        self.assertEquals(h.sdk_version_major, 0x05)
        self.assertEquals(h.sdk_version_minor, 0x00)
        self.assertEquals(h.app_version_major, 0x01)
        self.assertEquals(h.app_version_minor, 0x00)
        self.assertEquals(h.app_size, 3233)
        self.assertEquals(h.offset, 1288)
        self.assertEquals(h.crc, 0x94923e06)
        self.assertEquals(h.app_name, "WFG Demo")
        self.assertEquals(h.company_name, "Watchface-Generator.de")
        self.assertEquals(h.icon_resource_id, 1)
        self.assertEquals(h.symbol_table_addr, 148)
        self.assertEquals(h.flags, 1)
        self.assertEquals(h.num_relocation_entries, 4)
        self.assertEquals(h.uuid, UUID('13371337-d7aa-1feb-b878-99916289cd1e'))
        self.assertEquals(h.resource_crc, 0xcf886007)
        self.assertEquals(h.resource_timestamp, 1389382350)
        self.assertEquals(h.virtual_size, 3528)

    def test_deserialize_serialize_v1(self):
        h = PebbleAppHeader(V1_APP_HEADER)
        bytes = h.serialize()
        self.assertEquals(bytes, V1_APP_HEADER)

    def test_deserialize_serialize_v2(self):
        h = PebbleAppHeader(V2_APP_HEADER)
        bytes = h.serialize()
        self.assertEquals(bytes, V2_APP_HEADER)


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