ice/candidatepair_test.go

package ice

import "testing"

var (
	hostCandidate = &Candidate{
		Type:            CandidateTypeHost,
		LocalPreference: defaultLocalPreference,
		Component:       ComponentRTP,
	}
	prflxCandidate = &Candidate{
		Type:            CandidateTypePeerReflexive,
		LocalPreference: defaultLocalPreference,
		Component:       ComponentRTP,
	}
	srflxCandidate = &Candidate{
		Type:            CandidateTypeServerReflexive,
		LocalPreference: defaultLocalPreference,
		Component:       ComponentRTP,
	}
	relayCandidate = &Candidate{
		Type:            CandidateTypeRelay,
		LocalPreference: defaultLocalPreference,
		Component:       ComponentRTP,
	}
)

func TestCandidatePairPriority(t *testing.T) {
	for _, test := range []struct {
		Pair         *candidatePair
		WantPriority uint64
	}{
		{
			Pair: newCandidatePair(
				hostCandidate,
				hostCandidate,
				false,
			),
			WantPriority: 9151314440652587007,
		},
		{
			Pair: newCandidatePair(
				hostCandidate,
				hostCandidate,
				true,
			),
			WantPriority: 9151314440652587007,
		},
		{
			Pair: newCandidatePair(
				hostCandidate,
				prflxCandidate,
				true,
			),
			WantPriority: 7998392936314175488,
		},
		{
			Pair: newCandidatePair(
				hostCandidate,
				prflxCandidate,
				false,
			),
			WantPriority: 7998392936314175487,
		},
		{
			Pair: newCandidatePair(
				hostCandidate,
				srflxCandidate,
				true,
			),
			WantPriority: 7277816996102668288,
		},
		{
			Pair: newCandidatePair(
				hostCandidate,
				srflxCandidate,
				false,
			),
			WantPriority: 7277816996102668287,
		},
		{
			Pair: newCandidatePair(
				hostCandidate,
				relayCandidate,
				true,
			),
			WantPriority: 72057593987596288,
		},
		{
			Pair: newCandidatePair(
				hostCandidate,
				relayCandidate,
				false,
			),
			WantPriority: 72057593987596287,
		},
	} {
		if got, want := test.Pair.Priority(), test.WantPriority; got != want {
			t.Fatalf("CandidatePair(%v).Priority() = %d, want %d", test.Pair, got, want)
		}
	}
}

func TestCandidatePairEquality(t *testing.T) {
	pairA := newCandidatePair(hostCandidate, srflxCandidate, true)
	pairB := newCandidatePair(hostCandidate, srflxCandidate, false)

	if !pairA.Equal(pairB) {
		t.Fatalf("Expected %v to equal %v", pairA, pairB)
	}
}