Refactor tests directory structure to mirror bumble package layout

tests/audio/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

tests/crypto/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

tests/crypto/crypto_test.py

@@ -0,0 +1,213 @@
+# Copyright 2021-2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# -----------------------------------------------------------------------------
+# Imports
+# -----------------------------------------------------------------------------
+from unittest import mock
+
+import pytest
+
+from bumble import crypto
+from bumble.crypto import EccKey, aes_cmac, ah, c1, f4, f5, f6, g2, h6, h7, s1
+
+
+# -----------------------------------------------------------------------------
+# pylint: disable=invalid-name
+# -----------------------------------------------------------------------------
+@pytest.fixture(
+    scope="session", params=["bumble.crypto.builtin", "bumble.crypto.cryptography"]
+)
+def crypto_backend(request):
+    backend = pytest.importorskip(request.param)
+    with (
+        mock.patch.object(crypto, "e", backend.e),
+        mock.patch.object(crypto, "aes_cmac", backend.aes_cmac),
+        mock.patch.object(crypto, "EccKey", backend.EccKey),
+    ):
+        yield
+
+
+# -----------------------------------------------------------------------------
+def reversed_hex(hex_str: str) -> bytes:
+    return bytes.fromhex(hex_str)[::-1]
+
+
+# -----------------------------------------------------------------------------
+def test_ecc(crypto_backend):
+    key = EccKey.generate()
+    x = key.x
+    y = key.y
+
+    assert len(x) == 32
+    assert len(y) == 32
+
+    # Test DH with test vectors from the spec
+    private_A = (
+        '3f49f6d4 a3c55f38 74c9b3e3 d2103f50 4aff607b eb40b799 5899b8a6 cd3c1abd'
+    )
+    private_B = (
+        '55188b3d 32f6bb9a 900afcfb eed4e72a 59cb9ac2 f19d7cfb 6b4fdd49 f47fc5fd'
+    )
+    public_A_x = (
+        '20b003d2 f297be2c 5e2c83a7 e9f9a5b9 eff49111 acf4fddb cc030148 0e359de6'
+    )
+    public_A_y = (
+        'dc809c49 652aeb6d 63329abf 5a52155c 766345c2 8fed3024 741c8ed0 1589d28b'
+    )
+    public_B_x = (
+        '1ea1f0f0 1faf1d96 09592284 f19e4c00 47b58afd 8615a69f 559077b2 2faaa190'
+    )
+    public_B_y = (
+        '4c55f33e 429dad37 7356703a 9ab85160 472d1130 e28e3676 5f89aff9 15b1214a'
+    )
+    dhkey = 'ec0234a3 57c8ad05 341010a6 0a397d9b 99796b13 b4f866f1 868d34f3 73bfa698'
+
+    key_a = EccKey.from_private_key_bytes(bytes.fromhex(private_A))
+    shared_key = key_a.dh(bytes.fromhex(public_B_x), bytes.fromhex(public_B_y))
+    assert shared_key == bytes.fromhex(dhkey)
+
+    key_b = EccKey.from_private_key_bytes(bytes.fromhex(private_B))
+    shared_key = key_b.dh(bytes.fromhex(public_A_x), bytes.fromhex(public_A_y))
+    assert shared_key == bytes.fromhex(dhkey)
+
+
+# -----------------------------------------------------------------------------
+def test_c1(crypto_backend):
+    k = bytes(16)
+    r = reversed_hex('5783D52156AD6F0E6388274EC6702EE0')
+    pres = reversed_hex('05000800000302')
+    preq = reversed_hex('07071000000101')
+    iat = 1
+    ia = reversed_hex('A1A2A3A4A5A6')
+    rat = 0
+    ra = reversed_hex('B1B2B3B4B5B6')
+    result = c1(k, r, preq, pres, iat, rat, ia, ra)
+    assert result == reversed_hex('1e1e3fef878988ead2a74dc5bef13b86')
+
+
+# -----------------------------------------------------------------------------
+def test_s1(crypto_backend):
+    k = bytes(16)
+    r1 = reversed_hex('000F0E0D0C0B0A091122334455667788')
+    r2 = reversed_hex('010203040506070899AABBCCDDEEFF00')
+    result = s1(k, r1, r2)
+    assert result == reversed_hex('9a1fe1f0e8b0f49b5b4216ae796da062')
+
+
+# -----------------------------------------------------------------------------
+def test_aes_cmac(crypto_backend):
+    m = b''
+    k = bytes.fromhex('2b7e1516 28aed2a6 abf71588 09cf4f3c')
+    cmac = aes_cmac(m, k)
+    assert cmac == bytes.fromhex('bb1d6929 e9593728 7fa37d12 9b756746')
+
+    m = bytes.fromhex('6bc1bee2 2e409f96 e93d7e11 7393172a')
+    cmac = aes_cmac(m, k)
+    assert cmac == bytes.fromhex('070a16b4 6b4d4144 f79bdd9d d04a287c')
+
+    m = bytes.fromhex(
+        '6bc1bee2 2e409f96 e93d7e11 7393172a'
+        + 'ae2d8a57 1e03ac9c 9eb76fac 45af8e51'
+        + '30c81c46 a35ce411'
+    )
+    cmac = aes_cmac(m, k)
+    assert cmac == bytes.fromhex('dfa66747 de9ae630 30ca3261 1497c827')
+
+    m = bytes.fromhex(
+        '6bc1bee2 2e409f96 e93d7e11 7393172a'
+        + 'ae2d8a57 1e03ac9c 9eb76fac 45af8e51'
+        + '30c81c46 a35ce411 e5fbc119 1a0a52ef'
+        + 'f69f2445 df4f9b17 ad2b417b e66c3710'
+    )
+    cmac = aes_cmac(m, k)
+    assert cmac == bytes.fromhex('51f0bebf 7e3b9d92 fc497417 79363cfe')
+
+
+# -----------------------------------------------------------------------------
+def test_f4(crypto_backend):
+    u = reversed_hex(
+        '20b003d2 f297be2c 5e2c83a7 e9f9a5b9 eff49111 acf4fddb cc030148 0e359de6'
+    )
+    v = reversed_hex(
+        '55188b3d 32f6bb9a 900afcfb eed4e72a 59cb9ac2 f19d7cfb 6b4fdd49 f47fc5fd'
+    )
+    x = reversed_hex('d5cb8454 d177733e ffffb2ec 712baeab')
+    z = b'\0'
+    value = f4(u, v, x, z)
+    assert value == reversed_hex('f2c916f1 07a9bd1c f1eda1be a974872d')
+
+
+# -----------------------------------------------------------------------------
+def test_f5(crypto_backend):
+    w = reversed_hex(
+        'ec0234a3 57c8ad05 341010a6 0a397d9b 99796b13 b4f866f1 868d34f3 73bfa698'
+    )
+    n1 = reversed_hex('d5cb8454 d177733e ffffb2ec 712baeab')
+    n2 = reversed_hex('a6e8e7cc 25a75f6e 216583f7 ff3dc4cf')
+    a1 = reversed_hex('00561237 37bfce')
+    a2 = reversed_hex('00a71370 2dcfc1')
+    value = f5(w, n1, n2, a1, a2)
+    assert value[0] == reversed_hex('2965f176 a1084a02 fd3f6a20 ce636e20')
+    assert value[1] == reversed_hex('69867911 69d7cd23 980522b5 94750a38')
+
+
+# -----------------------------------------------------------------------------
+def test_f6(crypto_backend):
+    n1 = reversed_hex('d5cb8454 d177733e ffffb2ec 712baeab')
+    n2 = reversed_hex('a6e8e7cc 25a75f6e 216583f7 ff3dc4cf')
+    mac_key = reversed_hex('2965f176 a1084a02 fd3f6a20 ce636e20')
+    r = reversed_hex('12a3343b b453bb54 08da42d2 0c2d0fc8')
+    io_cap = reversed_hex('010102')
+    a1 = reversed_hex('00561237 37bfce')
+    a2 = reversed_hex('00a71370 2dcfc1')
+    value = f6(mac_key, n1, n2, r, io_cap, a1, a2)
+    assert value == reversed_hex('e3c47398 9cd0e8c5 d26c0b09 da958f61')
+
+
+# -----------------------------------------------------------------------------
+def test_g2(crypto_backend):
+    u = reversed_hex(
+        '20b003d2 f297be2c 5e2c83a7 e9f9a5b9 eff49111 acf4fddb cc030148 0e359de6'
+    )
+    v = reversed_hex(
+        '55188b3d 32f6bb9a 900afcfb eed4e72a 59cb9ac2 f19d7cfb 6b4fdd49 f47fc5fd'
+    )
+    x = reversed_hex('d5cb8454 d177733e ffffb2ec 712baeab')
+    y = reversed_hex('a6e8e7cc 25a75f6e 216583f7 ff3dc4cf')
+    value = g2(u, v, x, y)
+    assert value == 0x2F9ED5BA
+
+
+# -----------------------------------------------------------------------------
+def test_h6(crypto_backend):
+    KEY = reversed_hex('ec0234a3 57c8ad05 341010a6 0a397d9b')
+    KEY_ID = bytes.fromhex('6c656272')
+    assert h6(KEY, KEY_ID) == reversed_hex('2d9ae102 e76dc91c e8d3a9e2 80b16399')
+
+
+# -----------------------------------------------------------------------------
+def test_h7(crypto_backend):
+    KEY = reversed_hex('ec0234a3 57c8ad05 341010a6 0a397d9b')
+    SALT = bytes.fromhex('00000000 00000000 00000000 746D7031')
+    assert h7(SALT, KEY) == reversed_hex('fb173597 c6a3c0ec d2998c2a 75a57011')
+
+
+# -----------------------------------------------------------------------------
+def test_ah(crypto_backend):
+    irk = reversed_hex('ec0234a3 57c8ad05 341010a6 0a397d9b')
+    prand = reversed_hex('708194')
+    value = ah(irk, prand)
+    expected = reversed_hex('0dfbaa')
+    assert value == expected

tests/profiles/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.