diff --git a/onionr/tests/test_blocks.py b/onionr/tests/test_blocks.py index 8dfa103f..242b4bfa 100755 --- a/onionr/tests/test_blocks.py +++ b/onionr/tests/test_blocks.py @@ -2,23 +2,32 @@ import sys, os sys.path.append(".") 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 onionrstorage from utils import createdirs from onionrutils import bytesconverter -createdirs.create_dirs() +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() + class OnionrBlockTests(unittest.TestCase): def test_plaintext_insert(self): + setup_test() message = 'hello world' bl = onionrblocks.insert(message) + self.assertTrue(bl.startswith('0')) self.assertIn(bytesconverter.str_to_bytes(message), onionrstorage.getData(bl)) - #def test_encrypted_insert(self): - # key_pair_1 = nacl.signing.SigningKey.generate(encoder=nacl.encoding.base32) + def test_encrypted_insert(self): + 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() \ No newline at end of file