Onionr/onionr/tests/test_storagecounter.py

47 lines
1.4 KiB
Python
Raw Normal View History

import sys, os
sys.path.append(".")
import unittest, uuid
import logger
from utils import createdirs
import onionrsetup as setup
from utils import createdirs
import onionrblocks
import filepaths
import onionrexceptions
import storagecounter
import onionrstorage
def _test_setup():
2019-09-11 02:48:07 +00:00
import config
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()
setup.setup_config()
config.reload()
class TestStorageCounter(unittest.TestCase):
def test_basic_amount(self):
2019-09-11 02:48:07 +00:00
import config
_test_setup()
self.assertIsNotNone(config.get('allocations.disk'))
2019-09-11 02:53:28 +00:00
self.assertGreaterEqual(config.get('allocations.disk'), 1000000)
def test_insert_too_much(self):
2019-09-11 02:48:07 +00:00
import config
_test_setup()
config.set('allocations.disk', 1000)
self.assertRaises(onionrexceptions.DiskAllocationReached, onionrblocks.insert, "test")
def test_count(self):
_test_setup()
counter = storagecounter.StorageCounter()
start_value = counter.get_amount()
b_hash = onionrblocks.insert("test")
self.assertGreater(counter.get_amount(), start_value)
onionrstorage.removeblock.remove_block(b_hash)
self.assertEqual(counter.get_amount(), start_value)
unittest.main()