Skip to content
Snippets Groups Projects
Commit 6bc0b071 authored by Mariusz Trela's avatar Mariusz Trela
Browse files

A fix in `MockBlockProvider` class

parent 423c29d3
No related branches found
No related tags found
2 merge requests!456Release candidate v1 24,!389Mock tests for communities
...@@ -29,15 +29,17 @@ class MockBlockProvider(MockDataProvider): ...@@ -29,15 +29,17 @@ class MockBlockProvider(MockDataProvider):
data = {} data = {}
with open(file_name, "r") as src: with open(file_name, "r") as src:
data = load(src) data = load(src)
for block_num, transactions in data.items(): for block_num, block_content in data.items():
cls.add_block_data(block_num, transactions) cls.add_block_data(block_num, block_content)
@classmethod @classmethod
def add_block_data(cls, block_num, transactions): def add_block_data(cls, block_num, block_content):
if block_num in cls.block_data: if block_num in cls.block_data:
cls.block_data[str(block_num)].extend(transactions) assert 'transactions' in cls.block_data[str(block_num)]
assert 'transactions' in block_content
cls.block_data[str(block_num)]['transactions'] = cls.block_data[str(block_num)]['transactions'] + block_content['transactions']
else: else:
cls.block_data[str(block_num)] = transactions cls.block_data[str(block_num)] = block_content
@classmethod @classmethod
def get_block_data(cls, block_num, pop=False): def get_block_data(cls, block_num, pop=False):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment