mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-06-30 21:52:12 +02:00
76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
import pytest
|
|
|
|
from bigchaindb.models import Transaction
|
|
|
|
VOTES_ENDPOINT = '/api/v1/votes'
|
|
|
|
|
|
@pytest.mark.bdb
|
|
@pytest.mark.usefixtures('inputs')
|
|
def test_get_votes_endpoint(b, client):
|
|
tx = Transaction.create([b.me], [([b.me], 1)])
|
|
tx = tx.sign([b.me_private])
|
|
|
|
block = b.create_block([tx])
|
|
b.write_block(block)
|
|
|
|
# vote the block valid
|
|
vote = b.vote(block.id, b.get_last_voted_block().id, True)
|
|
b.write_vote(vote)
|
|
|
|
res = client.get(VOTES_ENDPOINT + '?block_id=' + block.id)
|
|
assert vote == res.json[0]
|
|
assert len(res.json) == 1
|
|
assert res.status_code == 200
|
|
|
|
|
|
@pytest.mark.bdb
|
|
@pytest.mark.usefixtures('inputs')
|
|
def test_get_votes_endpoint_multiple_votes(b, client):
|
|
from bigchaindb.common.crypto import generate_key_pair
|
|
|
|
tx = Transaction.create([b.me], [([b.me], 1)])
|
|
tx = tx.sign([b.me_private])
|
|
|
|
block = b.create_block([tx])
|
|
b.write_block(block)
|
|
last_block = b.get_last_voted_block().id
|
|
# vote the block valid
|
|
vote_valid = b.vote(block.id, last_block, True)
|
|
b.write_vote(vote_valid)
|
|
|
|
# vote the block invalid
|
|
# a note can only vote once so we need a new node_pubkey for the second
|
|
# vote
|
|
_, pk = generate_key_pair()
|
|
vote_invalid = b.vote(block.id, last_block, False)
|
|
vote_invalid['node_pubkey'] = pk
|
|
b.write_vote(vote_invalid)
|
|
|
|
res = client.get(VOTES_ENDPOINT + '?block_id=' + block.id)
|
|
assert len(res.json) == 2
|
|
assert res.status_code == 200
|
|
|
|
|
|
@pytest.mark.bdb
|
|
def test_get_votes_endpoint_returns_empty_list_not_found(client):
|
|
res = client.get(VOTES_ENDPOINT + '?block_id=')
|
|
assert [] == res.json
|
|
assert res.status_code == 200
|
|
|
|
res = client.get(VOTES_ENDPOINT + '?block_id=123')
|
|
assert [] == res.json
|
|
assert res.status_code == 200
|
|
|
|
|
|
@pytest.mark.bdb
|
|
def test_get_votes_endpoint_returns_400_bad_query_params(client):
|
|
res = client.get(VOTES_ENDPOINT)
|
|
assert res.status_code == 400
|
|
|
|
res = client.get(VOTES_ENDPOINT + '?ts_id=123')
|
|
assert res.status_code == 400
|
|
|
|
res = client.get(VOTES_ENDPOINT + '?tx_id=123&block_id=123')
|
|
assert res.status_code == 400
|