# Copyright © 2020 Interplanetary Database Association e.V.,
# Planetmint and IPDB software contributors.
# SPDX-License-Identifier: (Apache-2.0 AND CC-BY-4.0)
# Code is Apache-2.0 and docs are CC-BY-4.0

import pytest

from transactions.types.assets.create import Create
from planetmint.abci.block import Block
from ipld import marshal, multihash


BLOCKS_ENDPOINT = "/api/v1/blocks/"


@pytest.mark.bdb
@pytest.mark.usefixtures("inputs")
def test_get_block_endpoint(b, client, alice):
    import copy

    tx = Create.generate(
        [alice.public_key], [([alice.public_key], 1)], assets=[{"data": multihash(marshal({"cycle": "hero"}))}]
    )
    tx = tx.sign([alice.private_key])

    # with b.models.store_bulk_transactions we use `insert_many` where PyMongo
    # automatically adds an `_id` field to the tx, therefore we need the
    # deepcopy, for more info see:
    # https://api.mongodb.com/python/current/faq.html#writes-and-ids
    tx_dict = copy.deepcopy(tx.to_dict())
    b.models.store_bulk_transactions([tx])

    block = Block(app_hash="random_utxo", height=31, transactions=[tx.id])
    b.models.store_block(block._asdict())

    res = client.get(BLOCKS_ENDPOINT + str(block.height))
    expected_response = {"app_hash": "random_utxo", "height": block.height, "transaction_ids": [tx.id]}
    assert res.json == expected_response
    assert res.status_code == 200


@pytest.mark.bdb
@pytest.mark.usefixtures("inputs")
def test_get_block_returns_404_if_not_found(client):
    res = client.get(BLOCKS_ENDPOINT + "123")
    assert res.status_code == 404

    res = client.get(BLOCKS_ENDPOINT + "123/")
    assert res.status_code == 404


@pytest.mark.bdb
def test_get_block_containing_transaction(b, client, alice):
    tx = Create.generate(
        [alice.public_key], [([alice.public_key], 1)], assets=[{"data": multihash(marshal({"cycle": "hero"}))}]
    )
    tx = tx.sign([alice.private_key])
    b.models.store_bulk_transactions([tx])

    block = Block(app_hash="random_utxo", height=13, transactions=[tx.id])
    b.models.store_block(block._asdict())
    res = client.get("{}?transaction_id={}".format(BLOCKS_ENDPOINT, tx.id))
    expected_height = block.height
    assert res.json["height"] == expected_height
    assert res.status_code == 200


@pytest.mark.bdb
def test_get_blocks_by_txid_endpoint_returns_empty_list_not_found(client):
    res = client.get(BLOCKS_ENDPOINT + "?transaction_id=")
    assert res.status_code == 404

    res = client.get(BLOCKS_ENDPOINT + "?transaction_id=123")
    assert res.status_code == 404