def returnConnected(server, client): """Take two Protocol instances and connect them.""" cio = BytesIO() sio = BytesIO() client.makeConnection(FileWrapper(cio)) server.makeConnection(FileWrapper(sio)) pump = IOPump(client, server, cio, sio) # Challenge-response authentication: pump.flush() # Uh... pump.flush() return pump
def main(): """ Main function to be run if __name__ == "__main__". """ config = WorkerOptions() config.parseOptions() from twisted.trial._dist.worker import WorkerProtocol workerProtocol = WorkerProtocol(config['force-gc']) protocolIn = os.fdopen(_WORKER_AMP_STDIN) protocolOut = os.fdopen(_WORKER_AMP_STDOUT, 'w') workerProtocol.makeConnection(FileWrapper(protocolOut)) observer = WorkerLogObserver(workerProtocol) startLoggingWithObserver(observer.emit, False) while True: try: r = protocolIn.read(1) except IOError, e: if e.args[0] == errno.EINTR: sys.exc_clear() continue else: raise if r == '': break else: workerProtocol.dataReceived(r) protocolOut.flush() sys.stdout.flush() sys.stderr.flush()
def test_write(self): """ L{twisted.internet.protocol.FileWrapper.write} """ wrapper = FileWrapper(BytesIO()) wrapper.write(b"test1") self.assertEqual(wrapper.file.getvalue(), b"test1") wrapper = FileWrapper(BytesIO()) # BytesIO() cannot accept unicode, so this will # cause an exception to be thrown which will be # handled by FileWrapper.handle_exception(). wrapper.write("stuff") self.assertNotEqual(wrapper.file.getvalue(), "stuff")
def test_writeSequence(self): """ L{twisted.internet.protocol.FileWrapper.writeSequence} """ wrapper = FileWrapper(BytesIO()) wrapper.writeSequence([b"test1", b"test2"]) self.assertEqual(wrapper.file.getvalue(), b"test1test2") wrapper = FileWrapper(BytesIO()) # In Python 3, b"".join([u"a", u"b"]) will raise a TypeError self.assertRaises(TypeError, wrapper.writeSequence, ["test3", "test4"])
def main(_fdopen=os.fdopen): """ Main function to be run if __name__ == "__main__". @param _fdopen: If specified, the function to use in place of C{os.fdopen}. @param _fdopen: C{callable} """ config = WorkerOptions() config.parseOptions() from twisted.trial._dist.worker import WorkerProtocol workerProtocol = WorkerProtocol(config['force-gc']) protocolIn = _fdopen(_WORKER_AMP_STDIN) protocolOut = _fdopen(_WORKER_AMP_STDOUT, 'w') workerProtocol.makeConnection(FileWrapper(protocolOut)) observer = WorkerLogObserver(workerProtocol) startLoggingWithObserver(observer.emit, False) while True: try: r = protocolIn.read(1) if isinstance(r, unicode): r = r.encode("utf-8") except IOError as e: if e.args[0] == errno.EINTR: if sys.version_info < (3, 0): sys.exc_clear() continue else: raise if r == b'': break else: workerProtocol.dataReceived(r) protocolOut.flush() sys.stdout.flush() sys.stderr.flush() if config.tracer: sys.settrace(None) results = config.tracer.results() results.write_results(show_missing=True, summary=False, coverdir=config.coverdir().path)
def test_writeSequence(self): """ L{twisted.internet.protocol.FileWrapper.writeSequence} """ wrapper = FileWrapper(BytesIO()) wrapper.writeSequence([b"test1", b"test2"]) self.assertEqual(wrapper.file.getvalue(), b"test1test2") wrapper = FileWrapper(BytesIO()) if _PY3: # In Python 3, b"".join([u"a", u"b"]) will raise a TypeError self.assertRaises(TypeError, wrapper.writeSequence, [u"test3", u"test4"]) else: # In Python 2, b"".join([u"a", u"b"]) # will give u"ab", but writing unicode to BytesIO # will throw an exception which will be caught # and ignored by FileWrapper.handle_exception() wrapper.writeSequence([u"test3", u"test4"]) self.assertTrue(len(wrapper.file.getvalue()) == 0)
def test_write(self): """ L{twisted.internet.protocol.FileWrapper.write} """ wrapper = FileWrapper(BytesIO()) wrapper.write(b"test1") self.assertEqual(wrapper.file.getvalue(), b"test1") wrapper = FileWrapper(BytesIO()) # BytesIO() cannot accept unicode, so this will # cause an exception to be thrown which will be # handled by FileWrapper.handle_exception(). wrapper.write(u"stuff") self.assertNotEqual(wrapper.file.getvalue(), u"stuff")
def test_writeSequence(self): """ L{twisted.internet.protocol.FileWrapper.writeSequence} """ wrapper = FileWrapper(BytesIO()) wrapper.writeSequence([b"test1", b"test2"]) self.assertEqual(wrapper.file.getvalue(), b"test1test2") wrapper = FileWrapper(BytesIO()) if _PY3: # In Python 3, b"".join([u"a", u"b"]) will raise a TypeError self.assertRaises(TypeError, wrapper.writeSequence, [u"test3", u"test4"]) else: # In Python 2, b"".join([u"a", u"b"]) # will give u"ab", but writing unicode to BytesIO # will throw an exception which will be caught # and ignored by FileWrapper.handle_exception() wrapper.writeSequence([u"test3", u"test4"]) self.assertTrue(len(wrapper.file.getvalue()) == 0)
'<!DOCTYPE' and '>' as an argument. """ print '!DOCTYPE', repr(doctype) def gotTagEnd(self, name): '''Encountered closing tag Default behaviour is to print.''' print 'end', name if __name__ == '__main__': from cStringIO import StringIO testDocument = ''' <!DOCTYPE ignore all this shit, hah its malformed!!!!@$> <?xml version="suck it"?> <foo> A <bar /> <baz boz="buz">boz &zop;</baz> <![CDATA[ foo bar baz ]]> </foo> ''' x = XMLParser() x.makeConnection(FileWrapper(StringIO())) # fn = "/home/glyph/Projects/Twisted/doc/howto/ipc10paper.html" fn = "/home/glyph/gruesome.xml" # testDocument = open(fn).read() x.dataReceived(testDocument)
def setUp(self): self.f = StringIO() self.t = FileWrapper(self.f) self.vb = VumiBotProtocol(self.nick, [self.channel], self) self.vb.makeConnection(self.t) self.recvd_messages = []
def __init__(self, stringio, q2qhost, q2qpeer): FileWrapper.__init__(self, stringio) self.q2qhost = q2qhost self.q2qpeer = q2qpeer
def make_dummy_http_request(self): """Make a dummy HTTP request for tests.""" transport = FileWrapper(io.BytesIO()) channel = http.HTTPChannel() channel.makeConnection(transport) return http.Request(channel, True)
def __init__(self, stringio, q2qhost, q2qpeer): FileWrapper.__init__(self, stringio) self.q2qhost = q2qhost self.q2qpeer = q2qpeer