Merge "Add BaseDataTest" into tm-qpr-dev

/*

 * Copyright 2022 The Android Open Source Project

 *

 * 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

 *

 *      http://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.

 */

package com.android.bluetooth.bass_client;

import static com.google.common.truth.Truth.assertThat;

import static org.junit.Assert.assertThrows;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.junit.runners.JUnit4;

import java.util.LinkedHashSet;

import java.util.Set;

@RunWith(JUnit4.class)

public class BaseDataTest {

    @Test

    public void baseInformation() {

        BaseData.BaseInformation info = new BaseData.BaseInformation();

        assertThat(info.presentationDelay.length).isEqualTo(3);

        assertThat(info.codecId.length).isEqualTo(5);

        assertThat(info.isCodecIdUnknown()).isFalse();

        info.codecId[4] = (byte) 0xFE;

        assertThat(info.isCodecIdUnknown()).isTrue();

        // info.print() with different combination shouldn't crash.

        info.print();

        info.level = 1;

        info.codecConfigLength = 1;

        info.print();

        info.level = 2;

        info.metaDataLength = 1;

        info.keyMetadataDiff.add("metadata-diff");

        info.keyCodecCfgDiff.add("cfg-diff");

        info.print();

        info.level = 3;

        info.print();

    }

    @Test

    public void parseBaseData() {

        assertThrows(IllegalArgumentException.class, () -> BaseData.parseBaseData(null));

        byte[] serviceData = new byte[] {

                // LEVEL 1

                (byte) 0x01, (byte) 0x02, (byte) 0x03, // presentationDelay

                (byte) 0x01,  // numSubGroups

                // LEVEL 2

                (byte) 0x01,  // numSubGroups

                (byte) 0xFE,  // UNKNOWN_CODEC

                (byte) 0x02,  // codecConfigLength

                (byte) 0x01, (byte) 'A', // codecConfigInfo

                (byte) 0x03,  // metaDataLength

                (byte) 0x06, (byte) 0x07, (byte) 0x08,  // metaData

                // LEVEL 3

                (byte) 0x04,  // index

                (byte) 0x03,  // codecConfigLength

                (byte) 0x02, (byte) 'B', (byte) 'C' // codecConfigInfo

        };

        BaseData data = BaseData.parseBaseData(serviceData);

        BaseData.BaseInformation level = data.getLevelOne();

        assertThat(level.presentationDelay).isEqualTo(new byte[] { 0x01, 0x02, 0x03 });

        assertThat(level.numSubGroups).isEqualTo(1);

        assertThat(data.getLevelTwo().size()).isEqualTo(1);

        level = data.getLevelTwo().get(0);

        assertThat(level.numSubGroups).isEqualTo(1);

        assertThat(level.isCodecIdUnknown()).isTrue();

        assertThat(level.codecConfigLength).isEqualTo(2);

        assertThat(level.metaDataLength).isEqualTo(3);

        assertThat(data.getLevelThree().size()).isEqualTo(1);

        level = data.getLevelThree().get(0);

        assertThat(level.index).isEqualTo(4);

        assertThat(level.codecConfigLength).isEqualTo(3);

    }

}