diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 13:14:23 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 13:14:23 +0000 |
commit | 73df946d56c74384511a194dd01dbe099584fd1a (patch) | |
tree | fd0bcea490dd81327ddfbb31e215439672c9a068 /src/net/main_windows_test.go | |
parent | Initial commit. (diff) | |
download | golang-1.16-73df946d56c74384511a194dd01dbe099584fd1a.tar.xz golang-1.16-73df946d56c74384511a194dd01dbe099584fd1a.zip |
Adding upstream version 1.16.10.upstream/1.16.10upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | src/net/main_windows_test.go | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/src/net/main_windows_test.go b/src/net/main_windows_test.go new file mode 100644 index 0000000..07f21b7 --- /dev/null +++ b/src/net/main_windows_test.go @@ -0,0 +1,45 @@ +// 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. + +package net + +import "internal/poll" + +var ( + // Placeholders for saving original socket system calls. + origSocket = socketFunc + origWSASocket = wsaSocketFunc + origClosesocket = poll.CloseFunc + origConnect = connectFunc + origConnectEx = poll.ConnectExFunc + origListen = listenFunc + origAccept = poll.AcceptFunc +) + +func installTestHooks() { + socketFunc = sw.Socket + wsaSocketFunc = sw.WSASocket + poll.CloseFunc = sw.Closesocket + connectFunc = sw.Connect + poll.ConnectExFunc = sw.ConnectEx + listenFunc = sw.Listen + poll.AcceptFunc = sw.AcceptEx +} + +func uninstallTestHooks() { + socketFunc = origSocket + wsaSocketFunc = origWSASocket + poll.CloseFunc = origClosesocket + connectFunc = origConnect + poll.ConnectExFunc = origConnectEx + listenFunc = origListen + poll.AcceptFunc = origAccept +} + +// forceCloseSockets must be called only from TestMain. +func forceCloseSockets() { + for s := range sw.Sockets() { + poll.CloseFunc(s) + } +} |