Move test_*.py to examples dir

examples/test_client.py

@@ -0,0 +1,95 @@
+import logging
+import os
+import struct
+import time
+from asyncio import get_event_loop, gather, ensure_future
+from base64 import b64decode
+
+import yaml
+from colorlog import ColoredFormatter
+from nacl.signing import SigningKey
+
+from ssb.muxrpc import MuxRPCAPI, MuxRPCAPIException
+from ssb.packet_stream import PSClient, PSMessageType
+
+
+api = MuxRPCAPI()
+
+
+@api.define('createHistoryStream')
+def create_history_stream(connection, msg):
+    print('create_history_stream', msg)
+    # msg = PSMessage(PSMessageType.JSON, True, stream=True, end_err=True, req=-req)
+    # connection.write(msg)
+
+
+@api.define('blobs.createWants')
+def create_wants(connection, msg):
+    print('create_wants', msg)
+
+
+async def test_client():
+    async for msg in api.call('createHistoryStream', [{
+        'id': "@1+Iwm79DKvVBqYKFkhT6fWRbAVvNNVH4F2BSxwhYmx8=.ed25519",
+        'seq': 1,
+        'live': False,
+        'keys': False
+    }], 'source'):
+        print('> RESPONSE:', msg)
+
+    try:
+        print('> RESPONSE:', await api.call('whoami', [], 'sync'))
+    except MuxRPCAPIException as e:
+        print(e)
+
+    handler = api.call('gossip.ping', [], 'duplex')
+    handler.send(struct.pack('l', int(time.time() * 1000)), msg_type=PSMessageType.BUFFER)
+    async for msg in handler:
+        print('> RESPONSE:', msg)
+        handler.send(True, end=True)
+        break
+
+    async for data in api.call('blobs.get', ['&/6q7JOKythgnnzoBI5xxvotCr5HeFkAIZSAuqHiZfLw=.sha256'], 'source'):
+        if data.type.name == 'BUFFER':
+            with open('./funny_img.png', 'wb') as f:
+                f.write(data.data)
+
+
+async def _main(packet_stream):
+    await packet_stream.connect()
+    api.add_connection(packet_stream)
+    await gather(ensure_future(api), test_client())
+
+
+def main():
+    # create console handler and set level to debug
+    ch = logging.StreamHandler()
+    ch.setLevel(logging.INFO)
+
+    # create formatter
+    formatter = ColoredFormatter('%(log_color)s%(levelname)s%(reset)s:%(bold_white)s%(name)s%(reset)s - '
+                                 '%(cyan)s%(message)s%(reset)s')
+
+    # add formatter to ch
+    ch.setFormatter(formatter)
+
+    # add ch to logger
+    logger = logging.getLogger('packet_stream')
+    logger.setLevel(logging.INFO)
+    logger.addHandler(ch)
+
+    with open(os.path.expanduser('~/.ssb/secret')) as f:
+        config = yaml.load(f)
+
+    server_pub_key = b64decode(config['public'][:-8])
+    server_prv_key = b64decode(config['private'][:-8])
+    sign = SigningKey(server_prv_key[:32])
+
+    loop = get_event_loop()
+    packet_stream = PSClient('127.0.0.1', 8008, sign, server_pub_key, loop=loop)
+    loop.run_until_complete(_main(packet_stream))
+    loop.close()
+
+
+if __name__ == '__main__':
+    main()

examples/test_server.py

@@ -0,0 +1,45 @@
+import logging
+import os
+from asyncio import get_event_loop, ensure_future
+from base64 import b64decode
+
+import yaml
+from colorlog import ColoredFormatter
+from nacl.signing import SigningKey
+
+from ssb.packet_stream import PSServer
+
+
+with open(os.path.expanduser('~/.ssb/secret')) as f:
+    config = yaml.load(f)
+
+
+async def on_connect():
+    ensure_future(packet_stream, loop=loop)
+
+# create console handler and set level to debug
+ch = logging.StreamHandler()
+ch.setLevel(logging.DEBUG)
+
+# create formatter
+formatter = ColoredFormatter('%(log_color)s%(levelname)s%(reset)s:%(bold_white)s%(name)s%(reset)s - '
+                             '%(cyan)s%(message)s%(reset)s')
+
+# add formatter to ch
+ch.setFormatter(formatter)
+
+# add ch to logger
+logger = logging.getLogger('packet_stream')
+logger.setLevel(logging.DEBUG)
+logger.addHandler(ch)
+
+
+loop = get_event_loop()
+
+server_keypair = SigningKey(b64decode(config['private'][:-8])[:32])
+packet_stream = PSServer('127.0.0.1', 8008, server_keypair, loop=loop)
+packet_stream.on_connect(on_connect)
+packet_stream.listen()
+
+loop.run_forever()
+loop.close()