69 lines
3.7 KiB
Python
69 lines
3.7 KiB
Python
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'}), \
|
|
patch('requests.get'):
|
|
self.verify_reply('', 'No Command Specified')
|
|
|
|
def test_help_query(self) -> None:
|
|
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
|
|
Enter `list-commands` to show the list of available commands.
|
|
Version 1.0
|
|
''')
|
|
|
|
def test_list_commands_command(self) -> None:
|
|
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 '
|
|
'<source_id> : List the plans for the source\n - list-customers '
|
|
'<source_id> : List the customers in the source\n - list-subscriptions '
|
|
'<source_id> : List the subscriptions in the source\n')
|
|
|
|
def test_account_info_command(self) -> None:
|
|
with self.mock_config_info({'api_key': 'TEST'}):
|
|
with self.mock_http_conversation('account_info'):
|
|
self.verify_reply('account-info', '**Your account information:** \nId: 376418\nCompany: NA\nDefault '
|
|
'Currency: United States Dollar')
|
|
|
|
def test_list_sources_command(self) -> None:
|
|
with self.mock_config_info({'api_key': 'TEST'}):
|
|
with self.mock_http_conversation('list_sources'):
|
|
self.verify_reply('list-sources', '**Listing sources:** \n1.ID: 5f7QC5NC0Ywgcu\nProvider: '
|
|
'baremetrics\nProvider ID: None\n\n')
|
|
|
|
def test_list_plans_command(self) -> None:
|
|
r = '**Listing plans:** \n1.Name: Plan 1\nActive: True\nInterval: year\nInterval Count: 1\nAmounts: \n' \
|
|
' - 450000 USD\n\n2.Name: Plan 2\nActive: True\nInterval: year\nInterval Count: 1\nAmounts: \n' \
|
|
' - 450000 USD\n\n'
|
|
|
|
with self.mock_config_info({'api_key': 'TEST'}):
|
|
with self.mock_http_conversation('list_plans'):
|
|
self.verify_reply('list-plans TEST', r)
|
|
|
|
def test_list_customers_command(self) -> None:
|
|
r = '**Listing customers:** \n1.Name: Customer 1\nDisplay Name: Customer 1\nOID: customer_1\nActive: True\n' \
|
|
'Email: customer_1@baremetrics.com\nNotes: Here are some notes\nCurrent Plans: \n - Plan 1\n\n'
|
|
|
|
with self.mock_config_info({'api_key': 'TEST'}):
|
|
with self.mock_http_conversation('list_customers'):
|
|
self.verify_reply('list-customers TEST', r)
|
|
|
|
def test_list_subscriptions_command(self) -> None:
|
|
r = '**Listing subscriptions:** \n1.Customer Name: Customer 1\nCustomer Display Name: Customer 1\n' \
|
|
'Customer OID: customer_1\nCustomer Email: customer_1@baremetrics.com\nActive: True\n' \
|
|
'Plan Name: Plan 1\nPlan Amounts: \n - 1000 $\n\n'
|
|
|
|
with self.mock_config_info({'api_key': 'TEST'}):
|
|
with self.mock_http_conversation('list_subscriptions'):
|
|
self.verify_reply('list-subscriptions TEST', r)
|