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