1# Copyright 2021-2023 Google LLC
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7#      https://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14
15# -----------------------------------------------------------------------------
16# Imports
17# -----------------------------------------------------------------------------
18import pytest
19from bumble.codecs import AacAudioRtpPacket, BitReader
20
21
22# -----------------------------------------------------------------------------
23def test_reader():
24    reader = BitReader(b'')
25    with pytest.raises(ValueError):
26        reader.read(1)
27
28    reader = BitReader(b'hello')
29    with pytest.raises(ValueError):
30        reader.read(40)
31
32    reader = BitReader(bytes([0xFF]))
33    assert reader.read(1) == 1
34    with pytest.raises(ValueError):
35        reader.read(10)
36
37    reader = BitReader(bytes([0x78]))
38    value = 0
39    for _ in range(8):
40        value = (value << 1) | reader.read(1)
41    assert value == 0x78
42
43    data = bytes([x & 0xFF for x in range(66 * 100)])
44    reader = BitReader(data)
45    value = 0
46    for _ in range(100):
47        for bits in range(1, 33):
48            value = value << bits | reader.read(bits)
49    assert value == int.from_bytes(data, byteorder='big')
50
51
52def test_aac_rtp():
53    # pylint: disable=line-too-long
54    packet_data = bytes.fromhex(
55        '47fc0000b090800300202066000198000de120000000000000000000000000000000000000000000001c'
56    )
57    packet = AacAudioRtpPacket(packet_data)
58    adts = packet.to_adts()
59    assert adts == bytes.fromhex(
60        'fff1508004fffc2066000198000de120000000000000000000000000000000000000000000001c'
61    )
62
63
64# -----------------------------------------------------------------------------
65if __name__ == '__main__':
66    test_reader()
67    test_aac_rtp()
68