import pytest
from unittest.mock import patch, Mock

from django.core.management import CommandError

from polaris.management.commands.poll_pending_deposits import Command, rdi, rri, logger
from polaris.models import Transaction
from polaris.tests.processes.test_create_stellar_deposit import mock_account

test_module = "polaris.management.commands.poll_pending_deposits"


def mock_poll_pending_deposits_success(deposits):
    """
    Returns deposits that are "ready"
    pending_external deposits are not "ready" because funds have not
    been received by the anchor.
    """
    return [
        d
        for d in deposits
        if d.status == Transaction.STATUS.pending_user_transfer_start
    ]


@pytest.mark.django_db
@patch(f"{test_module}.create_stellar_deposit", Mock(return_value=True))
@patch(
    f"{test_module}.get_or_create_transaction_destination_account",
    Mock(return_value=(mock_account, False, False)),
)
@patch(f"{test_module}.check_for_multisig", Mock(return_value=False))
def test_poll_pending_deposits_success(client, acc1_usd_deposit_transaction_factory):
    transaction_user = acc1_usd_deposit_transaction_factory()
    transaction_external = acc1_usd_deposit_transaction_factory()
    transaction_external.status = Transaction.STATUS.pending_external
    transaction_external.save()
    og_pending_deposits = rri.poll_pending_deposits
    rri.poll_pending_deposits = mock_poll_pending_deposits_success
    rdi.after_deposit = Mock()
    Command.execute_deposits()

    transaction_user.refresh_from_db()
    assert transaction_user.status == Transaction.STATUS.pending_anchor
    assert transaction_user.status_eta == 5
    assert rdi.after_deposit.was_called

    transaction_external.refresh_from_db()
    assert transaction_external.status == Transaction.STATUS.pending_external

    rri.poll_pending_deposits = og_pending_deposits


@pytest.mark.django_db
@patch(f"{test_module}.create_stellar_deposit", Mock(return_value=True))
@patch(
    f"{test_module}.get_or_create_transaction_destination_account",
    Mock(return_value=(mock_account, False, False)),
)
def test_poll_pending_deposits_bad_integration(
    client,
    acc1_usd_deposit_transaction_factory,
    acc1_usd_withdrawal_transaction_factory,
):
    # execute_deposits() queries for pending deposits
    acc1_usd_deposit_transaction_factory()
    # integration returns withdraw transaction
    withdrawal_transaction = acc1_usd_withdrawal_transaction_factory()
    rri.poll_pending_deposits = Mock(return_value=[withdrawal_transaction])
    logger.error = Mock()

    with pytest.raises(CommandError):
        Command.execute_deposits()

import pytest
from unittest.mock import patch, Mock

from django.core.management import CommandError

from polaris.management.commands.poll_pending_deposits import Command, rdi, rri, logger
from polaris.models import Transaction
from polaris.tests.processes.test_create_stellar_deposit import mock_account

test_module = "polaris.management.commands.poll_pending_deposits"


def mock_poll_pending_deposits_success(deposits):
    """
    Returns deposits that are "ready"
    pending_external deposits are not "ready" because funds have not
    been received by the anchor.
    """
    return [
        d
        for d in deposits
        if d.status == Transaction.STATUS.pending_user_transfer_start
    ]


@pytest.mark.django_db
@patch(f"{test_module}.create_stellar_deposit", Mock(return_value=True))
@patch(
    f"{test_module}.get_or_create_transaction_destination_account",
    Mock(return_value=(mock_account, False, False)),
)
@patch(f"{test_module}.check_for_multisig", Mock(return_value=False))
def test_poll_pending_deposits_success(client, acc1_usd_deposit_transaction_factory):
    transaction_user = acc1_usd_deposit_transaction_factory()
    transaction_external = acc1_usd_deposit_transaction_factory()
    transaction_external.status = Transaction.STATUS.pending_external
    transaction_external.save()
    og_pending_deposits = rri.poll_pending_deposits
    rri.poll_pending_deposits = mock_poll_pending_deposits_success
    rdi.after_deposit = Mock()
    Command.execute_deposits()

    transaction_user.refresh_from_db()
    assert transaction_user.status == Transaction.STATUS.pending_anchor
    assert transaction_user.status_eta == 5
    assert rdi.after_deposit.was_called

    transaction_external.refresh_from_db()
    assert transaction_external.status == Transaction.STATUS.pending_external

    rri.poll_pending_deposits = og_pending_deposits


@pytest.mark.django_db
@patch(f"{test_module}.create_stellar_deposit", Mock(return_value=True))
@patch(
    f"{test_module}.get_or_create_transaction_destination_account",
    Mock(return_value=(mock_account, False, False)),
)
def test_poll_pending_deposits_bad_integration(
    client,
    acc1_usd_deposit_transaction_factory,
    acc1_usd_withdrawal_transaction_factory,
):
    # execute_deposits() queries for pending deposits
    acc1_usd_deposit_transaction_factory()
    # integration returns withdraw transaction
    withdrawal_transaction = acc1_usd_withdrawal_transaction_factory()
    rri.poll_pending_deposits = Mock(return_value=[withdrawal_transaction])
    logger.error = Mock()

    with pytest.raises(CommandError):
        Command.execute_deposits()

Returns deposits that are "ready"     pending_external deposits are not "ready" because funds have not     been received by the anchor. # execute_deposits() queries for pending deposits # integration returns withdraw transaction