pluto/handler/tests/test_markets.py

import logging
from unittest import TestCase
from unittest.mock import MagicMock

import lib.markets
import settings
from sources.agora import Agora
from tests.common import expected_to_update, fake_public_ads


class TestMarkets(TestCase):
    def setUp(self):
        logging.disable(logging.CRITICAL)
        self.markets = lib.markets.Markets()
        self.agora = Agora()
        self.markets.sinks = MagicMock()
        self.markets.sinks.currencies = [
            "AUD",
            "CAD",
            "CHF",
            "CZK",
            "DKK",
            "EUR",
            "GBP",
            "HKD",
            "HUF",
            "JPY",
            "MXN",
            "NOK",
            "NZD",
            "PLN",
            "RUB",
            "SEK",
            "SGD",
            "THB",
            "TRY",
            "USD",
            "ZAR",
        ]

    def test_autoprice(self):
        ads = [
            [
                "2b6dba4d-c9db-48f2-adba-4dc9dba8f2a0",
                "Xpoterlolipop",
                "182.80",
                "NATIONAL_BANK",
                "XMR",
                "USD",
                1.18,
            ],
            [
                "57e3e8d6-45fe-40da-a3e8-d645fe20da46",
                "SecureMole",
                "183.26",
                "NATIONAL_BANK",
                "XMR",
                "USD",
                1.19,
            ],
            [
                "87af6467-be02-476e-af64-67be02676e9a",
                "topmonero",
                "183.42",
                "NATIONAL_BANK",
                "XMR",
                "USD",
                1.19,
            ],
            [
                "65b452e3-a29f-4233-b452-e3a29fe23369",
                "topmonero",
                "183.42",
                "NATIONAL_BANK",
                "XMR",
                "USD",
                1.19,
            ],
            [
                "d2c6645c-6d56-4094-8664-5c6d5640941b",
                "topmonero",
                "183.42",
                "NATIONAL_BANK",
                "XMR",
                "USD",
                1.19,
            ],
        ]
        currency = "EUR"
        margin = self.markets.autoprice("topmonero", 1.1, 1.3, ads, currency)
        expected_margin = 1.18
        self.assertEqual(margin, expected_margin)

    def test_get_new_ad_equation(self):
        self.maxDiff = None
        settings.settings.Agora.MinMargin = 1.17
        settings.settings.Agora.MaxMargin = 1.3
        # 437 should be 1.3 but is 1.21
        to_update = self.markets.get_new_ad_equations("agora", fake_public_ads)
        self.assertCountEqual(to_update, expected_to_update)
        res_xmr = self.markets.get_new_ad_equations("agora", fake_public_ads, ["XMR"])
        expected_xmr_to_update = [x for x in expected_to_update if x[2] == "XMR"]
        self.assertCountEqual(res_xmr, expected_xmr_to_update)

        res_btc = self.markets.get_new_ad_equations("agora", fake_public_ads, ["BTC"])
        expected_btc_to_update = [x for x in expected_to_update if x[2] == "BTC"]
        self.assertCountEqual(res_btc, expected_btc_to_update)

        res_both = self.markets.get_new_ad_equations(
            "agora", fake_public_ads, ["XMR", "BTC"]
        )
        self.assertCountEqual(res_both, expected_to_update)

    def test_format_ad(self):
        settings.settings.Platform.Ad = """* Set **Country of recipient's bank** to **"United Kingdom"**
$PAYMENT$
* Set **Company name** to **"PATHOGEN LIMITED"**"""
        payment_details = {
            "sort_code": "02-03-04",
            "account_number": "0023-0045",
        }
        payment_details_text = self.markets.format_payment_details(
            "GBP", payment_details
        )
        ad_text = self.markets.format_ad("XMR", "GBP", payment_details_text)
        expected = """* Set **Country of recipient's bank** to **"United Kingdom"**
Payment details will be released after verification has passed.
If you've already completed verification, they will be sent immediately.
* Set **Company name** to **"PATHOGEN LIMITED"**"""

        self.assertEqual(ad_text, expected)

    def test_format_payment_details(self):
        payment_details = {
            "sort_code": "02-03-04",
            "account_number": "0023-0045",
        }
        payment_details_text = self.markets.format_payment_details(
            "GBP", payment_details
        )

        expected = """Payment details will be released after verification has passed.
If you've already completed verification, they will be sent immediately."""
        self.assertEqual(payment_details_text, expected)

        expected_real = """* Sort code: **02-03-04**
* Account number: **0023-0045**
Please send in GBP."""
        payment_details_text_real = self.markets.format_payment_details(
            "GBP", payment_details, real=True
        )
        self.assertEqual(payment_details_text_real, expected_real)