mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
7bd6d485bc
* Test Bigchain.get_blocks_status_containing_tx() exception case * Test Bigchaindb.has_previous_vote() * Add missing blank lines (pep 8) * Group imports * Move ImproperVoteError into .exceptions.py * Simplify logic * Simplify formatting * Imrpove the docstrings a bit * Move GenesisBlockAlreadyExistsError into .exceptions.py * Remove unused import * Add missing blank line * Remove extra blank line * Group imports from same module together * Add test for Bigchain.transaction_exists()
98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
from collections import namedtuple
|
|
|
|
from rethinkdb.ast import RqlQuery
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def config(request, monkeypatch):
|
|
config = {
|
|
'database': {
|
|
'host': 'host',
|
|
'port': 28015,
|
|
'name': 'bigchain',
|
|
},
|
|
'keypair': {
|
|
'public': 'pubkey',
|
|
'private': 'privkey',
|
|
},
|
|
'keyring': [],
|
|
'CONFIGURED': True,
|
|
'consensus_plugin': 'default'
|
|
}
|
|
|
|
monkeypatch.setattr('bigchaindb.config', config)
|
|
|
|
return config
|
|
|
|
|
|
def test_bigchain_class_default_initialization(config):
|
|
from bigchaindb.core import Bigchain
|
|
from bigchaindb.consensus import BaseConsensusRules
|
|
bigchain = Bigchain()
|
|
assert bigchain.host == config['database']['host']
|
|
assert bigchain.port == config['database']['port']
|
|
assert bigchain.dbname == config['database']['name']
|
|
assert bigchain.me == config['keypair']['public']
|
|
assert bigchain.me_private == config['keypair']['private']
|
|
assert bigchain.nodes_except_me == config['keyring']
|
|
assert bigchain.consensus == BaseConsensusRules
|
|
assert bigchain._conn is None
|
|
|
|
|
|
def test_bigchain_class_initialization_with_parameters(config):
|
|
from bigchaindb.core import Bigchain
|
|
from bigchaindb.consensus import BaseConsensusRules
|
|
init_kwargs = {
|
|
'host': 'some_node',
|
|
'port': '12345',
|
|
'dbname': 'atom',
|
|
'public_key': 'white',
|
|
'private_key': 'black',
|
|
'keyring': ['key_one', 'key_two'],
|
|
'consensus_plugin': 'default'
|
|
}
|
|
bigchain = Bigchain(**init_kwargs)
|
|
assert bigchain.host == init_kwargs['host']
|
|
assert bigchain.port == init_kwargs['port']
|
|
assert bigchain.dbname == init_kwargs['dbname']
|
|
assert bigchain.me == init_kwargs['public_key']
|
|
assert bigchain.me_private == init_kwargs['private_key']
|
|
assert bigchain.nodes_except_me == init_kwargs['keyring']
|
|
assert bigchain.consensus == BaseConsensusRules
|
|
assert bigchain._conn is None
|
|
|
|
|
|
def test_get_blocks_status_containing_tx(monkeypatch):
|
|
from bigchaindb.core import Bigchain
|
|
blocks = [
|
|
{'id': 1}, {'id': 2}
|
|
]
|
|
monkeypatch.setattr(
|
|
Bigchain, 'search_block_election_on_index', lambda x, y, z: blocks)
|
|
monkeypatch.setattr(
|
|
Bigchain, 'block_election_status', lambda x, y: Bigchain.BLOCK_VALID)
|
|
bigchain = Bigchain(public_key='pubkey', private_key='privkey')
|
|
with pytest.raises(Exception):
|
|
bigchain.get_blocks_status_containing_tx('txid')
|
|
|
|
|
|
def test_has_previous_vote(monkeypatch):
|
|
from bigchaindb.core import Bigchain
|
|
monkeypatch.setattr(
|
|
'bigchaindb.util.verify_vote_signature', lambda block, vote: False)
|
|
bigchain = Bigchain(public_key='pubkey', private_key='privkey')
|
|
block = {'votes': ({'node_pubkey': 'pubkey'},)}
|
|
with pytest.raises(Exception):
|
|
bigchain.has_previous_vote(block)
|
|
|
|
|
|
@pytest.mark.parametrize('items,exists', (((0,), True), ((), False)))
|
|
def test_transaction_exists(monkeypatch, items, exists):
|
|
from bigchaindb.core import Bigchain
|
|
monkeypatch.setattr(
|
|
RqlQuery, 'run', lambda x, y: namedtuple('response', 'items')(items))
|
|
bigchain = Bigchain(public_key='pubkey', private_key='privkey')
|
|
assert bigchain.transaction_exists('txid') is exists
|