more work on blocks tests

master
Kevin Froman 2019-09-10 20:59:31 -05:00
parent a50abcd20a
commit 92588c9f96
1 changed files with 17 additions and 8 deletions

View File

@ -2,23 +2,32 @@
import sys, os import sys, os
sys.path.append(".") sys.path.append(".")
import unittest, uuid, hashlib import unittest, uuid, hashlib
import nacl.exceptions
import nacl.signing, nacl.hash, nacl.encoding
TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
print("Test directory:", TEST_DIR)
os.environ["ONIONR_HOME"] = TEST_DIR
import onionrblocks import onionrblocks
import onionrstorage import onionrstorage
from utils import createdirs from utils import createdirs
from onionrutils import bytesconverter from onionrutils import bytesconverter
import onionrcrypto
import onionrblockapi
def setup_test():
TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
print("Test directory:", TEST_DIR)
os.environ["ONIONR_HOME"] = TEST_DIR
createdirs.create_dirs() createdirs.create_dirs()
class OnionrBlockTests(unittest.TestCase): class OnionrBlockTests(unittest.TestCase):
def test_plaintext_insert(self): def test_plaintext_insert(self):
setup_test()
message = 'hello world' message = 'hello world'
bl = onionrblocks.insert(message) bl = onionrblocks.insert(message)
self.assertTrue(bl.startswith('0'))
self.assertIn(bytesconverter.str_to_bytes(message), onionrstorage.getData(bl)) self.assertIn(bytesconverter.str_to_bytes(message), onionrstorage.getData(bl))
#def test_encrypted_insert(self): def test_encrypted_insert(self):
# key_pair_1 = nacl.signing.SigningKey.generate(encoder=nacl.encoding.base32) setup_test()
message = 'hello world2'
bl = onionrblocks.insert(message, asymPeer=onionrcrypto.pub_key)
self.assertIn(bytesconverter.str_to_bytes(message), onionrblockapi.Block(bl, decrypt=True).bcontent)
unittest.main() unittest.main()