cloudflared-mirror/connection/discovery_test.go

package connection

import (
	"net"
	"testing"

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

type mockEdgeServiceDiscoverer struct {
}

func (mr *mockEdgeServiceDiscoverer) Addr() *net.TCPAddr {
	return &net.TCPAddr{
		IP:   net.ParseIP("127.0.0.1"),
		Port: 63102,
	}
}

func (mr *mockEdgeServiceDiscoverer) AvailableAddrs() uint8 {
	return 1
}

func (mr *mockEdgeServiceDiscoverer) Refresh() error {
	return nil
}

func TestFlattenServiceIPs(t *testing.T) {
	result := flattenServiceIPs([][]*net.TCPAddr{
		[]*net.TCPAddr{
			&net.TCPAddr{Port: 1},
			&net.TCPAddr{Port: 2},
			&net.TCPAddr{Port: 3},
			&net.TCPAddr{Port: 4},
		},
		[]*net.TCPAddr{
			&net.TCPAddr{Port: 10},
			&net.TCPAddr{Port: 12},
			&net.TCPAddr{Port: 13},
		},
		[]*net.TCPAddr{
			&net.TCPAddr{Port: 21},
			&net.TCPAddr{Port: 22},
			&net.TCPAddr{Port: 23},
			&net.TCPAddr{Port: 24},
			&net.TCPAddr{Port: 25},
		},
	})
	assert.EqualValues(t, []*net.TCPAddr{
		&net.TCPAddr{Port: 1},
		&net.TCPAddr{Port: 10},
		&net.TCPAddr{Port: 21},
		&net.TCPAddr{Port: 2},
		&net.TCPAddr{Port: 12},
		&net.TCPAddr{Port: 22},
		&net.TCPAddr{Port: 3},
		&net.TCPAddr{Port: 13},
		&net.TCPAddr{Port: 23},
		&net.TCPAddr{Port: 4},
		&net.TCPAddr{Port: 24},
		&net.TCPAddr{Port: 25},
	}, result)
}