netlink_test.go 603 B

12345678910111213141516171819202122232425262728293031323334
  1. package netlink
  2. import (
  3. "log"
  4. "os"
  5. "runtime"
  6. "testing"
  7. "github.com/vishvananda/netns"
  8. )
  9. type tearDownNetlinkTest func()
  10. func setUpNetlinkTest(t *testing.T) tearDownNetlinkTest {
  11. if os.Getuid() != 0 {
  12. msg := "Skipped test because it requires root privileges."
  13. log.Printf(msg)
  14. t.Skip(msg)
  15. }
  16. // new temporary namespace so we don't pollute the host
  17. // lock thread since the namespace is thread local
  18. runtime.LockOSThread()
  19. var err error
  20. ns, err := netns.New()
  21. if err != nil {
  22. t.Fatal("Failed to create newns", ns)
  23. }
  24. return func() {
  25. ns.Close()
  26. runtime.UnlockOSThread()
  27. }
  28. }