def test_custom_index_with_header(self, get_elastic_client_mock, storage_get_mock, storage_put_mock, helpers):
model = l1_block_model.L1BlockModel(dc_id="one", block_id="2", timestamp="123", prev_id="0", prev_proof="apple")
self.elasticsearch.put_index_in_storage("BananaFolder", "MyUniqueId", model)
self.elasticsearch._es_client.index.assert_called_with(
body=json.dumps(
{
"version": "1",
"dcrn": "Block::L1::SearchIndex",
"dc_id": "one",
"block_id": 2,
"timestamp": 123,
"prev_id": 0,
"prev_proof": "apple",
"s3_object_folder": "BLOCK",
"s3_object_id": "2",
"l2_verifications": 0,
"l3_verifications": 0,
"l4_verifications": 0,
"l5_verifications": 0,
},
separators=(",", ":"),
),
doc_type="_doc",
id="MyUniqueId",
index="mixedcase-internal-id-bananafolder",
)
def make_l1_block():
return l1_block_model.L1BlockModel(
dc_id="an id",
block_id="8474745",
timestamp="129874",
prev_id="previous block",
prev_proof="the previous block proof",
stripped_transactions=["some", "transactions", "which", "are", "strings"],
)
def create_l1_block():
return l1_block_model.L1BlockModel(
dc_id="dc_id",
block_id="123",
timestamp="12345",
prev_proof="prev_proof",
prev_id="101",
transactions=[create_tx()],
scheme="trust",
proof="proof",
)
def test_put_index_calls_es_index_correctly(self, get_elastic_client_mock, storage_get_mock, storage_put_mock, helpers):
model = l1_block_model.L1BlockModel(dc_id="one", block_id="2", timestamp="123", prev_id="0", prev_proof="apple")
model2 = l1_block_model.L1BlockModel(dc_id="one", block_id="3", timestamp="123", prev_id="2", prev_proof="banana")
self.elasticsearch.put_many_index_only("BananaFolder", {model, model2})
helpers.bulk.assert_called_with(ANY, ANY, index="mixedcase-internal-id-bananafolder")
def test_setting_validations(self):
l1block = l1_block_model.L1BlockModel(dc_id="a", block_id="b")
self.assertEqual(l1block.get_associated_l1_dcid(), "a")
self.assertEqual(l1block.get_associated_l1_block_id(), {"b"})