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"})