fin-depo/test/test_kucoin.py

from decimal import Decimal

import fin_defs
import pytest

from fin_depo import defi_kucoin

from . import secrets

TEST_MARKET_ORDERS = False

needs_secrets = pytest.mark.skipif(
    not secrets.KUCOIN_KEY,
    reason='Secret kucoin_USERNAME required',
)

defi_kucoin.logger.setLevel('INFO')


@needs_secrets
def test_get_depo():
    """Can inspect kucoin depository."""
    fetcher = defi_kucoin.KucoinDepoFetcher(
        secrets.KUCOIN_KEY,
        secrets.KUCOIN_SECRET,
        secrets.KUCOIN_PASS,
    )

    depo = fetcher.get_depo()
    assert depo is not None


@needs_secrets
def test_place_market_order_requires_allow_trades():
    """Client fails if allow_trades is not enabled when doing market orders."""
    fetcher = defi_kucoin.KucoinDepoFetcher(
        secrets.KUCOIN_KEY,
        secrets.KUCOIN_SECRET,
        secrets.KUCOIN_PASS,
        allow_trades=False,
    )

    with pytest.raises(PermissionError) as m:
        fetcher.place_market_order(fin_defs.MPC, Decimal(1), fin_defs.USDT)

    assert 'KucoinDepoFetcher.allow_trades is not enabled: Cannot make trades' in str(m)


@needs_secrets
def test_place_buy_side_order():
    """Client can place buy side market orders."""
    fetcher = defi_kucoin.KucoinDepoFetcher(
        secrets.KUCOIN_KEY,
        secrets.KUCOIN_SECRET,
        secrets.KUCOIN_PASS,
        allow_trades=TEST_MARKET_ORDERS,
    )

    input_amount = Decimal('0.1')

    order_details = fetcher.place_market_order(
        fin_defs.USDT,
        input_amount,
        fin_defs.MPC,
    )

    assert order_details is not None
    assert order_details.order_id is not None
    assert order_details.raw_order_details is not None

    assert order_details.input_asset == fin_defs.USDT
    assert order_details.output_asset == fin_defs.MPC

    assert order_details.input_amount <= input_amount
    assert order_details.output_amount >= 0

    assert order_details.input_amount != order_details.output_amount

    assert order_details.fee_asset == fin_defs.USDT
    assert order_details.fee_amount <= Decimal('0.0002')


@needs_secrets
def test_place_sell_side_order():
    """Client can place sell side market orders."""
    fetcher = defi_kucoin.KucoinDepoFetcher(
        secrets.KUCOIN_KEY,
        secrets.KUCOIN_SECRET,
        secrets.KUCOIN_PASS,
        allow_trades=TEST_MARKET_ORDERS,
    )

    input_amount = Decimal('1')

    order_details = fetcher.place_market_order(
        fin_defs.MPC,
        input_amount,
        fin_defs.USDT,
    )

    assert order_details is not None
    assert order_details.order_id is not None
    assert order_details.raw_order_details is not None

    assert order_details.input_asset == fin_defs.MPC
    assert order_details.output_asset == fin_defs.USDT

    assert order_details.input_amount <= input_amount
    assert order_details.output_amount >= 0

    assert order_details.input_amount != order_details.output_amount

    assert order_details.fee_asset == fin_defs.USDT
    assert order_details.fee_amount is not None