netmonitor_test.go

// Copyright 2019 The Alpaca Authors
//
// 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 main

import (
	"errors"
	"net"
	"testing"

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

type mockAddr string

func (a mockAddr) Network() string {
	return "ip+net"
}

func (a mockAddr) String() string {
	return string(a)
}

func toAddrs(ss ...string) []net.Addr {
	addrs := make([]net.Addr, len(ss))
	for i, s := range ss {
		addrs[i] = mockAddr(s)
	}
	return addrs
}

func TestNetworkMonitor(t *testing.T) {
	var next []net.Addr
	nm := &netMonitorImpl{getAddrs: func() ([]net.Addr, error) { return next, nil }}
	// Start with just loopback interfaces
	next = toAddrs("127.0.0.1/8", "::1/128")
	assert.True(t, nm.addrsChanged())
	// Connect to network, and get local IPv4 and IPv6 addresses
	next = toAddrs("127.0.0.1/8", "192.168.1.6/24", "::1/128", "fe80::dfd9:fe1d:56d1:1f3a/64")
	assert.True(t, nm.addrsChanged())
	// Stay connected, nothing changed
	next = toAddrs("127.0.0.1/8", "192.168.1.6/24", "::1/128", "fe80::dfd9:fe1d:56d1:1f3a/64")
	assert.False(t, nm.addrsChanged())
	// DHCP lease expires, get new addresses
	next = toAddrs("127.0.0.1/8", "192.168.1.7/24", "::1/128", "fe80::dfd9:fe1d:56d1:1f3b/64")
	assert.True(t, nm.addrsChanged())
	// Disconnect, and go back to having just loopback addresses
	next = toAddrs("127.0.0.1/8", "::1/128")
	assert.True(t, nm.addrsChanged())
}

func TestFailToGetAddrs(t *testing.T) {
	alwaysFail := func() ([]net.Addr, error) { return nil, errors.New("failed") }
	nm := &netMonitorImpl{getAddrs: alwaysFail}
	assert.False(t, nm.addrsChanged())
}