class Plaid:
def __init__(self, clientId, clientSecret, accessToken=None):
self.client = Client(client_id=clientId, secret=clientSecret, access_token=accessToken)
def connect(self, accountType, username, password, email):
connect = self.client.connect(account_type=accountType, username=username, password=password, email=email)
if connect.ok:
json_response = json.loads(connect.content)
return json_response;
def answerMFA(self, accountType, mfaAnswer):
step = self.client.step(account_type=accountType, mfa=mfaAnswer)
if step.ok:
transactions = json.loads(step.content)
return transactions
def getTransactions(self, options=None):
transactions = self.client.transactions(options)
if transactions.ok:
transactions = json.loads(transactions.content)
return transactions
def delete(self):
self.client.delete_user()
def test_delete_user_requires_access_token():
client = Client('myclientid', 'mysecret')
with pytest.raises(Exception):
client.delete_user('bofa', 'foo')
def test_delete_user():
with patch('requests.delete') as mock_requests_delete:
client = Client('myclientid', 'mysecret', 'token')
client.delete_user()
assert mock_requests_delete.called
