Added svc data set function

Added function in gap.py to set service data.
Tests added to verify functionality.

Author: koffes

autopts/pybtp/btp/gap.py

@@ -1700,3 +1700,35 @@ def gap_set_broadcast_code(broadcast_code):
     logging.debug("%s %r", gap_set_broadcast_code.__name__, broadcast_code)
     stack = get_stack()
     stack.gap.big_broadcast_code = broadcast_code
+
+
+def gap_set_uuid16_svc_data(advData, uuid, service_data=None):
+    """Update service data. Will overwrite existing UUID16 service data if present"""
+    # Convert uuid to integer if it's a string
+    if not isinstance(uuid, int):
+        try:
+            uuid = int(uuid, 16)
+        except ValueError as err:
+            raise ValueError("Invalid UUID string format") from err
+
+    if (uuid < 0) or (uuid > 0xFFFF):
+        raise ValueError("Invalid UUID16 value")
+
+    if service_data is not None:
+        # Check if service_data is bytes (result of struct.pack)
+        if not isinstance(service_data, bytes):
+            raise TypeError(f"service_data must be bytes, got {type(service_data).__name__}")
+
+    # Initialize uuid16_svc_data if it doesn't exist
+    if AdType.uuid16_svc_data not in advData:
+        advData[AdType.uuid16_svc_data] = []
+
+    for entry in advData[AdType.uuid16_svc_data]:
+        existing_uuid = struct.unpack_from('<H', entry[:2])[0]
+        if existing_uuid == uuid:
+            advData[AdType.uuid16_svc_data].remove(entry)
+            break
+    if service_data is None:
+        advData[AdType.uuid16_svc_data] += [struct.pack('<H', uuid)]
+    else:
+        advData[AdType.uuid16_svc_data] += [struct.pack('<H', uuid) + service_data]

autopts/pybtp/types.py

@@ -117,6 +117,7 @@ class UUID:
     BASS = '184F'
     PACS = '1850'
     CAS = '1853'
+    TMAP = '1855'
     ASE_CP = '2BC6'
     CEP = '2900'
     CUD = '2901'
@@ -502,3 +503,22 @@ class BIGEncryption:
     BROADCAST_CODE_REQUIRED = 0x01
     DECRYPTING = 0x02
     BAD_CODE = 0x03
+
+
+class TMAPRole(IntFlag):
+    CALL_GATEWAY = defs.BIT(0)
+    CALL_TERMINAL = defs.BIT(1)
+    UNICAST_MEDIA_SENDER = defs.BIT(2)
+    UNICAST_MEDIA_RECEIVER = defs.BIT(3)
+    BROADCAST_MEDIA_SENDER = defs.BIT(4)
+    BROADCAST_MEDIA_RECEIVER = defs.BIT(5)
+
+
+class BAPAnnouncement:
+    GENERAL = 0x0
+    TARGETED = 0x1
+
+
+class CAPAnnouncement:
+    GENERAL = 0x0
+    TARGETED = 0x1

test/unittests.py

@@ -1,15 +1,20 @@
 import os
 import shutil
+import struct
 import sys
 import unittest
 from os.path import abspath, dirname
 from pathlib import Path
 from unittest.mock import patch
 
+import pytest
+
 from autopts.bot.common_features import report
 from autopts.client import FakeProxy, TestCaseRunStats
 from autopts.config import FILE_PATHS
 from autopts.ptsprojects.testcase_db import TestCaseTable
+from autopts.pybtp.btp.gap import gap_set_uuid16_svc_data
+from autopts.pybtp.types import AdType
 from autoptsclient_bot import import_bot_module, import_bot_projects
 from test.mocks.mocked_test_cases import mock_workspace_test_cases, test_case_list_generation_samples
 
@@ -250,6 +255,36 @@ def test_generate_stats(self):
                                 results, regressions, progresses, new_cases)
         assert os.path.exists(FILE_PATHS['REPORT_DIFF_TXT_FILE'])
 
+    def test_gap_set_uuid16_svc_data(self):
+        advData = {}
+        # Test invalid inputs
+        with pytest.raises(ValueError, match="Invalid UUID16 value"):
+            gap_set_uuid16_svc_data(advData, 0xFFFF1, None)
+        with pytest.raises(ValueError, match="Invalid UUID16 value"):
+            gap_set_uuid16_svc_data(advData, -1, None)
+        with pytest.raises(TypeError):
+            gap_set_uuid16_svc_data(advData, 0xAAAA, 4)
+
+        # Test valid input
+        try:
+            gap_set_uuid16_svc_data(advData, 0x0000, None)
+            gap_set_uuid16_svc_data(advData, 0xFFFF, None)
+            gap_set_uuid16_svc_data(advData, 0xAABB, struct.pack('<B', 1))
+
+        except Exception as e:
+            self.fail(f"Function raised unexpected exception: {e}")
+
+        assert advData[AdType.uuid16_svc_data][0] == struct.pack('<H', 0x0000)
+        assert advData[AdType.uuid16_svc_data][1] == struct.pack('<H', 0xFFFF)
+        assert advData[AdType.uuid16_svc_data][2] == struct.pack('<HB', 0xAABB, 1)
+
+        # Update a value in advData
+        gap_set_uuid16_svc_data(advData, 0xFFFF, struct.pack('<H', 0xABCD))
+
+        assert advData[AdType.uuid16_svc_data][0] == struct.pack('<H', 0x0000)
+        assert advData[AdType.uuid16_svc_data][1] == struct.pack('<HB', 0xAABB, 1)
+        assert advData[AdType.uuid16_svc_data][2] == struct.pack('<HH', 0xFFFF, 0xABCD)
+
 
 if __name__ == '__main__':
     unittest.main()