import sys

from bigchaindb.core import Bigchain
from bigchaindb.models import Transaction, Block
from bigchaindb.common.crypto import generate_key_pair
import time

test_size = int(sys.argv[1])


priv, pub = generate_key_pair()
b = Bigchain()

txs = []

prev_block = 'a'*64

def vote(block):
    global prev_block
    b.write_vote(b.vote(block.id, prev_block, True))
    prev_block=block.id



for i in range(test_size):
    tx = Transaction.create([pub], [([pub], i+1)])
    block = b.create_block([tx])
    b.write_block(block)
    vote(block)
    txs.append(tx)


for create in txs:
    tx = Transaction.transfer(create.to_inputs(), [([pub], 1)], create.id)
    block = b.create_block([tx])
    b.write_block(block)
    vote(block)


times = []

for i in range(10):
    start = time.time()
    outputs = b.get_outputs_filtered(pub, False)
    t = time.time() - start
    times.append(t)
    print("%.2fms" % (t*1000))
    assert len(outputs) == test_size