Skip to content

Commit 0df6121

Browse files
add error status attribute
1 parent f220bb3 commit 0df6121

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

main.py

+2
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ def __init__(self):
77
self.broker = BrokerFunctions()
88

99
self.token = None
10+
self.error = None
1011

1112
## SETUP
1213

@@ -16,6 +17,7 @@ def get_token(self, email, password, server="https://my.farm.bot"):
1617
token_data = self.api.get_token(email, password, server)
1718

1819
self.token = token_data
20+
self.error = self.api.api_connect.error
1921

2022
# Set broker tokens
2123
self.broker.broker_connect.token = self.api.api_connect.token

testing.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def test_get_token_bad_email(self, mock_post):
6262
headers={'content-type': 'application/json'},
6363
json={'user': {'email': '[email protected]', 'password': 'test_pass_123'}}
6464
)
65-
# self.assertEqual(fb.error, 'ERROR: Incorrect email address or password.')
65+
self.assertEqual(fb.error, 'HTTP ERROR: Incorrect email address or password.')
6666
self.assertIsNone(fb.token)
6767
self.assertEqual(mock_post.return_value.status_code, 422)
6868

@@ -80,7 +80,7 @@ def test_get_token_bad_server(self, mock_post):
8080
headers={'content-type': 'application/json'},
8181
json={'user': {'email': '[email protected]', 'password': 'test_pass_123'}}
8282
)
83-
# self.assertEqual(fb.error, 'ERROR: The server address does not exist.')
83+
self.assertEqual(fb.error, 'HTTP ERROR: The server address does not exist.')
8484
self.assertIsNone(fb.token)
8585
self.assertEqual(mock_post.return_value.status_code, 404)
8686

0 commit comments

Comments
 (0)