// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
// SPDX-License-Identifier: MIT

package rtcp

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestGetPadding(t *testing.T) {
	assert := assert.New(t)
	type testCase struct {
		input  int
		result int
	}

	cases := []testCase{
		{input: 0, result: 0},
		{input: 1, result: 3},
		{input: 2, result: 2},
		{input: 3, result: 1},
		{input: 4, result: 0},
		{input: 100, result: 0},
		{input: 500, result: 0},
	}
	for _, testCase := range cases {
		assert.Equalf(
			getPadding(testCase.input), testCase.result, "Test case returned wrong value for input %d", testCase.input,
		)
	}
}

func TestSetNBitsOfUint16(t *testing.T) {
	for _, test := range []struct {
		name        string
		source      uint16
		size        uint16
		index       uint16
		value       uint16
		result      uint16
		expectedErr error
	}{
		{
			"setOneBit", 0, 1, 8, 1, 128, nil,
		},
		{
			"setStatusVectorBit", 0, 1, 0, 1, 32768, nil,
		},
		{
			"setStatusVectorSecondBit", 32768, 1, 1, 1, 49152, nil,
		},
		{
			"setStatusVectorInnerBitsAndCutValue", 49152, 2, 6, 11111, 49920, nil,
		},
		{
			"setRunLengthSecondTwoBit", 32768, 2, 1, 1, 40960, nil,
		},
		{
			"setOneBitOutOfBounds", 32768, 2, 15, 1, 0, errInvalidSizeOrStartIndex,
		},
	} {
		test := test
		t.Run(test.name, func(t *testing.T) {
			got, err := setNBitsOfUint16(test.source, test.size, test.index, test.value)
			if test.expectedErr != nil {
				assert.ErrorIs(t, err, test.expectedErr)
			} else {
				assert.NoError(t, err)
			}

			assert.Equalf(t, test.result, got, "setNBitsOfUint16 %q", test.name)
		})
	}
}