mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
88 lines
2 KiB
Go
88 lines
2 KiB
Go
package ovmanager
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/docker/libnetwork/driverapi"
|
|
"github.com/docker/libnetwork/idm"
|
|
"github.com/docker/libnetwork/netlabel"
|
|
_ "github.com/docker/libnetwork/testutils"
|
|
"github.com/docker/libnetwork/types"
|
|
"gotest.tools/assert"
|
|
is "gotest.tools/assert/cmp"
|
|
)
|
|
|
|
func newDriver(t *testing.T) *driver {
|
|
d := &driver{
|
|
networks: networkTable{},
|
|
}
|
|
|
|
vxlanIdm, err := idm.New(nil, "vxlan-id", vxlanIDStart, vxlanIDEnd)
|
|
assert.NilError(t, err)
|
|
|
|
d.vxlanIdm = vxlanIdm
|
|
return d
|
|
}
|
|
|
|
func parseCIDR(t *testing.T, ipnet string) *net.IPNet {
|
|
subnet, err := types.ParseCIDR(ipnet)
|
|
assert.NilError(t, err)
|
|
return subnet
|
|
}
|
|
|
|
func TestNetworkAllocateFree(t *testing.T) {
|
|
d := newDriver(t)
|
|
|
|
ipamData := []driverapi.IPAMData{
|
|
{
|
|
Pool: parseCIDR(t, "10.1.1.0/24"),
|
|
},
|
|
{
|
|
Pool: parseCIDR(t, "10.1.2.0/24"),
|
|
},
|
|
}
|
|
|
|
vals, err := d.NetworkAllocate("testnetwork", nil, ipamData, nil)
|
|
assert.NilError(t, err)
|
|
|
|
vxlanIDs, ok := vals[netlabel.OverlayVxlanIDList]
|
|
assert.Check(t, is.Equal(true, ok))
|
|
assert.Check(t, is.Len(strings.Split(vxlanIDs, ","), 2))
|
|
|
|
err = d.NetworkFree("testnetwork")
|
|
assert.NilError(t, err)
|
|
}
|
|
|
|
func TestNetworkAllocateUserDefinedVNIs(t *testing.T) {
|
|
d := newDriver(t)
|
|
|
|
ipamData := []driverapi.IPAMData{
|
|
{
|
|
Pool: parseCIDR(t, "10.1.1.0/24"),
|
|
},
|
|
{
|
|
Pool: parseCIDR(t, "10.1.2.0/24"),
|
|
},
|
|
}
|
|
|
|
options := make(map[string]string)
|
|
// Intentionally add mode vnis than subnets
|
|
options[netlabel.OverlayVxlanIDList] = fmt.Sprintf("%d,%d,%d", vxlanIDStart, vxlanIDStart+1, vxlanIDStart+2)
|
|
|
|
vals, err := d.NetworkAllocate("testnetwork", options, ipamData, nil)
|
|
assert.NilError(t, err)
|
|
|
|
vxlanIDs, ok := vals[netlabel.OverlayVxlanIDList]
|
|
assert.Check(t, is.Equal(true, ok))
|
|
|
|
// We should only get exactly the same number of vnis as
|
|
// subnets. No more, no less, even if we passed more vnis.
|
|
assert.Check(t, is.Len(strings.Split(vxlanIDs, ","), 2))
|
|
assert.Check(t, is.Equal(fmt.Sprintf("%d,%d", vxlanIDStart, vxlanIDStart+1), vxlanIDs))
|
|
|
|
err = d.NetworkFree("testnetwork")
|
|
assert.NilError(t, err)
|
|
}
|