server_test.go

package corebgp

import (
	"net/netip"
	"reflect"
	"testing"

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

func TestServer(t *testing.T) {
	_, err := NewServer(netip.Addr{})
	assert.Error(t, err)

	_, err = NewServer(netip.MustParseAddr("::1"))
	assert.Error(t, err)

	s, err := NewServer(netip.MustParseAddr("127.0.0.1"))
	assert.NoError(t, err)

	err = s.AddPeer(PeerConfig{}, nil)
	assert.Error(t, err)

	err = s.AddPeer(PeerConfig{
		RemoteAddress: netip.MustParseAddr("127.0.0.2"),
		LocalAS:       64512,
		RemoteAS:      64513,
	}, nil, WithLocalAddress(netip.MustParseAddr("::1")))
	assert.Error(t, err)

	pcIPv4 := PeerConfig{
		RemoteAddress: netip.MustParseAddr("127.0.0.2"),
		LocalAS:       64512,
		RemoteAS:      64513,
	}
	err = s.AddPeer(pcIPv4, nil, WithLocalAddress(netip.MustParseAddr("127.0.0.1")))
	assert.NoError(t, err)
	err = s.AddPeer(pcIPv4, nil, WithLocalAddress(netip.MustParseAddr("127.0.0.1")))
	assert.ErrorIs(t, err, ErrPeerAlreadyExists)

	pcIPv6 := PeerConfig{
		RemoteAddress: netip.MustParseAddr("::2"),
		LocalAS:       64512,
		RemoteAS:      64513,
	}
	err = s.AddPeer(pcIPv6, nil, WithLocalAddress(netip.MustParseAddr("::1")))
	assert.NoError(t, err)
	err = s.AddPeer(pcIPv6, nil, WithLocalAddress(netip.MustParseAddr("::1")))
	assert.ErrorIs(t, err, ErrPeerAlreadyExists)

	pcs := s.ListPeers()
	if assert.Len(t, pcs, 2) {
		var found [2]bool
		for _, pc := range pcs {
			if reflect.DeepEqual(pc, pcIPv4) {
				found[0] = true
			}
			if reflect.DeepEqual(pc, pcIPv6) {
				found[1] = true
			}
		}
		assert.True(t, found[0])
		assert.True(t, found[1])
	}

	err = s.DeletePeer(pcIPv4.RemoteAddress)
	assert.NoError(t, err)
	err = s.DeletePeer(pcIPv4.RemoteAddress)
	assert.ErrorIs(t, err, ErrPeerNotExist)
}