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)
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