from acts.base_test import BaseTestClass
from acts.controllers import native_android_device
from acts_contrib.test_utils.bt.native_bt_test_utils import setup_native_bluetooth
from acts_contrib.test_utils.bt.bt_test_utils import generate_id_by_size


class BtNativeTest(BaseTestClass):
    tests = None

    def __init__(self, controllers):
        BaseTestClass.__init__(self, controllers)
        self.tests = (
            "test_binder_get_name",
            "test_binder_get_name_invalid_parameter",
            "test_binder_set_name_get_name",
            "test_binder_get_address", )

    def setup_class(self):
        setup_native_bluetooth(self.native_android_devices)
        self.droid = self.native_android_devices[0].droid
        if len(self.native_android_devices) > 1:
            self.droid1 = self.native_android_devices[1].droid
            self.tests = self.tests + ("test_two_devices_set_get_name", )

    def test_binder_get_name(self):
        result = self.droid.BtBinderGetName()
        self.log.info("Bluetooth device name: {}".format(result))
        return True

    def test_binder_get_name_invalid_parameter(self):
        try:
            self.droid.BtBinderGetName("unexpected_parameter")
            return False
        except Exception:
            return True

    def test_binder_set_name_get_name(self):
        test_name = generate_id_by_size(4)
        result = self.droid.BtBinderSetName(test_name)
        if not result:
            return False
        name = self.droid.BtBinderGetName()
        if test_name != name:
            return False
        return True

    def test_binder_get_address(self):
        result = self.droid.BtBinderGetAddress()
        self.log.info("Found BT address: {}".format(result))
        if not result:
            return False
        return True

    def test_two_devices_set_get_name(self):
        test_name = generate_id_by_size(4)
        for n in self.native_android_devices:
            d = n.droid
            d.BtBinderSetName(test_name)
            name = d.BtBinderGetName()
            if name != test_name:
                return False
        return True