Added testing of streaming multiple blocks from gossip peer
This commit is contained in:
parent
4caee50ce7
commit
6a6460ef31
78
tests/gossip-unittests/test_server_diffuse_mny.py
Normal file
78
tests/gossip-unittests/test_server_diffuse_mny.py
Normal file
@ -0,0 +1,78 @@
|
||||
import os, uuid
|
||||
TEST_DIR = 'testdata/%s-%s' % (str(uuid.uuid4())[:12], os.path.basename(__file__)) + '/'
|
||||
print("Test directory:", TEST_DIR)
|
||||
os.environ["ONIONR_HOME"] = TEST_DIR
|
||||
|
||||
from threading import Thread
|
||||
import asyncio
|
||||
import unittest
|
||||
import sys
|
||||
sys.path.append(".")
|
||||
sys.path.append("src/")
|
||||
|
||||
from ordered_set import OrderedSet
|
||||
import onionrblocks
|
||||
|
||||
import blockdb
|
||||
from gossip.server import gossip_server
|
||||
from filepaths import gossip_server_socket_file
|
||||
|
||||
|
||||
BLOCK_MAX_SIZE = 1024 * 2000
|
||||
BLOCK_MAX_SIZE_LEN = len(str(BLOCK_MAX_SIZE))
|
||||
BLOCK_ID_SIZE = 128
|
||||
BLOCK_STREAM_OFFSET_DIGITS = 8
|
||||
|
||||
|
||||
class OnionrDiffuseMany(unittest.TestCase):
|
||||
|
||||
|
||||
def test_many(self):
|
||||
|
||||
Thread(target=gossip_server, daemon=True).start()
|
||||
blocks = []
|
||||
for _ in range(10):
|
||||
bl = onionrblocks.blockcreator.create_anonvdf_block(
|
||||
b"my test block" + os.urandom(16), b"txt", 2800)
|
||||
blockdb.add_block_to_db(bl)
|
||||
blocks.append(bl)
|
||||
|
||||
async def diffuse_client():
|
||||
reader, writer = await asyncio.open_unix_connection(
|
||||
gossip_server_socket_file)
|
||||
|
||||
# tell we want to stream blocks
|
||||
writer.write(int(4).to_bytes(1, 'big'))
|
||||
await writer.drain()
|
||||
|
||||
# tell timestamp offset
|
||||
writer.write('0'.zfill(BLOCK_STREAM_OFFSET_DIGITS).encode('utf-8'))
|
||||
await writer.drain()
|
||||
|
||||
# Read blocks from offset
|
||||
for i in range(10):
|
||||
bl_id = await reader.readexactly(BLOCK_ID_SIZE)
|
||||
for bl in blocks:
|
||||
if bl.id == bl_id:
|
||||
break
|
||||
|
||||
# tell we want the block
|
||||
writer.write(int(1).to_bytes(1, 'big'))
|
||||
await writer.drain()
|
||||
|
||||
# check block size
|
||||
self.assertEqual(
|
||||
len(bl.raw),
|
||||
int((await reader.readexactly(BLOCK_MAX_SIZE_LEN)).decode('utf-8')))
|
||||
|
||||
self.assertEqual(bl.raw, await reader.readexactly(len(bl.raw)))
|
||||
writer.write(int(1).to_bytes(1, 'big'))
|
||||
await writer.drain()
|
||||
|
||||
writer.write(int(0).to_bytes(1, 'big'))
|
||||
await writer.drain()
|
||||
|
||||
|
||||
asyncio.run(diffuse_client())
|
||||
|
||||
unittest.main()
|
@ -26,6 +26,7 @@ BLOCK_STREAM_OFFSET_DIGITS = 8
|
||||
|
||||
class OnionrServerDiffuseTest(unittest.TestCase):
|
||||
|
||||
|
||||
def test_one_block(self):
|
||||
|
||||
Thread(target=gossip_server, daemon=True).start()
|
||||
@ -45,6 +46,17 @@ class OnionrServerDiffuseTest(unittest.TestCase):
|
||||
|
||||
self.assertEqual(bl.id, await reader.readexactly(BLOCK_ID_SIZE))
|
||||
|
||||
# we want the block
|
||||
writer.write(int(1).to_bytes(1, 'big'))
|
||||
assert writer.drain()
|
||||
|
||||
# check block size
|
||||
self.assertEqual(
|
||||
len(bl.raw),
|
||||
int((await reader.readexactly(BLOCK_MAX_SIZE_LEN)).decode('utf-8')))
|
||||
|
||||
self.assertEqual(bl.raw, await reader.readexactly(len(bl.raw)))
|
||||
|
||||
|
||||
asyncio.run(diffuse_client())
|
||||
|
Loading…
Reference in New Issue
Block a user