Python StreamServer Examples

Example #1

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_echo_data(self):
     # Echo a bunch of data and ensure it is echoed identically
     server = StreamServer(echo_handler)
     server.listen(('127.0.0.1', 0))
     client = StreamClient()
     client.connect(server.addresses[0])
     md1 = hashlib.sha256()
     md2 = hashlib.sha256()
     def produce():
         for i in range(1000):
             chunk = os.urandom(1024)
             client.write(chunk)
             md1.update(chunk)
         client.write_eof()
     def consume():
         while True:
             buf = client.read1(1024)
             if not buf:
                 break
             md2.update(buf)
     f1 = gruvi.spawn(produce)
     f2 = gruvi.spawn(consume)
     f1.join(); f2.join()
     self.assertEqual(md1.digest(), md2.digest())
     server.close()
     client.close()

Example #2

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_read_timeout(self):
     server = StreamServer(echo_handler)
     server.listen(self.pipename())
     client = StreamClient(timeout=0.01)
     client.connect(server.addresses[0])
     self.assertRaises(Timeout, client.readline)
     server.close()
     client.close()

Example #3

File: test_stream.py Project: geertj/gruvi

 def test_echo_tcp_ssl(self):
     server = StreamServer(echo_handler)
     server.listen(('127.0.0.1', 0), **self.ssl_s_args)
     client = StreamClient()
     client.connect(server.addresses[0], **self.ssl_c_args)
     client.write(b'foo\n')
     self.assertEqual(client.readline(), b'foo\n')
     server.close()
     client.close()

Example #4

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_echo_tcp(self):
     server = StreamServer(echo_handler)
     server.listen(('127.0.0.1', 0))
     client = StreamClient()
     client.connect(server.addresses[0])
     client.write(b'foo\n')
     client.write_eof()
     self.assertEqual(client.readline(), b'foo\n')
     server.close()
     client.close()

Example #5

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_echo_tcp_ssl(self):
     server = StreamServer(echo_handler)
     context = self.get_ssl_context()
     server.listen(('127.0.0.1', 0), ssl=context)
     client = StreamClient()
     client.connect(server.addresses[0], ssl=context)
     client.write(b'foo\n')
     self.assertEqual(client.readline(), b'foo\n')
     server.close()
     client.close()

Example #6

File: test_pyuv.py Project: cocagne/gruvi

 def test_tcp(self):
     server = StreamServer(echo_handler)
     server.listen(('localhost', 0))
     addr = server.transport.getsockname()
     client = StreamClient()
     client.connect(addr)
     buf = b'x' * 1024
     client.write(buf)
     result = client.read(1024)
     assert result == buf
     client.close()

Example #7

File: test_pyuv.py Project: cocagne/gruvi

 def test_pipe(self):
     server = StreamServer(echo_handler)
     path = self.pipename('temp.sock')
     server.listen(path)
     client = StreamClient()
     client.connect(path)
     buf = b'x' * 1024
     client.write(buf)
     result = client.read(1024)
     assert result == buf
     client.close()

Example #8

File: test_pyuv.py Project: cocagne/gruvi

 def test_tcp(self):
     server = StreamServer(echo_handler)
     server.listen(('localhost', 0))
     addr = server.transport.getsockname()
     client = StreamClient()
     client.connect(addr)
     buf = b'x' * 1024
     client.write(buf)
     result = client.read(1024)
     assert result == buf
     client.close()

Example #9

File: test_pyuv.py Project: cocagne/gruvi

 def test_pipe(self):
     server = StreamServer(echo_handler)
     path = self.pipename('temp.sock')
     server.listen(path)
     client = StreamClient()
     client.connect(path)
     buf = b'x' * 1024
     client.write(buf)
     result = client.read(1024)
     assert result == buf
     client.close()

Example #10

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_echo_pipe(self):
     server = StreamServer(echo_handler)
     server.listen(self.pipename())
     client = StreamClient()
     client.connect(server.addresses[0])
     client.write(b'foo\n')
     client.write_eof()
     self.assertEqual(client.readline(), b'foo\n')
     self.assertEqual(client.readline(), b'')
     server.close()
     client.close()

Example #11

File: test_ssl.py Project: cocagne/gruvi

 def test_stream(self):
     def echo_handler(stream, protocol, client):
         while True:
             buf = stream.read(4096)
             if not buf:
                 break
             nbytes = stream.write(buf)
     server = StreamServer(echo_handler)
     server.listen(('localhost', 0), ssl=True,
                   keyfile=self.certname, certfile=self.certname)
     addr = server.transport.getsockname()
     client = StreamClient()
     client.connect(addr, ssl=True)
     buf = b'x' * 1024
     client.write(buf)
     result = b''
     while len(result) != 1024:
         result += client.read(1024)
     assert result == buf
     client.close()
     server.close()

Example #12

File: test_stream.py Project: Ed-von-Schleck/gruvi

    def test_echo_data(self):
        # Echo a bunch of data and ensure it is echoed identically
        server = StreamServer(echo_handler)
        server.listen(('127.0.0.1', 0))
        client = StreamClient()
        client.connect(server.addresses[0])
        md1 = hashlib.sha256()
        md2 = hashlib.sha256()

        def produce():
            for i in range(1000):
                chunk = os.urandom(1024)
                client.write(chunk)
                md1.update(chunk)
            client.write_eof()

        def consume():
            while True:
                buf = client.read1(1024)
                if not buf:
                    break
                md2.update(buf)

        f1 = gruvi.spawn(produce)
        f2 = gruvi.spawn(consume)
        f1.join()
        f2.join()
        self.assertEqual(md1.digest(), md2.digest())
        server.close()
        client.close()

Example #13

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_read_timeout(self):
     server = StreamServer(echo_handler)
     server.listen(self.pipename())
     client = StreamClient(timeout=0.01)
     client.connect(server.addresses[0])
     self.assertRaises(Timeout, client.readline)
     server.close()
     client.close()

Example #14

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_echo_tcp(self):
     server = StreamServer(echo_handler)
     server.listen(('127.0.0.1', 0))
     client = StreamClient()
     client.connect(server.addresses[0])
     client.write(b'foo\n')
     client.write_eof()
     self.assertEqual(client.readline(), b'foo\n')
     server.close()
     client.close()

Example #15

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_echo_tcp_ssl(self):
     server = StreamServer(echo_handler)
     context = self.get_ssl_context()
     server.listen(('127.0.0.1', 0), ssl=context)
     client = StreamClient()
     client.connect(server.addresses[0], ssl=context)
     client.write(b'foo\n')
     self.assertEqual(client.readline(), b'foo\n')
     server.close()
     client.close()

Example #16

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_echo_pipe(self):
     server = StreamServer(echo_handler)
     server.listen(self.pipename())
     client = StreamClient()
     client.connect(server.addresses[0])
     client.write(b'foo\n')
     client.write_eof()
     self.assertEqual(client.readline(), b'foo\n')
     self.assertEqual(client.readline(), b'')
     server.close()
     client.close()

Example #17

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_connection_limit(self):
     server = StreamServer(echo_handler)
     server.listen(('127.0.0.1', 0))
     addr = server.addresses[0]
     server.max_connections = 10
     clients = []
     try:
         for i in range(15):
             client = StreamClient()
             client.connect(addr)
             client.write(b'foo\n')
             buf = client.readline()
             if buf == b'':  # conneciton closed: EOF
                 client.close()
             clients.append(client)
     except TransportError:
         pass
     self.assertLessEqual(len(server.connections), server.max_connections)
     for client in clients:
         client.close()
     server.close()

Example #18

File: test_stream.py Project: Ed-von-Schleck/gruvi

 def test_connection_limit(self):
     server = StreamServer(echo_handler)
     server.listen(('127.0.0.1', 0))
     addr = server.addresses[0]
     server.max_connections = 10
     clients = []
     try:
         for i in range(15):
             client = StreamClient()
             client.connect(addr)
             client.write(b'foo\n')
             buf = client.readline()
             if buf == b'':  # conneciton closed: EOF
                 client.close()
             clients.append(client)
     except TransportError:
         pass
     self.assertLessEqual(len(server.connections), server.max_connections)
     for client in clients:
         client.close()
     server.close()

Example #19

File: test_ssl.py Project: cocagne/gruvi

    def test_stream(self):
        def echo_handler(stream, protocol, client):
            while True:
                buf = stream.read(4096)
                if not buf:
                    break
                nbytes = stream.write(buf)

        server = StreamServer(echo_handler)
        server.listen(('localhost', 0),
                      ssl=True,
                      keyfile=self.certname,
                      certfile=self.certname)
        addr = server.transport.getsockname()
        client = StreamClient()
        client.connect(addr, ssl=True)
        buf = b'x' * 1024
        client.write(buf)
        result = b''
        while len(result) != 1024:
            result += client.read(1024)
        assert result == buf
        client.close()
        server.close()

Example #20

import logging
import argparse

from gruvi import get_hub, util
from gruvi.stream import StreamServer

logging.basicConfig()

parser = argparse.ArgumentParser()
parser.add_argument('port', type=int)
args = parser.parse_args()


def echo_handler(stream, protocol, client):
    peer = client.getpeername()
    print('Connection from {0}'.format(util.saddr(peer)))
    while True:
        buf = stream.read(4096)
        if not buf:
            break
        stream.write(buf)
    print('Connection closed')


server = StreamServer(echo_handler)
server.listen(('0.0.0.0', args.port))

hub = get_hub()
hub.switch(interrupt=True)

Example #21

File: echoserver1.py Project: cocagne/gruvi

# Example: echo server, using StreamServer

import logging
import argparse

from gruvi import get_hub, util
from gruvi.stream import StreamServer

logging.basicConfig()

parser = argparse.ArgumentParser()
parser.add_argument('port', type=int)
args = parser.parse_args()

def echo_handler(stream, protocol, client):
    peer = client.getpeername()
    print('Connection from {0}'.format(util.saddr(peer)))
    while True:
        buf = stream.read(4096)
        if not buf:
            break
        stream.write(buf)
    print('Connection closed')

server = StreamServer(echo_handler)
server.listen(('0.0.0.0', args.port))

hub = get_hub()
hub.switch(interrupt=True)