xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

baremetrics bot: Quit on invalid API key.

Browse source
This commit is contained in:
Robert Hönig 2018-01-04 16:14:35 +01:00 committed by showell
parent c86e62a06a
commit f6f09202ac
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
zulip_bots/zulip_bots/bots/baremetrics/baremetrics.py
View file

@ -21,16 +21,16 @@ class BaremetricsHandler(object):
'List the sources', 'List the plans for the source', 'List the customers in the source',
'List the subscriptions in the source']
self.check_api_key()
self.check_api_key(bot_handler)
def check_api_key(self) -> None:
def check_api_key(self, bot_handler) -> None:
url = "https://api.baremetrics.com/v1/account"
test_query_response = requests.get(url, headers=self.auth_header)
test_query_data = test_query_response.json()
try:
if test_query_data['error'] == "Unauthorized. Token not found":
logging.error('API Key not valid. Please see doc.md to find out how to get it.')
bot_handler.quit('API Key not valid. Please see doc.md to find out how to get it.')
except KeyError:
pass

10
zulip_bots/zulip_bots/bots/baremetrics/test_baremetrics.py
View file

@ -1,14 +1,17 @@
from unittest.mock import patch
from zulip_bots.test_lib import BotTestCase
class TestBaremetricsBot(BotTestCase):
bot_name = "baremetrics"
def test_bot_responds_to_empty_message(self) -> None:
with self.mock_config_info({'api_key': 'TEST'}):
with self.mock_config_info({'api_key': 'TEST'}), \
patch('requests.get'):
self.verify_reply('', 'No Command Specified')
def test_help_query(self) -> None:
with self.mock_config_info({'api_key': 'TEST'}):
with self.mock_config_info({'api_key': 'TEST'}), \
patch('requests.get'):
self.verify_reply('help', '''
This bot gives updates about customer behavior, financial performance, and analytics
for an organization using the Baremetrics Api.\n
@ -17,7 +20,8 @@ class TestBaremetricsBot(BotTestCase):
''')
def test_list_commands_command(self) -> None:
with self.mock_config_info({'api_key': 'TEST'}):
with self.mock_config_info({'api_key': 'TEST'}), \
patch('requests.get'):
self.verify_reply('list-commands', '**Available Commands:** \n - help : Display bot info\n - list-commands '
': Display the list of available commands\n - account-info : Display '
'the account info\n - list-sources : List the sources\n - list-plans '