def test_get_account_portfolio(self, MockOAuthSession):
"""test_get_account_positions(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions"""
# Set Mock returns
MockOAuthSession().get().json.return_value = {"account": "abc123"}
MockOAuthSession().get().text = r"<xml> returns </xml>"
account = accounts.ETradeAccounts("abc123", "xyz123", "abctoken",
"xyzsecret")
# Test Dev
result = account.get_account_portfolio("12345")
self.assertTrue(isinstance(result, dict))
# Test API URL
MockOAuthSession().get.assert_called_with(
"https://apisb.etrade.com/v1/accounts/12345/portfolio", params={})
result = account.get_account_portfolio("12345", resp_format="json")
self.assertTrue(isinstance(result, dict))
# Test Prod
account = accounts.ETradeAccounts("abc123",
"xyz123",
"abctoken",
"xyzsecret",
dev=False)
result = account.get_account_portfolio("12345")
self.assertTrue(isinstance(result, dict))
# Test API URL
MockOAuthSession().get.assert_called_with(
"https://api.etrade.com/v1/accounts/12345/portfolio", params={})
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
result = account.get_account_portfolio("12345", resp_format="xml")
self.assertTrue(isinstance(result, dict))
def test_get_account_balance(self, MockOAuthSession):
"""test_get_account_balance(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions"""
# Set Mock returns
MockOAuthSession().get().json.return_value = {"account": "abc123"}
MockOAuthSession().get().text = r"<xml> returns </xml>"
account = accounts.ETradeAccounts("abc123", "xyz123", "abctoken",
"xyzsecret")
# Test Dev XML
result = account.get_account_balance("12345", resp_format="xml")
self.assertTrue(isinstance(result, dict))
# Test Dev JSON
result = account.get_account_balance("12345", resp_format="json")
self.assertTrue(isinstance(result, dict))
# Test API URL
MockOAuthSession().get.assert_called_with(
"https://apisb.etrade.com/v1/accounts/12345/balance.json",
params={
"instType": "BROKERAGE",
"realTimeNAV": True
},
)
account = accounts.ETradeAccounts("abc123",
"xyz123",
"abctoken",
"xyzsecret",
dev=False)
# Test Prod JSON
result = account.get_account_balance("12345", resp_format="json")
self.assertTrue(isinstance(result, dict))
# Test API URL
MockOAuthSession().get.assert_called_with(
"https://api.etrade.com/v1/accounts/12345/balance.json",
params={
"instType": "BROKERAGE",
"realTimeNAV": True
},
)
# xml prod
result = account.get_account_balance("12345", resp_format="xml")
self.assertTrue(isinstance(result, dict))
# Test API URL
MockOAuthSession().get.assert_called_with(
"https://api.etrade.com/v1/accounts/12345/balance",
params={
"instType": "BROKERAGE",
"realTimeNAV": True
},
)
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
def test_get_account_balance(self, MockOAuthSession):
'''test_get_account_balance(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions'''
# Set Mock returns
MockOAuthSession().get().json.return_value = "{'account': 'abc123'}"
MockOAuthSession().get().text = r'<xml> returns </xml>'
account = accounts.ETradeAccounts('abc123', 'xyz123', 'abctoken',
'xyzsecret')
# Test Dev JSON
self.assertEqual(account.get_account_balance(12345),
"{'account': 'abc123'}")
# Test API URL
MockOAuthSession().get.assert_called_with(
('https://etwssandbox.etrade.com/accounts/'
'sandbox/rest/accountbalance/12345.json'))
# Test Prod JSON
self.assertEqual(account.get_account_balance(12345, dev=False),
"{'account': 'abc123'}")
# Test API URL
MockOAuthSession().get.assert_called_with(
('https://etws.etrade.com/accounts/'
'rest/accountbalance/12345.json'))
# Test Dev XML
self.assertEqual(account.get_account_balance(12345, resp_format='xml'),
r'<xml> returns </xml>')
#MockOAuthSession().get.assert_called_with(
# ('https://etws.etrade.com/accounts/'
# 'rest/accountbalance/12345')
# )
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
def test_get_account_balance(self, MockOAuthSession):
"""test_get_account_balance(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions"""
# Set Mock returns
MockOAuthSession().get().json.return_value = "{'account': 'abc123'}"
MockOAuthSession().get().text = r"<xml> returns </xml>"
account = accounts.ETradeAccounts("abc123", "xyz123", "abctoken", "xyzsecret")
# Test Dev JSON
self.assertEqual(account.get_account_balance(12345), "{'account': 'abc123'}")
# Test API URL
MockOAuthSession().get.assert_called_with(
"https://apisb.etrade.com/v1/accounts/12345/balance.json",
params={"instType": "BROKERAGE", "realTimeNAV": True},
)
# Test Prod JSON
self.assertEqual(
account.get_account_balance(12345, dev=False), "{'account': 'abc123'}"
)
# Test API URL
MockOAuthSession().get.assert_called_with(
"https://api.etrade.com/v1/accounts/12345/balance.json",
params={"instType": "BROKERAGE", "realTimeNAV": True},
)
# Test Dev XML
result = str(account.get_account_balance(12345, resp_format="xml"))
self.assertEqual(result, "{'xml': 'returns'}")
# MockOAuthSession().get.assert_called_with(
# ('https://etws.etrade.com/accounts/'
# 'rest/accountbalance/12345')
# )
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
def test_list_accounts(self, MockOAuthSession):
"""test_list_accounts(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions"""
# Set Mock returns
MockOAuthSession().get().json.return_value = "{'account': 'abc123'}"
MockOAuthSession().get().text = r"<xml> returns </xml>"
account = accounts.ETradeAccounts("abc123", "xyz123", "abctoken", "xyzsecret")
# Test Dev JSON
self.assertEqual(account.list_accounts(), "{'account': 'abc123'}")
# Test API URL
MockOAuthSession().get.assert_called_with(
("https://apisb.etrade.com/v1/accounts/list.json")
)
# Test Prod JSON
self.assertEqual(account.list_accounts(dev=False), "{'account': 'abc123'}")
# Test API URL
MockOAuthSession().get.assert_called_with(
("https://api.etrade.com/v1/accounts/list.json")
)
# Test Dev XML
result = str(account.list_accounts(resp_format="xml"))
self.assertEqual(result, "{'xml': 'returns'}")
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
def test_get_transaction_details(self, MockOAuthSession):
"""test_get_transaction_details(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions"""
# Set Mock returns
MockOAuthSession().get().json.return_value = "{'transaction': 'abc123'}"
MockOAuthSession().get().text = r"<xml> returns </xml>"
account = accounts.ETradeAccounts("abc123", "xyz123", "abctoken", "xyzsecret")
# Test Dev JSON
self.assertEqual(
account.get_transaction_details(12345, 67890), "{'transaction': 'abc123'}"
)
# Test API URL
MockOAuthSession().get.assert_called_with(
(
"https://apisb.etrade.com/accounts/v1"
"sandbox/rest/12345/transactions/67890.json"
),
params={},
)
# Test Prod JSON
self.assertEqual(
account.get_transaction_details(12345, 67890, dev=False),
"{'transaction': 'abc123'}",
)
# Test API URL
MockOAuthSession().get.assert_called_with(
(
"https://api.etrade.com/v1/accounts/"
"rest/12345/transactions/67890.json"
),
params={},
)
# Test Dev XML
self.assertEqual(
account.get_transaction_details(12345, 67890, resp_format="xml"),
r"<xml> returns </xml>",
)
MockOAuthSession().get.assert_called_with(
(
"https://apisb.etrade.com/v1/accounts/"
"sandbox/rest/12345/transactions/67890"
),
params={},
)
# Test Prod XML
self.assertEqual(
account.get_transaction_details(12345, 67890, dev=False, resp_format="xml"),
r"<xml> returns </xml>",
)
MockOAuthSession().get.assert_called_with(
("https://api.etrade.com/v1/accounts/" "rest/12345/transactions/67890"),
params={},
)
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
def test_get_transaction_history(self, MockOAuthSession):
'''test_get_transaction_history(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions'''
# Set Mock returns
MockOAuthSession().get(
).json.return_value = "{'transaction': 'abc123'}"
MockOAuthSession().get().text = r'<xml> returns </xml>'
account = accounts.ETradeAccounts('abc123', 'xyz123', 'abctoken',
'xyzsecret')
# Test Dev JSON
self.assertEqual(account.get_transaction_history(12345),
"{'transaction': 'abc123'}")
# Test API URL
MockOAuthSession().get.assert_called_with(
('https://etwssandbox.etrade.com/accounts/'
'sandbox/rest/12345/transactions.json'),
params={})
# Test Prod JSON
self.assertEqual(account.get_transaction_history(12345, dev=False),
"{'transaction': 'abc123'}")
# Test API URL
MockOAuthSession().get.assert_called_with(
('https://etws.etrade.com/accounts/'
'rest/12345/transactions.json'),
params={})
# Test Dev XML
self.assertEqual(
account.get_transaction_history(12345, resp_format='xml'),
r'<xml> returns </xml>')
# Test API URL
MockOAuthSession().get.assert_called_with(
('https://etwssandbox.etrade.com/accounts/'
'sandbox/rest/12345/transactions'),
params={})
# Test Prod XML
self.assertEqual(
account.get_transaction_history(12345,
dev=False,
resp_format='xml'),
r'<xml> returns </xml>')
# Test API URL
MockOAuthSession().get.assert_called_with(
('https://etws.etrade.com/accounts/'
'rest/12345/transactions'),
params={})
# Test optional_args
self.assertEqual(
account.get_transaction_history(12345, group='WITHDRAWALS'),
"{'transaction': 'abc123'}")
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
def test_list_accounts(self, MockOAuthSession):
'''test_list_accounts(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions'''
# Set Mock returns
MockOAuthSession().get().json.return_value = "{'account': 'abc123'}"
MockOAuthSession().get().text.return_value = r'<xml> returns </xml>'
account = accounts.ETradeAccounts('abc123', 'xyz123', 'abctoken', 'xyzsecret')
# Test Dev JSON
self.assertEqual(account.list_accounts(), "{'account': 'abc123'}")
# Test Prod JSON
self.assertEqual(account.list_accounts(dev=False), "{'account': 'abc123'}")
# Test Dev XML
self.assertEqual(account.list_accounts(resp_format='xml'), r'<xml> returns </xml>')
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get().text.called)
self.assertTrue(MockOAuthSession().get.called)
def test_get_account_positions(self, MockOAuthSession):
"""test_get_account_positions(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions"""
# Set Mock returns
MockOAuthSession().get().json.return_value = "{'account': 'abc123'}"
MockOAuthSession().get().text = r"<xml> returns </xml>"
account = accounts.ETradeAccounts("abc123", "xyz123", "abctoken", "xyzsecret")
# Test Dev JSON
self.assertEqual(
account.get_account_positions("12345"), "{'account': 'abc123'}"
)
# Test API URL
MockOAuthSession().get.assert_called_with(
("https://apisb.etrade.com/v1/accounts" "/12345/portfolio.json")
)
# Test Prod JSON
self.assertEqual(
account.get_account_positions("12345", dev=False), "{'account': 'abc123'}"
)
# Test API URL
MockOAuthSession().get.assert_called_with(
("https://api.etrade.com/v1/accounts" "/12345/portfolio.json")
) # Test Dev XML
self.assertEqual(
account.get_account_positions("12345", resp_format="xml"),
r"<xml> returns </xml>",
)
MockOAuthSession().get.assert_called_with(
("https://apisb.etrade.com/v1/accounts" "/12345/portfolio")
)
self.assertEqual(
account.get_account_positions("12345", dev=False, resp_format="xml"),
r"<xml> returns </xml>",
)
MockOAuthSession().get.assert_called_with(
("https://api.etrade.com/v1/accounts" "/12345/portfolio")
)
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
def test_delete_alert(self, MockOAuthSession):
"""test_list_alerts(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions"""
# Set Mock returns
MockOAuthSession().delete().json.return_value = "{'account': 'abc123'}"
MockOAuthSession().delete().text = r"<xml> returns </xml>"
account = accounts.ETradeAccounts("abc123", "xyz123", "abctoken", "xyzsecret")
# Test Dev JSON
self.assertEqual(account.delete_alert(1234), "{'account': 'abc123'}")
# Test API URL
MockOAuthSession().delete.assert_called_with(
("https://apisb.etrade.com/v1/accounts/" "sandbox/rest/alerts/1234.json")
)
# Test Prod JSON
self.assertEqual(account.delete_alert(1234, dev=False), "{'account': 'abc123'}")
# Test API URL
MockOAuthSession().delete.assert_called_with(
("https://api.etrade.com/v1/accounts/" "rest/alerts/1234.json")
)
# Test Dev XML
self.assertEqual(
account.delete_alert(1234, resp_format="xml"), r"<xml> returns </xml>"
)
MockOAuthSession().delete.assert_called_with(
("https://apisb.etrade.com/v1/accounts/" "sandbox/rest/alerts/1234")
)
self.assertEqual(
account.delete_alert(1234, dev=False, resp_format="xml"),
r"<xml> returns </xml>",
)
MockOAuthSession().delete.assert_called_with(
("https://api.etrade.com/v1/accounts/" "rest/alerts/1234")
)
self.assertTrue(MockOAuthSession().delete().json.called)
self.assertTrue(MockOAuthSession().delete.called)
def test_list_transactions(self, MockOAuthSession):
"""test_list_transactions(MockOAuthSession) -> None
param: MockOAuthSession
type: mock.MagicMock
description: MagicMock object for OAuth1Sessions"""
# Set Mock returns
MockOAuthSession().get(
).json.return_value = "{'transaction': 'abc123'}"
MockOAuthSession().get().text = r"<xml> returns </xml>"
account = accounts.ETradeAccounts("abc123", "xyz123", "abctoken",
"xyzsecret")
# Test Dev JSON
self.assertEqual(
account.list_transactions(12345, resp_format="json"),
"{'transaction': 'abc123'}",
)
# Test API URL
MockOAuthSession().get.assert_called_with(
("https://apisb.etrade.com/v1/accounts/12345/transactions.json"),
params={},
)
# Test Dev XML
self.assertEqual(
dict(account.list_transactions(12345, resp_format="xml")),
{"xml": "returns"},
)
# Test API URL
MockOAuthSession().get.assert_called_with(
("https://apisb.etrade.com/v1/accounts/12345/transactions"),
params={},
)
account = accounts.ETradeAccounts("abc123",
"xyz123",
"abctoken",
"xyzsecret",
dev=False)
# Test Prod JSON
self.assertEqual(
account.list_transactions(12345, resp_format="json"),
"{'transaction': 'abc123'}",
)
# Test API URL
MockOAuthSession().get.assert_called_with(
("https://api.etrade.com/v1/accounts/12345/transactions.json"),
params={},
)
# Test Prod XML
self.assertEqual(
dict(account.list_transactions(12345, resp_format="xml")),
{"xml": "returns"},
)
# Test API URL
MockOAuthSession().get.assert_called_with(
("https://api.etrade.com/v1/accounts/12345/transactions"),
params={})
# Test optional_args
self.assertEqual(
account.list_transactions(12345,
group="WITHDRAWALS",
resp_format="json"),
"{'transaction': 'abc123'}",
)
self.assertTrue(MockOAuthSession().get().json.called)
self.assertTrue(MockOAuthSession().get.called)
