36 lines
1,001 B
Python
36 lines
1,001 B
Python
|
from tests.integration.util import (
|
||
|
|
create_client,
|
||
|
|
SANDBOX_INSTITUTION
|
||
|
|
)
|
||
|
|
|
||
|
|
access_token = None
|
||
|
|
|
||
|
|
|
||
|
|
def setup_module(module):
|
||
|
|
client = create_client()
|
||
|
|
pt_response = client.Sandbox.public_token.create(
|
||
|
|
SANDBOX_INSTITUTION, ['investments'])
|
||
|
|
exchange_response = client.Item.public_token.exchange(
|
||
|
|
pt_response['public_token'])
|
||
|
|
global access_token
|
||
|
|
access_token = exchange_response['access_token']
|
||
|
|
|
||
|
|
|
||
|
|
def teardown_module(module):
|
||
|
|
client = create_client()
|
||
|
|
client.Item.remove(access_token)
|
||
|
|
|
||
|
|
|
||
|
|
def test_get():
|
||
|
|
client = create_client()
|
||
|
|
response = client.Holdings.get(access_token)
|
||
|
|
assert response['item'] is not None
|
||
|
|
assert response['accounts'] is not None
|
||
|
|
assert response['securities'] is not None
|
||
|
|
assert response['holdings'] is not None
|
||
|
|
|
||
|
|
# get selected accounts
|
||
|
|
account_id = response['accounts'][0]['account_id']
|
||
|
|
response = client.Holdings.get(access_token, account_ids=[account_id])
|
||
|
|
assert len(response['accounts']) == 1
|