Mirrors/sftpgo
1
0
Fork 0
mirror of https://github.com/drakkan/sftpgo.git synced 2025-12-07 06:40:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

ensure no client is connected before running max connections test cases

Browse Source
This commit is contained in:
Nicola Murino
2021-05-11 08:04:57 +02:00
parent c8f7fc9bc9
commit b67cd0d3df
6 changed files with 56 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
common/common_test.go
View File

@@ -276,9 +276,13 @@ func TestMaxConnectionPerHost(t *testing.T) {
Connections.AddClientConnection(ipAddr)
assert.False(t, Connections.IsNewConnectionAllowed(ipAddr))
assert.Equal(t, int32(3), Connections.GetClientConnections())
Connections.RemoveClientConnection(ipAddr)
Connections.RemoveClientConnection(ipAddr)
Connections.RemoveClientConnection(ipAddr)
assert.Equal(t, int32(0), Connections.GetClientConnections())
Config.MaxPerHostConnections = oldValue
}
@@ -357,6 +361,7 @@ func TestIdleConnections(t *testing.T) {
defer Connections.RUnlock()
return len(Connections.sshConnections) == 0
}, 1*time.Second, 200*time.Millisecond)
assert.Equal(t, int32(0), Connections.GetClientConnections())
stopIdleTimeoutTicker()
assert.True(t, customConn1.isClosed)
assert.True(t, customConn2.isClosed)