import typing

from .context import mango
from .fakes import fake_account_info, fake_seeded_public_key
from .data import load_cache

from datetime import datetime
from decimal import Decimal


def test_cache_constructor() -> None:
    account_info = fake_account_info(fake_seeded_public_key("cache"))
    meta_data = mango.Metadata(mango.layouts.DATA_TYPE.parse(bytearray(b'\x07')), mango.Version.V1, True)
    timestamp = datetime.now()
    price_cache = [mango.PriceCache(Decimal(26), timestamp)]
    root_bank_cache = [mango.RootBankCache(Decimal("0.00001"), Decimal("0.00001"), timestamp)]
    perp_market_cache = [mango.PerpMarketCache(Decimal("0.00002"), Decimal("0.00002"), timestamp)]
    actual = mango.Cache(account_info, mango.Version.V1, meta_data, price_cache, root_bank_cache, perp_market_cache)

    assert actual is not None
    assert actual.logger is not None
    assert actual.account_info == account_info
    assert actual.address == fake_seeded_public_key("cache")
    assert actual.meta_data == meta_data
    assert actual.meta_data.data_type == mango.layouts.DATA_TYPE.Cache
    assert actual.price_cache == price_cache
    assert actual.root_bank_cache == root_bank_cache
    assert actual.perp_market_cache == perp_market_cache


def test_load_cache() -> None:
    cache = load_cache("tests/testdata/1deposit/cache.json")

    #
    # These values are all verified with the same file loaded in the TypeScript client.
    #

    actual_pc: typing.Sequence[typing.Optional[mango.PriceCache]] = cache.price_cache
    assert actual_pc[0] is not None and actual_pc[0].price == Decimal("0.33642499999999841975")
    assert actual_pc[1] is not None and actual_pc[1].price == Decimal("47380.32499999999999928946")
    assert actual_pc[2] is not None and actual_pc[2].price == Decimal("3309.69549999999999911893")
    assert actual_pc[3] is not None and actual_pc[3].price == Decimal("0.17261599999999788224")
    assert actual_pc[4] is not None and actual_pc[4].price == Decimal("8.79379999999999739657")
    assert actual_pc[5] is not None and actual_pc[5].price == Decimal("1")
    assert actual_pc[6] is not None and actual_pc[6].price == Decimal("1.00039999999999906777")
    assert actual_pc[7] is None
    assert actual_pc[8] is None
    assert actual_pc[9] is None
    assert actual_pc[10] is None
    assert actual_pc[11] is None
    assert actual_pc[12] is None
    assert actual_pc[13] is None
    assert actual_pc[14] is None

    actual_rbc: typing.Sequence[typing.Optional[mango.RootBankCache]] = cache.root_bank_cache
    assert actual_rbc[0] is not None and actual_rbc[0].deposit_index == Decimal("1001923.86460821722014813417")
    assert actual_rbc[0] is not None and actual_rbc[0].borrow_index == Decimal("1002515.45257855337824182129")
    assert actual_rbc[1] is not None and actual_rbc[1].deposit_index == Decimal("1000007.37249653914441083202")
    assert actual_rbc[1] is not None and actual_rbc[1].borrow_index == Decimal("1000166.98522159213999316307")
    assert actual_rbc[2] is not None and actual_rbc[2].deposit_index == Decimal("1000000.19554886875829424753")
    assert actual_rbc[2] is not None and actual_rbc[2].borrow_index == Decimal("1000001.13273253565107623331")
    assert actual_rbc[3] is not None and actual_rbc[3].deposit_index == Decimal("1000037.82149923799070379005")
    assert actual_rbc[3] is not None and actual_rbc[3].borrow_index == Decimal("1000044.28925241010965052624")
    assert actual_rbc[4] is not None and actual_rbc[4].deposit_index == Decimal("1000000.0000132182767842437")
    assert actual_rbc[4] is not None and actual_rbc[4].borrow_index == Decimal("1000000.14235973938041368569")
    assert actual_rbc[5] is not None and actual_rbc[5].deposit_index == Decimal("1000000.35244386506945346582")
    assert actual_rbc[5] is not None and actual_rbc[5].borrow_index == Decimal("1000000.66156146420993522383")
    assert actual_rbc[6] is not None and actual_rbc[6].deposit_index == Decimal("1000473.25161608998580575758")
    assert actual_rbc[6] is not None and actual_rbc[6].borrow_index == Decimal("1000524.37279217702128875089")
    assert actual_rbc[7] is None
    assert actual_rbc[7] is None
    assert actual_rbc[8] is None
    assert actual_rbc[8] is None
    assert actual_rbc[9] is None
    assert actual_rbc[9] is None
    assert actual_rbc[10] is None
    assert actual_rbc[10] is None
    assert actual_rbc[11] is None
    assert actual_rbc[11] is None
    assert actual_rbc[12] is None
    assert actual_rbc[12] is None
    assert actual_rbc[13] is None
    assert actual_rbc[13] is None
    assert actual_rbc[14] is None
    assert actual_rbc[14] is None
    assert actual_rbc[15] is not None and actual_rbc[15].deposit_index == Decimal("1000154.42276607534055088422")
    assert actual_rbc[15] is not None and actual_rbc[15].borrow_index == Decimal("1000219.00868743509063563124")

    actual_pmc: typing.Sequence[typing.Optional[mango.PerpMarketCache]] = cache.perp_market_cache
    assert actual_pmc[0] is None
    assert actual_pmc[1] is not None and actual_pmc[1].long_funding == Decimal("-751864.70031280454435673732")
    assert actual_pmc[1] is not None and actual_pmc[1].short_funding == Decimal("-752275.3557979761382519257")
    assert actual_pmc[2] is not None and actual_pmc[2].long_funding == Decimal("0")
    assert actual_pmc[2] is not None and actual_pmc[2].short_funding == Decimal("0")
    assert actual_pmc[3] is not None and actual_pmc[3].long_funding == Decimal("-636425.51790158202868497028")
    assert actual_pmc[3] is not None and actual_pmc[3].short_funding == Decimal("-636425.51790158202868497028")
    assert actual_pmc[4] is None
    assert actual_pmc[5] is None
    assert actual_pmc[6] is None
    assert actual_pmc[7] is None
    assert actual_pmc[8] is None
    assert actual_pmc[9] is None
    assert actual_pmc[10] is None
    assert actual_pmc[11] is None
    assert actual_pmc[12] is None
    assert actual_pmc[13] is None
    assert actual_pmc[14] is None