Files
cloudpods/pkg/util/iproute2/address_test.go
Yousong Zhou 05b72a9d02 iproute2: address: make Exact idempotent
Idempotence against address not available for Exact is not a bad idea if
we look at the name of that func

This should accomodate the race condition where keepalived will remove
vips should the primary address used for vrrp comms is removed, causing
later addr del to fail with "cannot assign requested address"
2020-07-14 16:40:15 +08:00

114 lines
2.8 KiB
Go

// Copyright 2019 Yunion
//
// 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 iproute2
import (
"testing"
)
func TestAddress(t *testing.T) {
t.Run("exact some", func(t *testing.T) {
ifname := genDummyName(t)
dum := addDummy(t, ifname)
defer delDummy(t, dum)
emptyT := func(t *testing.T) {
l := NewAddress(ifname)
l.Exact()
if err := l.Err(); err != nil {
t.Fatalf("got err: %v", err)
}
if ipnets, err := l.List4(); err != nil {
t.Fatalf("list4 err: %v", err)
} else if len(ipnets) > 0 {
t.Fatalf("want empty, got %#v", ipnets)
}
}
want := "10.168.222.236/24"
l := NewAddress(ifname, want, "fe80::222:d5ff:fe9e:28d1/64")
l.Exact()
if err := l.Err(); err != nil {
t.Fatalf("got err: %v", err)
}
if ipnets, err := l.List4(); err != nil {
t.Fatalf("list4 err: %v", err)
} else if len(ipnets) != 1 {
t.Fatalf("want 1, got %#v", ipnets)
} else if got := ipnets[0]; got.String() != want {
t.Fatalf("want %s, got %s", want, got.String())
}
t.Run("empty", emptyT)
t.Run("empty empty", emptyT)
})
t.Run("exact indempotent", func(t *testing.T) {
ifname := genDummyName(t)
dum := addDummy(t, ifname)
defer delDummy(t, dum)
surprise := "10.127.190.240/32"
{
l := NewAddress(ifname, "10.127.190.217/24", surprise)
l.Exact()
if err := l.Err(); err != nil {
t.Fatalf("got err: %v", err)
}
}
{
l := NewAddress(ifname)
l.testcb = func() {
if err := NewAddress(ifname, surprise).Del().Err(); err != nil {
t.Fatalf("sneak failed: %v", err)
}
}
if addrs, _ := l.List4(); len(addrs) == 0 {
t.Fatalf("expect at least 1 v4 addr")
}
l.Exact()
if err := l.Err(); err != nil {
t.Fatalf("got err: %v", err)
}
}
})
}
func TestAddress_nopriv(t *testing.T) {
t.Run("bad address", func(t *testing.T) {
addresses := []string{
"192.168.2.1.1",
"192.168.2.1/33",
"192.168.2.1",
"0.0.0.0",
}
address := NewAddress("lo", addresses...)
address.Exact().Add().Del().List4()
if nerr := len(address.errs); nerr != len(addresses) {
t.Errorf("want %d err, got %d: %v ", len(addresses), nerr, address.Err())
}
})
t.Run("good address", func(t *testing.T) {
address := NewAddress("lo",
"192.168.2.1/0",
"192.168.2.1/1",
"0.0.0.0/0",
)
address.List4()
if err := address.Err(); err != nil {
t.Errorf("got err: %v", err)
}
})
}