source: geniuswebcore/test/geniusweb/simplerunner/ClassPathConnectionFactoryTest.py@ 78

import time
import unittest

from uri.uri import URI

from geniusweb.references.PartyRef import PartyRef
from geniusweb.simplerunner.ClassPathConnectionFactory import ClassPathConnectionFactory, \
        ConnectionPair


class ClassPathConnectionFactoryTest(unittest.TestCase):
        factory=ClassPathConnectionFactory()
        PARTYREF = PartyRef(URI("pythonpath:test.testparty.TestParty.TestParty"))


        def testWrongURI(self): 
                self.assertRaises(ValueError, 
                                lambda:self.factory.connect(PartyRef(URI("http://blabla"))))


        def testNullPath(self):
                # bad because classpath should not start with //
                self.assertRaises(ValueError, lambda:self.factory.connect(PartyRef(URI(
                        "pythonpath://some.class.path"))))

        def testUnknownParty(self):
                self.assertRaises(ValueError, lambda:self.factory.connect(PartyRef(URI(
                        "pythonpath:blabla.bla"))))

        def testParty(self) :
                conn = self.factory.connect(self.PARTYREF)
                self.assertIsNotNone(conn)
                conn.close()

        def testConnectionPairProtocolClosesNormally(self):
                cpair = ConnectionPair(self.PARTYREF);
                self.assertEquals(2, len(cpair.getOpenConnections()))
                cpair.getProtocolToPartyConn().close()
                time.sleep(0.1)
                self.assertEqual(0, len(cpair.getOpenConnections()))
                        
        def testConnectionPairProtocolClosesWithExc(self):
                cpair = ConnectionPair(self.PARTYREF)
                self.assertEqual(2, len(cpair.getOpenConnections()))
                cpair.getProtocolToPartyConn().setError(ValueError("test"))
                time.sleep(0.1)
                self.assertEqual(0, len(cpair.getOpenConnections()))

        def testConnectionPairPartyClosesNormally(self):
                cpair = ConnectionPair(self.PARTYREF)
                self.assertEqual(2, len(cpair.getOpenConnections()))
                cpair._getPartyToProtocolConn().close()
                time.sleep(0.1)

                self.assertEqual(0, len(cpair.getOpenConnections()))

        def testConnectionPairPartyClosesWithExc(self):
                cpair = ConnectionPair(self.PARTYREF)
                self.assertEqual(2, len(cpair.getOpenConnections()))
                cpair._getPartyToProtocolConn().setError(ValueError("test"))
                time.sleep(0.1)
                self.assertEqual(0, len(cpair.getOpenConnections()))