package netlink

import (
	"syscall"
	"testing"
)

func TestAddrAddDel(t *testing.T) {
	tearDown := setUpNetlinkTest(t)
	defer tearDown()

	link, err := LinkByName("lo")
	if err != nil {
		t.Fatal(err)
	}

	addr, err := ParseAddr("127.1.1.1/24 local")
	if err != nil {
		t.Fatal(err)
	}

	if err = AddrAdd(link, addr); err != nil {
		t.Fatal(err)
	}

	addrs, err := AddrList(link, syscall.AF_INET)
	if err != nil {
		t.Fatal(err)
	}

	if len(addrs) != 1 || !addr.Equal(addrs[0]) || addrs[0].Label != addr.Label {
		t.Fatal("Address not added properly")
	}

	if err = AddrDel(link, addr); err != nil {
		t.Fatal(err)
	}
	addrs, err = AddrList(link, syscall.AF_INET)
	if err != nil {
		t.Fatal(err)
	}

	if len(addrs) != 0 {
		t.Fatal("Address not removed properly")
	}
}