36 lines
1.4 KiB
Python
36 lines
1.4 KiB
Python
|
from .context import mango
|
||
|
|
from .fakes import fake_account_info, fake_seeded_public_key
|
||
|
|
|
||
|
|
from decimal import Decimal
|
||
|
|
from mango.layouts import layouts
|
||
|
|
|
||
|
|
|
||
|
|
def test_construction():
|
||
|
|
account_info = fake_account_info()
|
||
|
|
meta_data = mango.Metadata(layouts.DATA_TYPE.Group, mango.Version.V1, True)
|
||
|
|
group = fake_seeded_public_key("group")
|
||
|
|
owner = fake_seeded_public_key("owner")
|
||
|
|
in_basket = [Decimal(1), Decimal(0), Decimal(3)]
|
||
|
|
deposits = [Decimal(10), Decimal(0), Decimal(5)]
|
||
|
|
borrows = [Decimal(0), Decimal(0), Decimal(0)]
|
||
|
|
spot_open_orders = [fake_seeded_public_key("spot1"), fake_seeded_public_key(
|
||
|
|
"spot2"), fake_seeded_public_key("spot3")]
|
||
|
|
|
||
|
|
# TODO - this isn't right.
|
||
|
|
perp_accounts = [fake_seeded_public_key("perp1"), fake_seeded_public_key("perp2"), fake_seeded_public_key("perp3")]
|
||
|
|
|
||
|
|
actual = mango.Account(account_info, mango.Version.V1, meta_data, group, owner,
|
||
|
|
in_basket, deposits, borrows, spot_open_orders, perp_accounts)
|
||
|
|
|
||
|
|
assert actual is not None
|
||
|
|
assert actual.logger is not None
|
||
|
|
assert actual.version == mango.Version.V1
|
||
|
|
assert actual.meta_data == meta_data
|
||
|
|
assert actual.group == group
|
||
|
|
assert actual.owner == owner
|
||
|
|
assert actual.in_basket == in_basket
|
||
|
|
assert actual.deposits == deposits
|
||
|
|
assert actual.borrows == borrows
|
||
|
|
assert actual.spot_open_orders == spot_open_orders
|
||
|
|
assert actual.perp_accounts == perp_accounts
|