package bridge

import (
	"net"
	"testing"

	"github.com/docker/libnetwork"
	"github.com/vishvananda/netlink"
)

func TestLinkCreate(t *testing.T) {
	defer libnetwork.SetupTestNetNS(t)()
	d := &driver{}

	config := &Configuration{
		BridgeName: DefaultBridgeName,
		EnableIPv6: true}
	netw, err := d.CreateNetwork("dummy", config)
	if err != nil {
		t.Fatalf("Failed to create bridge: %v", err)
	}

	interfaces, err := netw.Link("ep")
	if err != nil {
		t.Fatalf("Failed to create a link: %v", err)
	}

	if len(interfaces) != 1 {
		t.Fatalf("Expected exactly one interface. Instead got %d interface(s)", len(interfaces))
	}

	if interfaces[0].DstName == "" {
		t.Fatal("Invalid Dstname returned")
	}

	_, err = netlink.LinkByName(interfaces[0].SrcName)
	if err != nil {
		t.Fatalf("Could not find source link %s: %v", interfaces[0].SrcName, err)
	}

	ip, _, err := net.ParseCIDR(interfaces[0].Address)
	if err != nil {
		t.Fatalf("Invalid IPv4 address returned, ip = %s: %v", interfaces[0].Address, err)
	}

	b := netw.(*bridgeNetwork)
	if !b.bridge.bridgeIPv4.Contains(ip) {
		t.Fatalf("IP %s is not a valid ip in the subnet %s", ip.String(), b.bridge.bridgeIPv4.String())
	}

	ip6, _, err := net.ParseCIDR(interfaces[0].AddressIPv6)
	if err != nil {
		t.Fatalf("Invalid IPv6 address returned, ip = %s: %v", interfaces[0].AddressIPv6, err)
	}

	if !b.bridge.bridgeIPv6.Contains(ip6) {
		t.Fatalf("IP %s is not a valid ip in the subnet %s", ip6.String(), bridgeIPv6.String())
	}

	if interfaces[0].Gateway != b.bridge.bridgeIPv4.IP.String() {
		t.Fatalf("Invalid default gateway. Expected %s. Got %s", b.bridge.bridgeIPv4.IP.String(),
			interfaces[0].Gateway)
	}

	if interfaces[0].GatewayIPv6 != b.bridge.bridgeIPv6.IP.String() {
		t.Fatalf("Invalid default gateway for IPv6. Expected %s. Got %s", b.bridge.bridgeIPv6.IP.String(),
			interfaces[0].GatewayIPv6)
	}
}

func TestLinkCreateTwo(t *testing.T) {
	defer libnetwork.SetupTestNetNS(t)()
	d := &driver{}

	config := &Configuration{
		BridgeName: DefaultBridgeName,
		EnableIPv6: true}
	netw, err := d.CreateNetwork("dummy", config)
	if err != nil {
		t.Fatalf("Failed to create bridge: %v", err)
	}

	_, err = netw.Link("ep")
	if err != nil {
		t.Fatalf("Failed to create a link: %v", err)
	}

	_, err = netw.Link("ep1")
	if err != nil {
		if err != ErrEndpointExists {
			t.Fatalf("Failed with a wrong error :%v", err)
		}
	} else {
		t.Fatalf("Expected to fail while trying to add more than one endpoint")
	}
}

func TestLinkCreateNoEnableIPv6(t *testing.T) {
	defer libnetwork.SetupTestNetNS(t)()
	d := &driver{}

	config := &Configuration{
		BridgeName: DefaultBridgeName}
	netw, err := d.CreateNetwork("dummy", config)
	if err != nil {
		t.Fatalf("Failed to create bridge: %v", err)
	}

	interfaces, err := netw.Link("ep")
	if err != nil {
		t.Fatalf("Failed to create a link: %v", err)
	}

	if interfaces[0].AddressIPv6 != "" ||
		interfaces[0].GatewayIPv6 != "" {
		t.Fatalf("Expected IPv6 address and GatewayIPv6 to be empty when IPv6 enabled. Instead got IPv6 = %s and GatewayIPv6 = %s",
			interfaces[0].AddressIPv6, interfaces[0].GatewayIPv6)
	}
}