예제 #1
0
class PeotryTestCase(unittest.TestCase):

    def setUp(self):
        self.wf = "../welcome.txt"
        self.send_data = open(self.wf).read()
        self.service = PoetryService(self.wf)
        factory = PoetryFactory(self.service)
        from twisted.internet import reactor
        self.port = reactor.listenTCP(0, factory, interface="127.0.0.1")
        self.portnum = self.port.getHost().port

    def tearDown(self):
        port, self.port = self.port, None
        return port.stopListening()

    def test_client(self):
        """The correct poem is returned by get_poetry."""
        self.service.startService()
        d = get_poetry('127.0.0.1', self.portnum)

        def got_poem(poem):
            self.assertEquals(poem, self.send_data)

        d.addCallback(got_poem)

        return d

    def test_failure(self):
        """The correct failure is returned by get_poetry when
        connecting to a port with no server."""
        d = get_poetry('127.0.0.1', 0)
        return self.assertFailure(d, ConnectError)
예제 #2
0
 def setUp(self):
     self.wf = "../welcome.txt"
     self.send_data = open(self.wf).read()
     self.service = PoetryService(self.wf)
     factory = PoetryFactory(self.service)
     from twisted.internet import reactor
     self.port = reactor.listenTCP(0, factory, interface="127.0.0.1")
     self.portnum = self.port.getHost().port