handle_linux_test.go

package netlink

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestSetGetSocketTimeout(t *testing.T) {
	timeout := 10 * time.Second
	if err := SetSocketTimeout(10 * time.Second); err != nil {
		t.Fatalf("Set socket timeout for default handle failed: %v", err)
	}

	if val := GetSocketTimeout(); val != timeout {
		t.Fatalf("Unexpected socket timeout value: got=%v, expected=%v", val, timeout)
	}
}

func TestConfigureHandle(t *testing.T) {
	orig := pkgHandle
	origDone := configDone
	t.Cleanup(func() {
		configMu.Lock()
		defer configMu.Unlock()

		pkgHandle = orig
		configDone = origDone
	})

	assert.NoError(t, ConfigureHandle(HandleOptions{DisableVFInfoCollection: true}))
	assert.NotEqual(t, orig, pkgHandle)
	assert.NoError(t, pkgHandle.Close())
	assert.Error(t, ConfigureHandle(HandleOptions{}))
}