planetmint/tests/web/test_outputs.py

# 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 planetmint.transactions.types.assets.create import Create
from planetmint.transactions.types.assets.transfer import Transfer
from unittest.mock import MagicMock, patch


OUTPUTS_ENDPOINT = "/api/v1/outputs/"


@pytest.mark.bdb
@pytest.mark.userfixtures("inputs")
def test_get_outputs_endpoint(client, user_pk):
    m = MagicMock()
    m.txid = "a"
    m.output = 0
    with patch("planetmint.Planetmint.get_outputs_filtered") as gof:
        gof.return_value = [m, m]
        res = client.get(OUTPUTS_ENDPOINT + "?public_key={}".format(user_pk))
        assert res.json == [{"transaction_id": "a", "output_index": 0}, {"transaction_id": "a", "output_index": 0}]
    assert res.status_code == 200
    gof.assert_called_once_with(user_pk, None)


def test_get_outputs_endpoint_unspent(client, user_pk):
    m = MagicMock()
    m.txid = "a"
    m.output = 0
    with patch("planetmint.Planetmint.get_outputs_filtered") as gof:
        gof.return_value = [m]
        params = "?spent=False&public_key={}".format(user_pk)
        res = client.get(OUTPUTS_ENDPOINT + params)
    assert res.json == [{"transaction_id": "a", "output_index": 0}]
    assert res.status_code == 200
    gof.assert_called_once_with(user_pk, False)


@pytest.mark.bdb
@pytest.mark.userfixtures("inputs")
def test_get_outputs_endpoint_spent(client, user_pk):
    m = MagicMock()
    m.txid = "a"
    m.output = 0
    with patch("planetmint.Planetmint.get_outputs_filtered") as gof:
        gof.return_value = [m]
        params = "?spent=true&public_key={}".format(user_pk)
        res = client.get(OUTPUTS_ENDPOINT + params)
    assert res.json == [{"transaction_id": "a", "output_index": 0}]
    assert res.status_code == 200
    gof.assert_called_once_with(user_pk, True)


@pytest.mark.bdb
@pytest.mark.userfixtures("inputs")
def test_get_outputs_endpoint_without_public_key(client):
    res = client.get(OUTPUTS_ENDPOINT)
    assert res.status_code == 400


@pytest.mark.bdb
@pytest.mark.userfixtures("inputs")
def test_get_outputs_endpoint_with_invalid_public_key(client):
    expected = {"message": {"public_key": "Invalid base58 ed25519 key"}}
    res = client.get(OUTPUTS_ENDPOINT + "?public_key=abc")
    assert expected == res.json
    assert res.status_code == 400


@pytest.mark.bdb
@pytest.mark.userfixtures("inputs")
def test_get_outputs_endpoint_with_invalid_spent(client, user_pk):
    expected = {"message": {"spent": 'Boolean value must be "true" or "false" (lowercase)'}}
    params = "?spent=tru&public_key={}".format(user_pk)
    res = client.get(OUTPUTS_ENDPOINT + params)
    assert expected == res.json
    assert res.status_code == 400


@pytest.mark.abci
def test_get_divisble_transactions_returns_500(b, client):
    from planetmint.transactions.common import crypto
    import json

    TX_ENDPOINT = "/api/v1/transactions"

    def mine(tx_list):
        b.store_bulk_transactions(tx_list)

    alice_priv, alice_pub = crypto.generate_key_pair()
    bob_priv, bob_pub = crypto.generate_key_pair()
    carly_priv, carly_pub = crypto.generate_key_pair()

    create_tx = Create.generate([alice_pub], [([alice_pub], 4)])
    create_tx.sign([alice_priv])

    res = client.post(TX_ENDPOINT, data=json.dumps(create_tx.to_dict()))
    assert res.status_code == 202

    mine([create_tx])

    transfer_tx = Transfer.generate(create_tx.to_inputs(), [([alice_pub], 3), ([bob_pub], 1)], asset_id=create_tx.id)
    transfer_tx.sign([alice_priv])

    res = client.post(TX_ENDPOINT, data=json.dumps(transfer_tx.to_dict()))
    assert res.status_code == 202

    mine([transfer_tx])

    transfer_tx_carly = Transfer.generate([transfer_tx.to_inputs()[1]], [([carly_pub], 1)], asset_id=create_tx.id)
    transfer_tx_carly.sign([bob_priv])

    res = client.post(TX_ENDPOINT, data=json.dumps(transfer_tx_carly.to_dict()))
    assert res.status_code == 202

    mine([transfer_tx_carly])

    asset_id = create_tx.id

    url = TX_ENDPOINT + "?asset_id=" + asset_id
    assert client.get(url).status_code == 200
    assert len(client.get(url).json) == 3

    url = OUTPUTS_ENDPOINT + "?public_key=" + alice_pub
    assert client.get(url).status_code == 200

    url = OUTPUTS_ENDPOINT + "?public_key=" + bob_pub
    assert client.get(url).status_code == 200

    url = OUTPUTS_ENDPOINT + "?public_key=" + carly_pub
    assert client.get(url).status_code == 200