swifttunnel_linux_test.go

//go:build linux

package swiftunnel

import (
	"github.com/XenonCommunity/swiftunnel/swiftypes"
	"net"
	"testing"
)

func TestNewSwiftInterface(t *testing.T) {
	ip, ipNet, err := net.ParseCIDR("172.0.10.2/24")
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}

	config := Config{
		AdapterName: "tun0",
		AdapterType: swiftypes.AdapterTypeTUN,
		MTU:         1500,
		UnicastConfig: &swiftypes.UnicastConfig{
			IPNet: ipNet,
			IP:    ip,
		},
	}

	// Create a new SwiftInterface
	adapter, err := NewSwiftInterface(config)
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}
	defer adapter.Close()

	// Check if the adapter name is set correctly
	name, err := adapter.GetAdapterName()
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}

	if name != config.AdapterName {
		t.Errorf("expected adapter name %s, got %s", config.AdapterName, name)
	}

	// Check if the adapter index can be retrieved
	index, err := adapter.GetAdapterIndex()
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}
	if index == 0 {
		t.Error("expected a valid adapter index, got 0")
	}

}

func TestSetMTU(t *testing.T) {
	adapter, err := NewSwiftInterface(Config{
		AdapterName: "tun0",
		AdapterType: swiftypes.AdapterTypeTUN,
	})
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}
	defer adapter.Close()

	if err := adapter.SetMTU(1400); err != nil {
		t.Fatalf("expected no error setting MTU, got %v", err)
	}

}

func TestSetUnicastIpAddressEntry(t *testing.T) {
	adapter, err := NewSwiftInterface(Config{
		AdapterName: "tun0",
		AdapterType: swiftypes.AdapterTypeTUN,
	})
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}
	defer adapter.Close()

	ip, ipNet, err := net.ParseCIDR("172.0.10.2/24")
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}

	unicastConfig := &swiftypes.UnicastConfig{
		IPNet: ipNet,
		IP:    ip,
	}
	if err := adapter.SetUnicastIpAddressEntry(unicastConfig); err != nil {
		t.Fatalf("expected no error setting IP address, got %v", err)
	}
}

func TestTunReadCloser_Read(t *testing.T) {
	ip, ipNet, err := net.ParseCIDR("172.0.10.2/24")
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}

	config := Config{
		AdapterName: "tun0",
		AdapterType: swiftypes.AdapterTypeTUN,
		MTU:         1500,
		UnicastConfig: &swiftypes.UnicastConfig{
			IPNet: ipNet,
			IP:    ip,
		},
	}

	// Create a new SwiftInterface
	adapter, err := NewSwiftInterface(config)
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}
	defer adapter.Close()

	if err := adapter.SetMTU(1400); err != nil {
		t.Fatalf("expected no error setting MTU, got %v", err)
	}

	if err := adapter.SetStatus(swiftypes.InterfaceUp); err != nil {
		t.Fatalf("expected no error setting status, got %v", err)
	}

}