cloudflare/cloudflared

Public

mirrored from https://github.com/cloudflare/cloudflaredAvailable

CodeCommitsIssuesPull requestsActionsInsightsSecurity
2018.12.0

Branches

Tags

  • No tags available.
0Branches0Tags
Go to file
Add file
Code

Clone

HTTPS

Download ZIP

origin/discovery_test.go

45lines · modepreview

package origin

import (
	"net"
	"testing"

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

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)
}