OrgGo/shirou_gopsutil
1
0
Fork 0
mirror of https://github.com/shirou/gopsutil.git synced 2025-04-26 13:48:59 +08:00
Code Issues Projects Releases Wiki Activity

Wait for server connection to be established before checking

Browse Source 
connections.
This commit is contained in:
Tom Barker 2021-08-23 16:30:51 -04:00
parent bc4661937d
commit 9248140c98
2 changed files with 60 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
process/process_test.go
View File

@ -516,6 +516,7 @@ func Test_Connections(t *testing.T) {
t.Fatalf("unable to parse tcpServerAddr port: %v", err)
}
serverEstablished := make(chan struct{})
go func() { // TCP listening goroutine
conn, err := l.Accept()
if err != nil {
@ -523,6 +524,7 @@ func Test_Connections(t *testing.T) {
}
defer conn.Close()
serverEstablished <- struct{}{}
_, err = ioutil.ReadAll(conn)
if err != nil {
panic(err)
@ -535,6 +537,10 @@ func Test_Connections(t *testing.T) {
}
defer conn.Close()
// Rarely the call to net.Dial returns before the server connection is
// established. Wait so that the test doesn't fail.
<-serverEstablished
c, err := p.Connections()
skipIfNotImplementedErr(t, err)
if err != nil {
@ -543,14 +549,33 @@ func Test_Connections(t *testing.T) {
if len(c) == 0 {
t.Fatal("no connections found")
}
found := 0
serverConnections := 0
for _, connection := range c {
if connection.Status == "ESTABLISHED" && (connection.Laddr.IP == tcpServerAddrIP && connection.Laddr.Port == uint32(tcpServerAddrPort)) || (connection.Raddr.IP == tcpServerAddrIP && connection.Raddr.Port == uint32(tcpServerAddrPort)) {
found++
if connection.Laddr.IP == tcpServerAddrIP && connection.Laddr.Port == uint32(tcpServerAddrPort) && connection.Raddr.Port != 0 {
if connection.Status != "ESTABLISHED" {
t.Fatalf("expected server connection to be ESTABLISHED, have %+v", connection)
}
serverConnections++
}
}
if found != 2 { // two established connections, one for the server, the other for the client
t.Fatalf("wrong connections: %+v", c)
clientConnections := 0
for _, connection := range c {
if connection.Raddr.IP == tcpServerAddrIP && connection.Raddr.Port == uint32(tcpServerAddrPort) {
if connection.Status != "ESTABLISHED" {
t.Fatalf("expected client connection to be ESTABLISHED, have %+v", connection)
}
clientConnections++
}
}
if serverConnections != 1 { // two established connections, one for the server, the other for the client
t.Fatalf("expected 1 server connection, have %d.\nDetails: %+v", serverConnections, c)
}
if clientConnections != 1 { // two established connections, one for the server, the other for the client
t.Fatalf("expected 1 server connection, have %d.\nDetails: %+v", clientConnections, c)
}
}

35
v3/process/process_test.go
View File

@ -518,6 +518,7 @@ func Test_Connections(t *testing.T) {
t.Fatalf("unable to parse tcpServerAddr port: %v", err)
}
serverEstablished := make(chan struct{})
go func() { // TCP listening goroutine
conn, err := l.Accept()
if err != nil {
@ -525,6 +526,7 @@ func Test_Connections(t *testing.T) {
}
defer conn.Close()
serverEstablished <- struct{}{}
_, err = ioutil.ReadAll(conn)
if err != nil {
panic(err)
@ -537,6 +539,10 @@ func Test_Connections(t *testing.T) {
}
defer conn.Close()
// Rarely the call to net.Dial returns before the server connection is
// established. Wait so that the test doesn't fail.
<-serverEstablished
c, err := p.Connections()
skipIfNotImplementedErr(t, err)
if err != nil {
@ -545,14 +551,33 @@ func Test_Connections(t *testing.T) {
if len(c) == 0 {
t.Fatal("no connections found")
}
found := 0
serverConnections := 0
for _, connection := range c {
if connection.Status == "ESTABLISHED" && (connection.Laddr.IP == tcpServerAddrIP && connection.Laddr.Port == uint32(tcpServerAddrPort)) || (connection.Raddr.IP == tcpServerAddrIP && connection.Raddr.Port == uint32(tcpServerAddrPort)) {
found++
if connection.Laddr.IP == tcpServerAddrIP && connection.Laddr.Port == uint32(tcpServerAddrPort) && connection.Raddr.Port != 0 {
if connection.Status != "ESTABLISHED" {
t.Fatalf("expected server connection to be ESTABLISHED, have %+v", connection)
}
serverConnections++
}
}
if found != 2 { // two established connections, one for the server, the other for the client
t.Fatalf("wrong connections: %+v", c)
clientConnections := 0
for _, connection := range c {
if connection.Raddr.IP == tcpServerAddrIP && connection.Raddr.Port == uint32(tcpServerAddrPort) {
if connection.Status != "ESTABLISHED" {
t.Fatalf("expected client connection to be ESTABLISHED, have %+v", connection)
}
clientConnections++
}
}
if serverConnections != 1 { // two established connections, one for the server, the other for the client
t.Fatalf("expected 1 server connection, have %d.\nDetails: %+v", serverConnections, c)
}
if clientConnections != 1 { // two established connections, one for the server, the other for the client
t.Fatalf("expected 1 server connection, have %d.\nDetails: %+v", clientConnections, c)
}
}