Merge "aics: add test discover as secondary service" into main

from avatar import PandoraDevices, BumblePandoraDevice

from mobly import base_test, signals

from mobly.asserts import assert_in  # type: ignore

from mobly.asserts import assert_in, assert_not_in  # type: ignore

from pandora.host_pb2 import RANDOM

from pandora_experimental.gatt_grpc import GATT

from pandora_experimental.gatt_pb2 import PRIMARY

from bumble.att import UUID

from bumble.gatt import GATT_AUDIO_INPUT_CONTROL_SERVICE

from bumble.gatt import GATT_VOLUME_CONTROL_SERVICE, GATT_AUDIO_INPUT_CONTROL_SERVICE

from bumble.profiles.aics import AICSService

from bumble.profiles.vcp import VolumeControlService

class AicsTest(base_test.BaseTestClass):

    async def setup_test(self) -> None:

        await asyncio.gather(self.dut.reset(), self.ref.reset())

        self.ref.device.add_service(AICSService())  # type: ignore

        aics_service = AICSService()

        volume_control_service = VolumeControlService(included_services=[aics_service])

        self.ref.device.add_service(aics_service)  # type: ignore

        self.ref.device.add_service(volume_control_service)  # type: ignore

    def test_gatt_discover_aics_service(self) -> None:

    def connect_dut_to_ref(self):

        advertise = self.ref.host.Advertise(legacy=True, connectable=True)

        dut_ref_connection = self.dut.host.ConnectLE(public=self.ref.address, own_address_type=RANDOM).connection

        assert dut_ref_connection

        advertise.cancel()  # type: ignore

        dut_gatt = GATT(self.dut.channel)  # type: ignore

        return dut_ref_connection

    def test_do_not_discover_aics_as_primary_service(self) -> None:

        dut_ref_connection = self.connect_dut_to_ref()

        dut_gatt = GATT(self.dut.channel)

        services = dut_gatt.DiscoverServices(dut_ref_connection).services

        uuids = [UUID(service.uuid) for service in services]

        uuids = [UUID(service.uuid) for service in services if service.service_type == PRIMARY]

        assert_in(GATT_VOLUME_CONTROL_SERVICE, uuids)

        assert_not_in(GATT_AUDIO_INPUT_CONTROL_SERVICE, uuids)

    def test_gatt_discover_aics_service(self) -> None:

        dut_ref_connection = self.connect_dut_to_ref()

        dut_gatt = GATT(self.dut.channel)

        services = dut_gatt.DiscoverServices(dut_ref_connection).services

        filtered_services = [service for service in services if UUID(service.uuid) == GATT_VOLUME_CONTROL_SERVICE]

        assert len(filtered_services) == 1

        vcp_service = filtered_services[0]

        assert_in(GATT_AUDIO_INPUT_CONTROL_SERVICE, uuids)

        included_services_uuids = [UUID(included_service.uuid) for included_service in vcp_service.included_services]

        assert_in(GATT_AUDIO_INPUT_CONTROL_SERVICE, included_services_uuids)