#  Copyright (C) 2024 The Android Open Source Project
#
#  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.
"""Ranging base test."""

import logging
import re

from mobly import base_test
from mobly import records
from mobly import test_runner
from mobly.controllers import android_device

from test_utils import uwb_test_utils

RELEASE_ID_REGEX = re.compile(r"\w+\.\d+\.\d+")


class RangingBaseTest(base_test.BaseTestClass):
    """Base class for Uwb tests."""

    def setup_class(self):
        """Sets up the Android devices for Uwb test."""
        super().setup_class()
        self.android_devices = self.register_controller(android_device,
                                                        min_number=2)
        for ad in self.android_devices:
            ad.load_snippet("ranging", "multidevices.snippet.ranging")

        # for ad in self.android_devices:
        #     uwb_test_utils.initialize_uwb_country_code_if_not_set(ad)

    def setup_test(self):
        super().setup_test()
        for ad in self.android_devices:
            dev1 = ad.ranging
            dev1.logInfo("*** TEST START: " + self.current_test_info.name + " ***")

    def teardown_test(self):
        super().teardown_test()
        # for ad in self.android_devices:
        #     ad.ranging.logInfo("*** TEST END: " + self.current_test_info.name + " ***")

    def teardown_class(self):
        super().teardown_class()


if __name__ == "__main__":
    test_runner.main()