github.com/muhammedhassanm/blockchain@v0.0.0-20200120143007-697261defd4d/sawtooth-core-master/validator/tests/test_completer/mock.py (about)

     1  # Copyright 2017 Intel Corporation
     2  #
     3  # Licensed under the Apache License, Version 2.0 (the "License");
     4  # you may not use this file except in compliance with the License.
     5  # You may obtain a copy of the License at
     6  #
     7  #     http://www.apache.org/licenses/LICENSE-2.0
     8  #
     9  # Unless required by applicable law or agreed to in writing, software
    10  # distributed under the License is distributed on an "AS IS" BASIS,
    11  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  # See the License for the specific language governing permissions and
    13  # limitations under the License.
    14  # ------------------------------------------------------------------------------
    15  
    16  
    17  class MockGossip():
    18      def __init__(self):
    19          self.requested_blocks = []
    20          self.requested_batches = []
    21          self.requested_batches_by_txn_id = []
    22  
    23      def broadcast_block_request(self, block_id):
    24          self.requested_blocks.append(block_id)
    25  
    26      def broadcast_batch_by_batch_id_request(self, batch_id):
    27          self.requested_batches.append(batch_id)
    28  
    29      def broadcast_batch_by_transaction_id_request(self, transaction_ids):
    30          for txn_id in transaction_ids:
    31              self.requested_batches_by_txn_id.append(txn_id)