// Copyright 2020 xgfone, 2023 idk
//
// 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 peerprotocol

import (
	"testing"
)

func TestBitField(t *testing.T) {
	bf := NewBitFieldFromBools([]bool{
		false, true, false, true, false, true, true, true,
		false, false, true, false, false, true, true, false})
	if bf.IsSet(0) {
		t.Error(0)
	} else if !bf.IsSet(1) {
		t.Error(1)
	} else if !bf.IsSet(7) {
		t.Error(7)
	} else if bf.IsSet(8) {
		t.Error(8)
	} else if bf.IsSet(15) {
		t.Error(15)
	}

	bf.Set(9)
	if !bf.IsSet(9) {
		t.Error(9)
	}

	bf.Unset(10)
	if bf.IsSet(10) {
		t.Error(10)
	}

	bs := bf.Bools()
	if len(bs) != 16 {
		t.Fatal(bs)
	} else if !bs[9] {
		t.Error(9)
	} else if bs[10] {
		t.Error(10)
	}

	bf = NewBitField(16, true)
	if !bf.IsSet(0) || !bf.IsSet(1) || !bf.IsSet(2) || !bf.IsSet(3) ||
		!bf.IsSet(4) || !bf.IsSet(5) || !bf.IsSet(6) || !bf.IsSet(7) {
		t.Error(bf)
	}
}

func TestPieces(t *testing.T) {
	ps := Pieces{2, 3, 4, 5}
	ps = ps.Append(1)
	if len(ps) != 5 || ps[0] != 1 {
		t.Fatal(ps)
	}

	ps = ps.Append(5)
	if len(ps) != 5 {
		t.Fatal(ps)
	}

	ps = ps.Remove(6)
	if len(ps) != 5 {
		t.Fatal(ps)
	}

	ps = ps.Remove(3)
	if len(ps) != 4 || ps[0] != 1 || ps[1] != 2 || ps[2] != 4 || ps[3] != 5 {
		t.Fatal(ps)
	}
}