name : stdio_test_halfclose.py # -*- test-case-name: twisted.test.test_stdio.StandardInputOutputTests.test_readConnectionLost -*- # Copyright (c) Twisted Matrix Laboratories. # See LICENSE for details. """ Main program for the child process run by L{twisted.test.test_stdio.StandardInputOutputTests.test_readConnectionLost} to test that IHalfCloseableProtocol.readConnectionLost works for process transports. """ import sys from zope.interface import implementer from twisted.internet import protocol, stdio from twisted.internet.interfaces import IHalfCloseableProtocol from twisted.python import log, reflect @implementer(IHalfCloseableProtocol) class HalfCloseProtocol(protocol.Protocol): """ A protocol to hook up to stdio and observe its transport being half-closed. If all goes as expected, C{exitCode} will be set to C{0}; otherwise it will be set to C{1} to indicate failure. """ exitCode = None def connectionMade(self): """ Signal the parent process that we're ready. """ self.transport.write(b"x") def readConnectionLost(self): """ This is the desired event. Once it has happened, stop the reactor so the process will exit. """ self.exitCode = 0 reactor.stop() def connectionLost(self, reason): """ This may only be invoked after C{readConnectionLost}. If it happens otherwise, mark it as an error and shut down. """ if self.exitCode is None: self.exitCode = 1 log.err(reason, "Unexpected call to connectionLost") reactor.stop() def writeConnectionLost(self): # IHalfCloseableProtocol.writeConnectionLost pass if __name__ == "__main__": reflect.namedAny(sys.argv[1]).install() log.startLogging(open(sys.argv[2], "wb")) from twisted.internet import reactor halfCloseProtocol = HalfCloseProtocol() stdio.StandardIO(halfCloseProtocol) reactor.run() # type: ignore[attr-defined] sys.exit(halfCloseProtocol.exitCode)