mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
4a4cbc8de0
Solution: Added a little helper function to make sure there is at least one block in the chain when the test is executed
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
# Copyright BigchainDB GmbH and BigchainDB contributors
|
|
# SPDX-License-Identifier: (Apache-2.0 AND CC-BY-4.0)
|
|
# Code is Apache-2.0 and docs are CC-BY-4.0
|
|
|
|
from functools import singledispatch
|
|
|
|
from bigchaindb.backend.localmongodb.connection import LocalMongoDBConnection
|
|
|
|
|
|
@singledispatch
|
|
def flush_db(connection, dbname):
|
|
raise NotImplementedError
|
|
|
|
|
|
@flush_db.register(LocalMongoDBConnection)
|
|
def flush_localmongo_db(connection, dbname):
|
|
connection.conn[dbname].bigchain.delete_many({})
|
|
connection.conn[dbname].blocks.delete_many({})
|
|
connection.conn[dbname].transactions.delete_many({})
|
|
connection.conn[dbname].assets.delete_many({})
|
|
connection.conn[dbname].metadata.delete_many({})
|
|
connection.conn[dbname].utxos.delete_many({})
|
|
connection.conn[dbname].validators.delete_many({})
|
|
|
|
|
|
def generate_block(bigchain):
|
|
from bigchaindb.common.crypto import generate_key_pair
|
|
from bigchaindb.models import Transaction
|
|
import time
|
|
|
|
alice = generate_key_pair()
|
|
tx = Transaction.create([alice.public_key],
|
|
[([alice.public_key], 1)],
|
|
asset=None)\
|
|
.sign([alice.private_key])
|
|
|
|
code, message = bigchain.write_transaction(tx, 'broadcast_tx_commit')
|
|
assert code == 202
|
|
time.sleep(2)
|