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, ['liabilities'])
    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()
    # get liabilities for all accounts
    response = client.Liabilities.get(access_token)
    assert response['item'] is not None
    assert response['accounts'] is not None
    assert response['liabilities'] is not None

    # get liabiliteis for specified account
    account_id = response['accounts'][0]['account_id']
    response = client.Liabilities.get(access_token,
                                      account_ids=[account_id])
    assert response['liabilities'] is not None
    assert len(response['accounts']) == 1