def test_account_return(self): account_data = {"id": 1, "name": "TestName"} mock_data_interface = Mock() mock_data_interface.get.return_value = account_data account = Account(mock_data_interface) self.assertDictEqual(account_data, account.get_account(1))
def test_get_current_balance(self, mock_request): mock_response = Mock() mock_response.status_code = 200 mock_response.text = 'Some text data' mock_request.get.return_value = mock_response account = Account(Mock()) self.assertEqual( {'status': 200, 'data': 'Some text data'}, account.get_current_balance(1) )
def test_account_return_exception(self): mock_data_interface = Mock() mock_data_interface.get.side_effect = ConnectionError() account = Account(mock_data_interface) self.assertEqual('Connection error occurred. Try Again.', account.get_account(1))