diff --git a/fin_depo/defi_partisia_blockchain.py b/fin_depo/defi_partisia_blockchain.py index 337d904..7a85d6f 100644 --- a/fin_depo/defi_partisia_blockchain.py +++ b/fin_depo/defi_partisia_blockchain.py @@ -166,7 +166,7 @@ class PartisiaBlockchainAccountDepoFetcher(DepoFetcher): {fin_defs.MPC: balances.mpc}, ) depo_byoc = DepoSingle( - 'BYOC', + 'Bring Your Own Coin', balances.update_time, {BYOC_ASSETS[k]: v for k, v in balances.byoc.items()}, ) diff --git a/test/test_nordnet.py b/test/test_nordnet.py index 7999dcb..55c27dc 100644 --- a/test/test_nordnet.py +++ b/test/test_nordnet.py @@ -14,11 +14,11 @@ needs_secrets = pytest.mark.skipif( @needs_secrets def test_get_depo(): session = requests.Session() - nordnet = investbank_nordnet.NordnetDepoFetcher( + fetcher = investbank_nordnet.NordnetDepoFetcher( session, secrets.NORDNET_USERNAME, secrets.NORDNET_PASSWORD, ) - depo = nordnet.get_depo() + depo = fetcher.get_depo() assert depo is not None diff --git a/test/test_partisia_blockchain.py b/test/test_partisia_blockchain.py new file mode 100644 index 0000000..eaedd6c --- /dev/null +++ b/test/test_partisia_blockchain.py @@ -0,0 +1,23 @@ +import fin_defs +import requests + +from fin_depo import defi_partisia_blockchain + + +def test_get_depo(): + session = requests.Session() + fetcher = defi_partisia_blockchain.PartisiaBlockchainAccountDepoFetcher( + session, + '00f3a778fbcf663e4f0d47b41aa4ec51e0f9d39d60', + ) + + depo = fetcher.get_depo() + assert depo is not None + assert depo.name == 'Partisia Blockchain' + assert depo.updated_time is not None + assert depo.nested[0].name == 'Native' + assert depo.nested[0].updated_time == depo.updated_time + assert depo.nested[1].name == 'Bring Your Own Coin' + assert depo.nested[1].updated_time == depo.updated_time + + assert depo.get_amount_of_asset(fin_defs.MPC)