OpenGoPro/demos/python/sdk_wireless_camera_control/tests/unit/test_services.py

# test_services.py/Open GoPro, Version 2.0 (C) Copyright 2021 GoPro, Inc. (http://gopro.com/OpenGoPro).
# This copyright was auto-generated on Wed, Sep  1, 2021  5:05:54 PM

# pylint: disable = redefined-outer-name

import uuid

import pytest

from open_gopro.network.ble import BleUUID, Characteristic, Descriptor, GattDB, Service
from open_gopro.network.ble.services import BLE_BASE_UUID, UUIDs, UUIDsMeta


def test_128_bit_uuid():
    u = BleUUID("128 bit from str", hex="12345678123456781234567812345678")
    assert len(u.hex)
    u = BleUUID("128 bit from int", int=0x12345678123456781234567812345678)
    assert len(u.hex)
    u = BleUUID("128 bit from bytes", bytes=b"\x12\x34\x56\x78" * 4)
    assert len(u.hex)
    u = BleUUID(
        "128 bit from byte le",
        bytes_le=b"\x78\x56\x34\x12\x34\x12\x78\x56" + b"\x12\x34\x56\x78\x12\x34\x56\x78",
    )
    assert len(u.hex)


def test_16_bit_uuid():
    u = BleUUID("16 bit from str", format=BleUUID.Format.BIT_16, hex="FEA6")
    assert len(u.hex)
    u = BleUUID("16 bit from int", format=BleUUID.Format.BIT_16, int=1234)
    assert len(u.hex)
    u = BleUUID("16 bit from bytes", format=BleUUID.Format.BIT_16, bytes=bytes([0xAB, 0xCD]))
    assert len(u.hex)


def test_uuid_negative():
    with pytest.raises(ValueError):
        u = BleUUID("16 bit from bytes le", format=BleUUID.Format.BIT_16, bytes_le=bytes([0xCD, 0xAB]))
    with pytest.raises(ValueError):
        u = BleUUID("Multiple inputs", format=BleUUID.Format.BIT_16, hex="", int=1)
    with pytest.raises(ValueError):
        u = BleUUID("Bad string", format=BleUUID.Format.BIT_16, hex="AB")
    with pytest.raises(ValueError):
        u = BleUUID("Bad bytes", format=BleUUID.Format.BIT_16, bytes=bytes([0xAB, 0xCD, 0xEF]))


def test_ble_uuids():
    UUID_STR = "00001801-0000-1000-8000-00805f9b34fb"
    UUID_INT = 486857058725721441610830112830715

    class TestUUIDs(metaclass=UUIDsMeta):
        TEST_UUID = BleUUID("Test", hex=UUID_STR)

    assert UUID_STR in TestUUIDs
    assert BLE_BASE_UUID.format("ABCD") not in TestUUIDs
    assert UUID_INT in TestUUIDs
    assert 0xABCD not in TestUUIDs
    assert TestUUIDs.TEST_UUID in TestUUIDs
    assert uuid.UUID(hex=UUID_STR) in TestUUIDs
    assert TestUUIDs[UUID_STR].hex
    assert TestUUIDs[UUID_INT].hex
    assert TestUUIDs.TEST_UUID.hex
    assert TestUUIDs[uuid.UUID(hex=UUID_STR)].hex
    assert len([x for x in TestUUIDs]) == 1


def test_ble_uuids_negative():
    with pytest.raises(TypeError):

        class TestUUIDs(UUIDs):
            BAD_ATTRIBUTE = 1


def test_descriptor(mock_descriptor: Descriptor):
    assert mock_descriptor.handle > 0
    assert mock_descriptor.name == mock_descriptor.uuid.name


def test_characteristic(mock_characteristic: Characteristic):
    assert mock_characteristic.handle > 0
    assert mock_characteristic.name == mock_characteristic.uuid.name
    assert len(mock_characteristic.descriptors)

    assert mock_characteristic.is_readable
    assert not mock_characteristic.is_writeable
    assert not mock_characteristic.is_notifiable
    assert not mock_characteristic.is_indicatable


def test_service(mock_service: Service):
    assert mock_service.start_handle > 0
    assert mock_service.name == mock_service.uuid.name
    assert len(mock_service.characteristics) > 0


def test_characteristic_view(mock_gatt_db: GattDB):
    # Get all attributes by nested looping through services
    chars: list[Characteristic] = []
    for service in mock_gatt_db.services.values():
        for char in service.characteristics.values():
            chars.append(char)

    assert len(chars) == len(mock_gatt_db.characteristics)

    for char in chars:
        assert char.uuid in mock_gatt_db.characteristics

    for char in mock_gatt_db.characteristics:
        assert len(char.uuid.hex)

    assert list(mock_gatt_db.characteristics.keys()) == [c.uuid for c in chars]
    assert list([c.uuid for c in mock_gatt_db.characteristics.values()]) == [c.uuid for c in chars]


def test_gatt_db(mock_gatt_db: GattDB):
    handles = set([c.handle for c in mock_gatt_db.characteristics])
    uuids = set([c.uuid for c in mock_gatt_db.characteristics])
    assert handles == set([mock_gatt_db.uuid2handle(uuid) for uuid in uuids])
    assert uuids == set([mock_gatt_db.handle2uuid(handle) for handle in handles])

    mock_gatt_db.dump_to_csv()