32 lines
901 B
Python
32 lines
901 B
Python
import time
|
|
|
|
from Crypt import CryptConnection
|
|
|
|
class TestConnection:
|
|
def testSslConnection(self, connection_server):
|
|
server = connection_server
|
|
assert server.running
|
|
|
|
# Connect to myself
|
|
connection = server.getConnection("127.0.0.1", 1544)
|
|
assert connection.handshake
|
|
assert connection.crypt
|
|
|
|
# Close connection
|
|
connection.close()
|
|
time.sleep(0.01)
|
|
assert len(server.connections) == 0
|
|
|
|
def testRawConnection(self, connection_server):
|
|
server = connection_server
|
|
crypt_supported_bk = CryptConnection.manager.crypt_supported
|
|
CryptConnection.manager.crypt_supported = []
|
|
|
|
connection = server.getConnection("127.0.0.1", 1544)
|
|
assert not connection.crypt
|
|
|
|
# Close connection
|
|
connection.close()
|
|
time.sleep(0.01)
|
|
assert len(server.connections) == 0
|