Onionr/tests/test_blockio.py

121 lines
3.8 KiB
Python
Raw Normal View History

2021-01-17 19:44:57 +00:00
#!/usr/bin/env python3
import sys, os
2021-01-25 22:16:51 +00:00
2021-01-17 19:44:57 +00:00
sys.path.append(".")
sys.path.append("src/")
import uuid
TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
print("Test directory:", TEST_DIR)
os.environ["ONIONR_HOME"] = TEST_DIR
import unittest
import time
2021-01-25 22:16:51 +00:00
from utils import identifyhome, createdirs
2021-01-17 19:44:57 +00:00
from onionrsetup import setup_config, setup_default_plugins
createdirs.create_dirs()
setup_config()
setup_default_plugins()
import kasten
from onionrblocks.generators import anonvdf
2021-01-24 23:29:02 +00:00
from onionrblocks import blockcreator
2021-01-17 19:44:57 +00:00
from utils import identifyhome
import safedb
import blockio
2021-01-25 22:16:51 +00:00
from blockio.clean.cleanblocklistentries import clean_block_list_entries
2021-01-17 19:44:57 +00:00
2021-01-17 19:44:57 +00:00
def _remove_db(path):
try:
os.remove(path)
except FileNotFoundError:
pass
2021-01-17 19:44:57 +00:00
class TestBlockIO(unittest.TestCase):
def test_list_all_blocks(self):
db_file = identifyhome.identify_home() + 'test.db'
db = safedb.SafeDB(db_file)
expected_l = []
for i in range(5):
bl = blockcreator.create_anonvdf_block(b"hello" + int(i).to_bytes(1, "big"), b"txt" + int(i).to_bytes(1, "big"), 5)
blockio.store_block(bl, db)
expected_l.append(bl.id)
l = blockio.load.list_all_blocks(db)
self.assertEqual(len(l), 5)
for i in l:
self.assertIn(bytes(i), expected_l)
db.close()
_remove_db(db_file)
2021-01-25 22:16:51 +00:00
def test_clean_blocklist_entries(self):
db_file = identifyhome.identify_home() + 'test.db'
db = safedb.SafeDB(db_file)
bl = blockcreator.create_anonvdf_block(b"hello" + int(10).to_bytes(1, "big"), b"txt", 5)
blockio.store_block(bl, db)
db.db_conn[b'bl-txt'] = b''
clean_block_list_entries(db)
self.assertRaises(KeyError, db.get, 'bl-txt')
db.close()
_remove_db(db_file)
def test_clean_expired(self):
db_file = identifyhome.identify_home() + 'test.db'
db = safedb.SafeDB(db_file)
for i in range(3):
bl = blockcreator.create_anonvdf_block(b"hello" + int(i).to_bytes(1, "big"), b"txt", 5)
blockio.store_block(bl, db)
blockio.clean_expired_blocks(db)
time.sleep(1)
self.assertEqual(len(list(blockio.list_blocks_by_type("txt", db))), 3)
2021-01-25 22:16:51 +00:00
time.sleep(10.1)
blockio.clean_expired_blocks(db)
self.assertEqual(len(db.db_conn[b'bl-txt']), 0)
db.close()
_remove_db(db_file)
2021-01-17 19:44:57 +00:00
def test_store_block(self):
packed = kasten.generator.pack.pack(b"test", "tst")
bl: kasten.Kasten = anonvdf.AnonVDFGenerator.generate(packed, rounds=1000)
db_file = identifyhome.identify_home() + 'test.db'
db = safedb.SafeDB(db_file)
blockio.store_block(bl, db)
db.close()
_remove_db(db_file)
def test_store_dupe(self):
db_file = identifyhome.identify_home() + 'test.db'
_remove_db(db_file)
2021-01-17 19:44:57 +00:00
packed = kasten.generator.pack.pack(b"test", "tst")
bl: kasten.Kasten = anonvdf.AnonVDFGenerator.generate(packed, rounds=1000)
db = safedb.SafeDB(db_file)
blockio.store_block(bl, db)
self.assertRaises(ValueError, blockio.store_block, bl, db)
db.close()
_remove_db(db_file)
2021-01-24 23:29:02 +00:00
def test_list_blocks(self):
db_file = identifyhome.identify_home() + 'test.db'
_remove_db(db_file)
2021-01-24 23:29:02 +00:00
db = safedb.SafeDB(db_file)
expected_list = []
for i in range(10):
bl = blockcreator.create_anonvdf_block(b'test' + int(i).to_bytes(1, 'big'), 'txt', 60)
blockio.store_block(bl, db)
expected_list.append(bl.id)
l = blockio.list_blocks_by_type('txt', db)
self.assertEqual(len(list(l)), len(expected_list))
for i in l:
self.assertIn(bytes(i), expected_list)
db.close()
_remove_db(db_file)
2021-01-17 19:44:57 +00:00
unittest.main()