From 4a7b72d500379fe6fad8577e0cadb0f8b797bd92 Mon Sep 17 00:00:00 2001
From: Bartek Wrona <wrona@syncad.com>
Date: Wed, 9 Dec 2020 21:42:46 +0100
Subject: [PATCH] Mocks should not impact production flow if no data is loaded.
---
hive/indexer/mock_block_provider.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/hive/indexer/mock_block_provider.py b/hive/indexer/mock_block_provider.py
index 32a38b84b..429d7a296 100644
--- a/hive/indexer/mock_block_provider.py
+++ b/hive/indexer/mock_block_provider.py
@@ -50,6 +50,9 @@ class MockBlockProvider(MockDataProvider):
@classmethod
def get_block_data(cls, block_num, make_on_empty=False):
+ if len(cls.block_data) == 0:
+ return None
+
data = cls.block_data.get(block_num, None)
#if data is not None:
@@ -57,8 +60,7 @@ class MockBlockProvider(MockDataProvider):
if make_on_empty and data is None:
data = cls.make_empty_block(block_num)
- if data is None:
- raise MockDataProviderException("No more blocks to serve")
+
return data
@classmethod
--
GitLab