"""
Bluetooth Base Test

This test class serves as a base class for others to inherit from.
It also serves as a device-cleaner to help reset devices between tests.

"""

import logging

from mobly import base_test
from mobly.controllers import android_device
from mobly.controllers.android_device_lib.snippet_client_v2 import Config

from utilities import spectatio_utils
from utilities import bt_utils
from utilities.main_utils import common_main, get_test_args
from utilities.video_utils_service import VideoRecording


class BluetoothBaseTest(base_test.BaseTestClass):

    def setup_class(self):
        # Registering android_device controller module, and declaring that the test
        # requires at least two Android devices.
        # This setup will need to be overwritten or extended if a test uses three devices.
        logging.info("Running basic class setup.")
        self.ads = self.register_controller(android_device, min_number=2)
        # The device used to discover Bluetooth devices.
        self.discoverer = android_device.get_device(
            self.ads, label='auto')
        # Sets the tag that represents this device in logs.
        self.discoverer.debug_tag = 'discoverer'
        # The device that is expected to be discovered
        self.target = android_device.get_device(self.ads, label='phone')
        self.target.debug_tag = 'target'

        test_args = get_test_args(shell_escape=True)
        logging.info("\tLoading Snippets.  Test args = {}".format(test_args))
        snippet_config = Config(am_instrument_options=test_args)
        logging.info("\tSnippet config: {}".format(snippet_config))
        self.target.load_snippet('mbs', android_device.MBS_PACKAGE, config=snippet_config)
        self.discoverer.load_snippet('mbs', android_device.MBS_PACKAGE, config=snippet_config)

        logging.info("\tInitializing Utilities")
        self.call_utils = (spectatio_utils.CallUtils(self.discoverer))
        self.bt_utils = (bt_utils.BTUtils(self.discoverer, self.target))

        logging.info("\tInitializing video services on HU")
        self.video_utils_service = VideoRecording(self.discoverer, self.__class__.__name__)
        logging.info("\tInitializing video services on target")
        self.video_utils_service_target = VideoRecording(self.target, self.__class__.__name__)

    def setup_test(self):
        # Make sure bluetooth is on.
        self.call_utils.press_home()
        logging.info("Running basic test setup.")
        logging.info("\tEnabling bluetooth on Target and Discoverer.")
        self.target.mbs.btEnable()
        self.discoverer.mbs.btEnable()

    def teardown_test(self):
        # Turn Bluetooth off on both devices.
        logging.info("Running basic test teardown.")
        self.call_utils.press_home()
        self.call_utils.press_phone_home_icon_using_adb_command(self.target)
        self.hu_recording_handler()
        self.bt_utils.unpair()
        logging.info("Disable Bluetooth on Discoverer device")
        self.discoverer.mbs.btDisable()
        logging.info("Disable Bluetooth on Target device")
        self.target.mbs.btDisable()

    def teardown_no_video_recording(self):
        # Turn Bluetooth off on both devices.
        logging.info("Running basic test teardown.")
        self.call_utils.press_home()
        self.call_utils.press_phone_home_icon_using_adb_command(self.target)
        self.bt_utils.unpair()
        logging.info("Disable Bluetooth on Discoverer device")
        self.discoverer.mbs.btDisable()
        logging.info("Disable Bluetooth on Target device")
        self.target.mbs.btDisable()


    def hu_recording_handler(self):
        logging.info("Stopping the screen recording on Discoverer Device")
        self.video_utils_service.stop_screen_recording()
        logging.info("Pull the screen recording from Discoverer device")
        self.video_utils_service.pull_recording_file(self.log_path)
        logging.info("delete the screen recording from the Discoverer device")
        self.video_utils_service.delete_screen_recording_from_device()

        logging.info("Stopping the screen recording on Target Device")
        self.video_utils_service_target.stop_screen_recording()
        logging.info("Pull the screen recording from Target device")
        self.video_utils_service_target.pull_recording_file(self.log_path)
        logging.info("delete the screen recording from the Target device")
        self.video_utils_service_target.delete_screen_recording_from_device()

    def enable_recording(self):
        logging.info("Enabling video recording for Discoverer device")
        self.video_utils_service.enable_screen_recording()
        logging.info("Enabling video recording for Target device")
        self.video_utils_service_target.enable_screen_recording()
        logging.info("Video recording started on Discoverer and Target")

if __name__ == '__main__':
    common_main()