Commit bf173404 authored by Julien Muchembled's avatar Julien Muchembled

qa: fix resource leak with loopback connections in threaded tests

parent 3bfd4cd4
...@@ -321,6 +321,7 @@ class ServerNode(Node): ...@@ -321,6 +321,7 @@ class ServerNode(Node):
master_nodes = kw.get('master_nodes', cluster.master_nodes) master_nodes = kw.get('master_nodes', cluster.master_nodes)
name = kw.get('name', cluster.name) name = kw.get('name', cluster.name)
port = address[1] port = address[1]
if address is not BIND:
self._node_list[port] = weakref.proxy(self) self._node_list[port] = weakref.proxy(self)
self._init_args = init_args = kw.copy() self._init_args = init_args = kw.copy()
init_args['cluster'] = cluster init_args['cluster'] = cluster
...@@ -933,20 +934,17 @@ class NEOThreadedTest(NeoTestBase): ...@@ -933,20 +934,17 @@ class NEOThreadedTest(NeoTestBase):
tic = Serialized.tic tic = Serialized.tic
@contextmanager
def getLoopbackConnection(self): def getLoopbackConnection(self):
app = MasterApplication(getSSL=NEOCluster.SSL, app = MasterApplication(address=BIND,
getReplicas=0, getPartitions=1) getSSL=NEOCluster.SSL, getReplicas=0, getPartitions=1)
try:
handler = EventHandler(app) handler = EventHandler(app)
app.listening_conn = ListeningConnection(app, handler, app.server) app.listening_conn = ListeningConnection(app, handler, app.server)
node = app.nm.createMaster(address=app.listening_conn.getAddress(), yield ClientConnection(app, handler, app.nm.createMaster(
uuid=app.uuid) address=app.listening_conn.getAddress(), uuid=app.uuid))
conn = ClientConnection.__new__(ClientConnection) finally:
def reset(): app.close()
conn.__dict__.clear()
conn.__init__(app, handler, node)
conn.reset = reset
reset()
return conn
def getUnpickler(self, conn): def getUnpickler(self, conn):
reader = conn._reader reader = conn._reader
......
...@@ -1209,14 +1209,14 @@ class Test(NEOThreadedTest): ...@@ -1209,14 +1209,14 @@ class Test(NEOThreadedTest):
self.assertEqual(s.dm.getLastIDs()[0], truncate_tid) self.assertEqual(s.dm.getLastIDs()[0], truncate_tid)
def testConnectionTimeout(self): def testConnectionTimeout(self):
conn = self.getLoopbackConnection() with self.getLoopbackConnection() as conn:
conn.KEEP_ALIVE conn.KEEP_ALIVE
with Patch(conn, KEEP_ALIVE=0):
while conn.connecting:
conn.em.poll(1)
def onTimeout(orig): def onTimeout(orig):
conn.idle() conn.idle()
orig() orig()
with Patch(conn, KEEP_ALIVE=0):
while conn.connecting:
conn.em.poll(1)
with Patch(conn, onTimeout=onTimeout): with Patch(conn, onTimeout=onTimeout):
conn.em.poll(1) conn.em.poll(1)
self.assertFalse(conn.isClosed()) self.assertFalse(conn.isClosed())
......
...@@ -36,12 +36,8 @@ class SSLTests(SSLMixin, test.Test): ...@@ -36,12 +36,8 @@ class SSLTests(SSLMixin, test.Test):
testDeadlockAvoidance = None testDeadlockAvoidance = None
testUndoConflict = testUndoConflictDuringStore = None testUndoConflict = testUndoConflictDuringStore = None
def testAbortConnection(self): def testAbortConnection(self, after_handshake=1):
for after_handshake in 1, 0: with self.getLoopbackConnection() as conn:
try:
conn.reset()
except UnboundLocalError:
conn = self.getLoopbackConnection()
conn.ask(Packets.Ping()) conn.ask(Packets.Ping())
connector = conn.getConnector() connector = conn.getConnector()
del connector.connect_limit[connector.addr] del connector.connect_limit[connector.addr]
...@@ -58,6 +54,9 @@ class SSLTests(SSLMixin, test.Test): ...@@ -58,6 +54,9 @@ class SSLTests(SSLMixin, test.Test):
conn.em.poll(1) conn.em.poll(1)
self.assertIs(conn.getConnector(), None) self.assertIs(conn.getConnector(), None)
def testAbortConnectionBeforeHandshake(self):
self.testAbortConnection(0)
class SSLReplicationTests(SSLMixin, testReplication.ReplicationTests): class SSLReplicationTests(SSLMixin, testReplication.ReplicationTests):
# do not repeat slowest tests with SSL # do not repeat slowest tests with SSL
testBackupNodeLost = testBackupNormalCase = None testBackupNodeLost = testBackupNormalCase = None
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment