Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 1ffc4a4d authored by Charlie Boutier's avatar Charlie Boutier
Browse files

Pandora: Refactor Bumble service registration 

Bug: 342195977
Flag: TEST_ONLY
Test: atest BumbleBluetoothTests
Change-Id: Iaa2ca6e90f0290ac95e9f51a88e2c4f725c64ca4
parent 21a68fcc

framework/tests/bumble/src/bumble_server.py

+10 −7
Original line number Diff line number Diff line
@@ -47,13 +47,7 @@ ROOTCANAL_PORT_CUTTLEFISH = 7300 
    help='Bumble json configuration file',

)

def main(grpc_port: int, rootcanal_port: int, transport: str, config: str) -> None:

    bumble_server.register_servicer_hook(

        lambda bumble, _, server: add_AshaServicer_to_server(AshaService(bumble.device), server))

    bumble_server.register_servicer_hook(

        lambda bumble, _, server: add_DckServicer_to_server(DckService(bumble.device), server))

    bumble_server.register_servicer_hook(

        lambda bumble, _, server: add_GATTServicer_to_server(GATTService(bumble.device), server))



    register_bumble_services()

    if '<rootcanal-port>' in transport:

        transport = transport.replace('<rootcanal-port>', str(rootcanal_port))
@@ -70,6 +64,15 @@ def main(grpc_port: int, rootcanal_port: int, transport: str, config: str) -> No 
    asyncio.run(serve(device, config=server_config, port=grpc_port))





def register_bumble_services():

    bumble_server.register_servicer_hook(

        lambda bumble, _, server: add_AshaServicer_to_server(AshaService(bumble.device), server))

    bumble_server.register_servicer_hook(

        lambda bumble, _, server: add_DckServicer_to_server(DckService(bumble.device), server))

    bumble_server.register_servicer_hook(

        lambda bumble, _, server: add_GATTServicer_to_server(GATTService(bumble.device), server))





def retrieve_config(config: str) -> Dict[str, Any]:

    if not config:

        return {}