netstate/util_test.go

// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package netstate

import (
	"v.io/x/lib/netconfig"
)

import "net"

type ma struct {
	n, a string
}

func (a *ma) Network() string {
	return a.n
}

func (a *ma) String() string {
	return a.a
}

func MkAddr(n, a string) net.Addr {
	ip := net.ParseIP(a)
	if ip == nil {
		ip, _, _ = net.ParseCIDR(a)
	}
	return &ma{n: n, a: ip.String()}
}

func NewIPAddr(n, a string) Address {
	ip := net.ParseIP(a)
	return NewAddr(n, ip.String())
}

func NewAddr(n, a string) Address {
	return &address{
		addr: &ma{n: n, a: a},
	}
}

func NewInterface(name string, index int, addrs []net.Addr, routes []*netconfig.IPRoute) NetworkInterface {
	return &ipifc{
		name:     name,
		index:    index,
		addrs:    addrs,
		ipRoutes: routes,
	}
}

func CreateAndUseMockCache(ifcs []NetworkInterface, routetable RouteTable) func() {
	prev := internalCache
	internalCache = &netstateCache{
		current:    true,
		interfaces: ifcs,
		routes:     routetable,
	}
	return func() {
		internalCache = prev
		InvalidateCache()
	}
}