示例#1
0
 def testAcceptorWithPrio(self):
     reactor = CallTrace()
     acceptor = Acceptor(reactor, self.port, lambda sok: None, prio=5)
     client = socket()
     client.connect(('127.0.0.1', self.port))
     acceptor._accept()
     self.assertEquals(5, reactor.calledMethods[0].kwargs['prio'])
 def testAcceptorWithPrio(self):
     reactor = CallTrace()
     acceptor = Acceptor(reactor, self.port, lambda sok: None, prio=5)
     client = socket()
     client.connect(('127.0.0.1', self.port))
     acceptor._accept()
     self.assertEquals(5, reactor.calledMethods[0].kwargs['prio'])
 def testReuseAddress(self):
     reactor = CallTrace()
     acceptor = Acceptor(reactor, self.port, lambda sok: None)
     client = socket()
     client.connect(('127.0.0.1', self.port))
     acceptor._accept()
     acceptor.close()
     acceptor = Acceptor(reactor, self.port, lambda sok: None)
示例#4
0
class DebugPrompt(object):
    def __init__(self, reactor, port, globals):
        self._reactor = reactor
        self._port = port
        self._globals = globals

    def observer_init(self):
        self._acceptor = Acceptor(reactor=self._reactor,
                                  port=self._port,
                                  sinkFactory=self._connect,
                                  bindAddress='127.0.0.1')

    def _connect(self, sok):
        next(self.handle_conversation(sok))

    def close(self):
        self._acceptor.close()

    @identify
    def handle_conversation(self, sok):
        response = PROMPT
        this = yield
        while True:
            self._reactor.addWriter(sok, this.__next__)
            yield
            self._reactor.removeWriter(sok)
            sok.sendall(response)
            self._reactor.addReader(sok, this.__next__)
            yield
            self._reactor.removeReader(sok)
            command = sok.recv(1024).strip()
            if not command:
                sok.close()
                yield
            buff = StringIO()
            try:
                stdout_prev = sys.stdout
                stderr_prev = sys.stderr
                sys.stdout = buff
                sys.stderr = buff
                exec(command, self._globals)
            except:
                print_exc(limit=0)
            finally:
                sys.stdout = stdout_prev
                sys.stderr = stderr_prev
            response = buff.getvalue().encode() + PROMPT
 def testBindAddressCustom(self):
     reactor = CallTrace()
     acceptor = Acceptor(reactor,
                         self.port,
                         lambda sok: None,
                         bindAddress='127.0.0.1',
                         prio=5)
     self.assertEquals(('127.0.0.1', self.port),
                       acceptor._sok.getsockname())
 def testConnect(self):
     reactor = CallTrace()
     acceptor = Acceptor(reactor, self.port, lambda sok: None)
     self.assertEquals('addReader', reactor.calledMethods[0].name)
     acceptCallback = reactor.calledMethods[0].args[1]
     sok = socket()
     sok.connect(('localhost', self.port))
     acceptCallback()
     self.assertEquals(['addReader'], reactor.calledMethodNames())
     reactor.calledMethods[0].args[0].close()
     sok.close()
示例#7
0
 def testStartListening(self):
     reactor = CallTrace()
     port = randint(2**10, 2**16)
     acceptor = Acceptor(reactor, port, None)
     self.assertEquals('addReader', reactor.calledMethods[0].name)
     sok = reactor.calledMethods[0].args[0]
     out = Popen(['netstat', '--numeric', '--listening', '--tcp'], stdout=PIPE, stderr=PIPE).communicate()[0]
     self.assertTrue(str(port) in out, out)
     sok.close()
     callback = reactor.calledMethods[0].args[1]
     self.assertTrue(callable(callback))
示例#8
0
 def testReuseAddress(self):
     reactor = CallTrace()
     acceptor = Acceptor(reactor, self.port, lambda sok: None)
     client = socket()
     client.connect(('127.0.0.1', self.port))
     acceptor._accept()
     acceptor.close()
     acceptor = Acceptor(reactor, self.port, lambda sok: None)
示例#9
0
 def testCreateSink(self):
     reactor = CallTrace('reactor')
     port = randint(2**10, 2**16)
     class sinkFactory(object):
         def __init__(inner, *args, **kwargs): self.args, self.kwargs = args, kwargs
     acceptor = Acceptor(reactor, port, sinkFactory)
     acceptCallback = reactor.calledMethods[0].args[1]
     sok = socket()
     sok.connect(('localhost', port))
     acceptCallback()
     self.assertEquals('addReader', reactor.calledMethods[1].name)
     sink = reactor.calledMethods[1].args[1]
     self.assertEquals(socket, type(self.args[0]))
     reactor.calledMethods[0].args[0].close()
示例#10
0
 def testReadData(self):
     reactor = CallTrace('reactor')
     port = randint(2**10, 2**16)
     class sinkFactory(object):
         def __init__(inner, *args, **kwargs):
             self.args, self.kwargs = args, kwargs
         def next(inner):
             self.next = True
     acceptor = Acceptor(reactor, port, sinkFactory)
     acceptCallback = reactor.calledMethods[0].args[1]
     sok = socket()
     sok.connect(('localhost', port))
     acceptCallback()
     sink = reactor.calledMethods[1].args[1]
     self.next = False
     sink.next()
     self.assertTrue(self.next)
     reactor.calledMethods[0].args[0].close()
 def testBindAddress_DefaultsTo_0_0_0_0(self):
     reactor = CallTrace()
     acceptor = Acceptor(reactor, self.port, lambda sok: None, prio=5)
     self.assertEquals(('0.0.0.0', self.port), acceptor._sok.getsockname())
示例#12
0
 def observer_init(self):
     self._acceptor = Acceptor(reactor=self._reactor,
                               port=self._port,
                               sinkFactory=self._connect,
                               bindAddress='127.0.0.1')