diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-16 19:19:13 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-16 19:19:13 +0000 |
commit | ccd992355df7192993c666236047820244914598 (patch) | |
tree | f00fea65147227b7743083c6148396f74cd66935 /src/net/main_unix_test.go | |
parent | Initial commit. (diff) | |
download | golang-1.21-ccd992355df7192993c666236047820244914598.tar.xz golang-1.21-ccd992355df7192993c666236047820244914598.zip |
Adding upstream version 1.21.8.upstream/1.21.8
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/net/main_unix_test.go')
-rw-r--r-- | src/net/main_unix_test.go | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/src/net/main_unix_test.go b/src/net/main_unix_test.go new file mode 100644 index 0000000..e7a5b4f --- /dev/null +++ b/src/net/main_unix_test.go @@ -0,0 +1,55 @@ +// Copyright 2015 The Go Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +//go:build unix + +package net + +import "internal/poll" + +var ( + // Placeholders for saving original socket system calls. + origSocket = socketFunc + origClose = poll.CloseFunc + origConnect = connectFunc + origListen = listenFunc + origAccept = poll.AcceptFunc + origGetsockoptInt = getsockoptIntFunc + + extraTestHookInstallers []func() + extraTestHookUninstallers []func() +) + +func installTestHooks() { + socketFunc = sw.Socket + poll.CloseFunc = sw.Close + connectFunc = sw.Connect + listenFunc = sw.Listen + poll.AcceptFunc = sw.Accept + getsockoptIntFunc = sw.GetsockoptInt + + for _, fn := range extraTestHookInstallers { + fn() + } +} + +func uninstallTestHooks() { + socketFunc = origSocket + poll.CloseFunc = origClose + connectFunc = origConnect + listenFunc = origListen + poll.AcceptFunc = origAccept + getsockoptIntFunc = origGetsockoptInt + + for _, fn := range extraTestHookUninstallers { + fn() + } +} + +// forceCloseSockets must be called only from TestMain. +func forceCloseSockets() { + for s := range sw.Sockets() { + poll.CloseFunc(s) + } +} |