Update unit tests after recent leak fix

git-svn-id: https://svn.erp5.org/repos/neo/trunk@2827 71dcc9de-d417-0410-9af5-da40c76e7ee4

neo/tests/master/testClientHandler.py

@@ -167,6 +167,7 @@ class MasterClientHandlerTests(NeoUnitTestBase):
         client_uuid = self.identifyToMasterNode(node_type=NodeTypes.CLIENT,
                                                 port = self.client_port)
         conn = self.getFakeConnection(client_uuid, self.client_address)
+        self.app.listening_conn = object() # mark as running
         lptid = self.app.pt.getID()
         self.assertEqual(self.app.nm.getByUUID(client_uuid).getState(),
                 NodeStates.RUNNING)

neo/tests/storage/testInitializationHandler.py

@@ -53,6 +53,7 @@ class StorageInitializationHandlerTests(NeoUnitTestBase):
 
     def test_03_connectionClosed(self):
         conn = self.getClientConnection()
+        self.app.listening_conn = object() # mark as running
         self.assertRaises(PrimaryFailure, self.verification.connectionClosed, conn,)
         # nothing happens
         self.checkNoPacketSent(conn)

neo/tests/storage/testMasterHandler.py

@@ -60,6 +60,7 @@ class StorageMasterHandlerTests(NeoUnitTestBase):
     def test_07_connectionClosed2(self):
         # primary has closed the connection
         conn = self.getMasterConnection()
+        self.app.listening_conn = object() # mark as running
         self.assertRaises(PrimaryFailure, self.operation.connectionClosed, conn)
         self.checkNoPacketSent(conn)
 

neo/tests/storage/testVerificationHandler.py

@@ -58,6 +58,7 @@ class StorageVerificationHandlerTests(NeoUnitTestBase):
     # Tests
     def test_03_connectionClosed(self):
         conn = self.getClientConnection()
+        self.app.listening_conn = object() # mark as running
         self.assertRaises(PrimaryFailure, self.verification.connectionClosed, conn,)
         # nothing happens
         self.checkNoPacketSent(conn)