//go:build linux
// +build linux

package ipvlan

import (
	"testing"

	"github.com/vishvananda/netlink"
)

// TestValidateLink tests the parentExists function
func TestValidateLink(t *testing.T) {
	validIface := "lo"
	invalidIface := "foo12345"

	// test a valid parent interface validation
	if ok := parentExists(validIface); !ok {
		t.Fatalf("failed validating loopback %s", validIface)
	}
	// test an invalid parent interface validation
	if ok := parentExists(invalidIface); ok {
		t.Fatalf("failed to invalidate interface %s", invalidIface)
	}
}

// TestValidateSubLink tests valid 802.1q naming convention
func TestValidateSubLink(t *testing.T) {
	validSubIface := "lo.10"
	invalidSubIface1 := "lo"
	invalidSubIface2 := "lo:10"
	invalidSubIface3 := "foo123.456"

	// test a valid parent_iface.vlan_id
	_, _, err := parseVlan(validSubIface)
	if err != nil {
		t.Fatalf("failed subinterface validation: %v", err)
	}
	// test an invalid vid with a valid parent link
	_, _, err = parseVlan(invalidSubIface1)
	if err == nil {
		t.Fatalf("failed subinterface validation test: %s", invalidSubIface1)
	}
	// test a valid vid with a valid parent link with an invalid delimiter
	_, _, err = parseVlan(invalidSubIface2)
	if err == nil {
		t.Fatalf("failed subinterface validation test: %v", invalidSubIface2)
	}
	// test an invalid parent link with a valid vid
	_, _, err = parseVlan(invalidSubIface3)
	if err == nil {
		t.Fatalf("failed subinterface validation test: %v", invalidSubIface3)
	}
}

// TestSetIPVlanMode tests the ipvlan mode setter
func TestSetIPVlanMode(t *testing.T) {
	// test ipvlan l2 mode
	mode, err := setIPVlanMode(modeL2)
	if err != nil {
		t.Fatalf("error parsing %v vlan mode: %v", mode, err)
	}
	if mode != netlink.IPVLAN_MODE_L2 {
		t.Fatalf("expected %d got %d", netlink.IPVLAN_MODE_L2, mode)
	}
	// test ipvlan l3 mode
	mode, err = setIPVlanMode(modeL3)
	if err != nil {
		t.Fatalf("error parsing %v vlan mode: %v", mode, err)
	}
	if mode != netlink.IPVLAN_MODE_L3 {
		t.Fatalf("expected %d got %d", netlink.IPVLAN_MODE_L3, mode)
	}
	// test ipvlan l3s mode
	mode, err = setIPVlanMode(modeL3S)
	if err != nil {
		t.Fatalf("error parsing %v vlan mode: %v", mode, err)
	}
	if mode != netlink.IPVLAN_MODE_L3S {
		t.Fatalf("expected %d got %d", netlink.IPVLAN_MODE_L3S, mode)
	}
	// test invalid mode
	mode, err = setIPVlanMode("foo")
	if err == nil {
		t.Fatal("invalid ipvlan mode should have returned an error")
	}
	if mode != 0 {
		t.Fatalf("expected 0 got %d", mode)
	}
	// test null mode
	mode, err = setIPVlanMode("")
	if err == nil {
		t.Fatal("invalid ipvlan mode should have returned an error")
	}
	if mode != 0 {
		t.Fatalf("expected 0 got %d", mode)
	}
}

// TestSetIPVlanFlag tests the ipvlan flag setter
func TestSetIPVlanFlag(t *testing.T) {
	// test ipvlan bridge flag
	flag, err := setIPVlanFlag(flagBridge)
	if err != nil {
		t.Fatalf("error parsing %v vlan flag: %v", flag, err)
	}
	if flag != netlink.IPVLAN_FLAG_BRIDGE {
		t.Fatalf("expected %d got %d", netlink.IPVLAN_FLAG_BRIDGE, flag)
	}
	// test ipvlan private flag
	flag, err = setIPVlanFlag(flagPrivate)
	if err != nil {
		t.Fatalf("error parsing %v vlan flag: %v", flag, err)
	}
	if flag != netlink.IPVLAN_FLAG_PRIVATE {
		t.Fatalf("expected %d got %d", netlink.IPVLAN_FLAG_PRIVATE, flag)
	}
	// test ipvlan vepa flag
	flag, err = setIPVlanFlag(flagVepa)
	if err != nil {
		t.Fatalf("error parsing %v vlan flag: %v", flag, err)
	}
	if flag != netlink.IPVLAN_FLAG_VEPA {
		t.Fatalf("expected %d got %d", netlink.IPVLAN_FLAG_VEPA, flag)
	}
	// test invalid flag
	flag, err = setIPVlanFlag("foo")
	if err == nil {
		t.Fatal("invalid ipvlan flag should have returned an error")
	}
	if flag != 0 {
		t.Fatalf("expected 0 got %d", flag)
	}
	// test null flag
	flag, err = setIPVlanFlag("")
	if err == nil {
		t.Fatal("invalid ipvlan flag should have returned an error")
	}
	if flag != 0 {
		t.Fatalf("expected 0 got %d", flag)
	}
}