Example #1
0
class TestAcceptor:
    def setup_class(self):
        self.receiver = Mock()
        self.acceptor = Acceptor(('127.0.0.1', PORT), self.receiver)

    def test_stream(self):
        assert self.acceptor.stream().getsockname() == ('127.0.0.1', PORT)

    def test_accept(self):
        peer = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        peer.setblocking(0)
        try:
            peer.connect(('127.0.0.1', PORT))
        except Exception as e:
            err = peer.getsockopt(socket.SOL_SOCKET, socket.SO_ERROR)
            if err != 0 and err != errno.EINPROGRESS:
                raise e

        select.select([self.acceptor.stream()], [], [])
        self.acceptor.read_event()

        assert self.receiver.accepted_connection.called
Example #2
0
class TestAcceptor:
    def setup_class(self):
        self.receiver = Mock()
        self.acceptor = Acceptor(('127.0.0.1', PORT), self.receiver)
        
    def test_stream(self):
        assert self.acceptor.stream().getsockname() == ('127.0.0.1', PORT)

    def test_accept(self):
        peer = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        peer.setblocking(0)
        try:
            peer.connect(('127.0.0.1', PORT))
        except Exception as e:
            err = peer.getsockopt(socket.SOL_SOCKET, socket.SO_ERROR)
            if err != 0 and err != errno.EINPROGRESS:
                raise e

        select.select([self.acceptor.stream()],[],[])
        self.acceptor.read_event()

        assert self.receiver.accepted_connection.called