LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

fixed tests

Browse source
This commit is contained in:
Lex Berezhny 2021-08-06 10:52:47 -04:00
parent 4e8d10cb44
commit b4c3307cdf
3 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lbry/blob/disk_space_manager.py
View file

@ -9,7 +9,7 @@ class DiskSpaceManager:
@property
def space_used_bytes(self):
used = 0
data_dir = self.config.data_dir
data_dir = os.path.join(self.config.data_dir, 'blobfiles')
for item in os.listdir(data_dir):
blob_path = os.path.join(data_dir, item)
if os.path.isfile(blob_path):

3
tests/integration/datanetwork/test_file_commands.py
View file

@ -520,6 +520,7 @@ class DiskSpaceManagement(CommandTestCase):
self.assertIn('disk_space', status)
self.assertEqual(status['disk_space']['used'], '0')
await self.stream_create('foo', '0.01', data=('0' * 3 * 1024 * 1024).encode())
await self.stream_create('foo', '0.01', data=('0' * 2 * 1024 * 1024).encode())
status = await self.daemon.jsonrpc_status()
self.assertIn('disk_space', status)
self.assertEqual(status['disk_space']['used'], '3')
self.assertEqual(status['disk_space']['used'], '5')

3
tests/unit/blob/test_disk_space_manager.py
View file

@ -11,6 +11,7 @@ class ConfigurationTests(unittest.TestCase):
def test_space_calculation(self):
with tempfile.TemporaryDirectory() as temp_dir:
os.mkdir(os.path.join(temp_dir, 'blobfiles'))
config = Config(
data_dir=temp_dir,
wallet_dir=temp_dir,
@ -18,6 +19,6 @@ class ConfigurationTests(unittest.TestCase):
)
dsm = DiskSpaceManager(config)
self.assertEqual(0, dsm.space_used_mb)
with open(os.path.join(config.data_dir, '3mb-file'), 'w') as blob:
with open(os.path.join(config.data_dir, 'blobfiles', '3mb-file'), 'w') as blob:
blob.write('0' * 3 * 1024 * 1024)
self.assertEqual(3, dsm.space_used_mb)