From 2f429fcb86e0d470bc983327230b18b31c2a2d22 Mon Sep 17 00:00:00 2001 From: Viraat Chandra Date: Mon, 1 Jan 2018 15:13:05 +0600 Subject: [PATCH] zulip_bots: Add test for `create-plan` command of Baremetrics bot. --- .../bots/baremetrics/baremetrics.py | 16 +++-- .../baremetrics/fixtures/create_plan.json | 64 +++++++++++++++++++ .../bots/baremetrics/test_baremetrics.py | 23 +++++-- 3 files changed, 90 insertions(+), 13 deletions(-) create mode 100644 zulip_bots/zulip_bots/bots/baremetrics/fixtures/create_plan.json diff --git a/zulip_bots/zulip_bots/bots/baremetrics/baremetrics.py b/zulip_bots/zulip_bots/bots/baremetrics/baremetrics.py index 3d248e8..0ebfc9f 100644 --- a/zulip_bots/zulip_bots/bots/baremetrics/baremetrics.py +++ b/zulip_bots/zulip_bots/bots/baremetrics/baremetrics.py @@ -2,7 +2,6 @@ from typing import Any, List import requests -import logging class BaremetricsHandler(object): def initialize(self, bot_handler: Any) -> None: @@ -13,11 +12,14 @@ class BaremetricsHandler(object): 'Authorization': 'Bearer ' + self.api_key } - self.commands = ['help', 'list-commands', 'account-info', 'list-sources', 'list-plans ', - 'list-customers ', 'list-subscriptions ', 'create-plan ' - ' ' - ' ' - ''] + self.commands = ['help', + 'list-commands', + 'account-info', + 'list-sources', + 'list-plans ', + 'list-customers ', + 'list-subscriptions ', + 'create-plan '] self.descriptions = ['Display bot info', 'Display the list of available commands', 'Display the account info', 'List the sources', 'List the plans for the source', 'List the customers in the source', @@ -199,7 +201,7 @@ class BaremetricsHandler(object): 'amount': int(parameters[4]), 'interval': parameters[5], 'interval_count': int(parameters[6]) - } + } # type: Any url = 'https://api.baremetrics.com/v1/{}/plans'.format(parameters[0]) create_plan_response = requests.post(url, data=data_header, headers=self.auth_header) diff --git a/zulip_bots/zulip_bots/bots/baremetrics/fixtures/create_plan.json b/zulip_bots/zulip_bots/bots/baremetrics/fixtures/create_plan.json new file mode 100644 index 0000000..b491fc5 --- /dev/null +++ b/zulip_bots/zulip_bots/bots/baremetrics/fixtures/create_plan.json @@ -0,0 +1,64 @@ +{ + "request": { + "method": "POST", + "api_url": "https://api.baremetrics.com/v1/TEST/plans", + "headers": { + "Authorization": "Bearer TEST" + }, + "data": { + "oid": "1", + "name": "TEST", + "currency": "USD", + "amount": 123, + "interval": "TEST", + "interval_count": 123 + } + }, + "response": { + "plan": { + "oid": "1", + "source_id": "TEST", + "source": "TEST", + "name": "TEST", + "interval": "TEST", + "interval_count": 123, + "trial_duration": null, + "trial_duration_unit": null, + "created": null, + "active": true, + "setup_fees": 0, + "amounts": [ + { + "currency": "USD", + "symbol": "$", + "symbol_right": false, + "amount": 1 + } + ] + } + }, + "response-headers": { + "X-TokenExpires": "0", + "Server": "cloudflare-nginx", + "X-RateLimit-Remaining": "3593", + "X-Powered-By": "Phusion Passenger 5.0.30", + "Set-Cookie": "__cfduid=dd26464e3e6779b6a05b27d2681aea0851514374567; expires=Thu, 27-Dec-18 11:36:07 GMT; path=/; domain=.baremetrics.com; HttpOnly; Secure, LSW_WEB=\"LSW_WEB1\"; path=/", + "X-Runtime": "0.031730", + "Access-Control-Allow-Credentials": "false", + "Access-Control-Allow-Methods": "GET, OPTIONS, POST, PUT, DELETE", + "Content-Encoding": "gzip", + "Connection": "keep-alive", + "ETag": "W/\"110de39b319ed195bf1414ac4f3c9a01\"", + "Date": "Wed, 27 Dec 2017 11:36:08 GMT", + "X-RateLimit-Limit": "3600", + "X-Version": "721", + "X-Commit": "2dcec7190a2bd01c8d3d8527d386486c4bb5474c", + "Status": "200 OK", + "Content-Type": "application/json; charset=utf-8", + "Transfer-Encoding": "chunked", + "Access-Control-Allow-Headers": "Authorization,DNT,X-CustomHeader,Keep-Alive,User-Agent,X-Requested-With,If-Modified-Since,Cache-Control,Content-Type", + "Cache-Control": "max-age=0, private, must-revalidate", + "X-Request-Id": "0b292131-81bb-4df2-bdb9-089a50b17e63", + "CF-RAY": "3d3bfaf6feee2eff-DEL" + } +} diff --git a/zulip_bots/zulip_bots/bots/baremetrics/test_baremetrics.py b/zulip_bots/zulip_bots/bots/baremetrics/test_baremetrics.py index 49a3092..951faae 100644 --- a/zulip_bots/zulip_bots/bots/baremetrics/test_baremetrics.py +++ b/zulip_bots/zulip_bots/bots/baremetrics/test_baremetrics.py @@ -24,12 +24,17 @@ class TestBaremetricsBot(BotTestCase): 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 ' - ' : List the plans for the source\n - list-customers ' - ' : List the customers in the source\n - list-subscriptions ' - ' : List the subscriptions in the source\n') + 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 : List the plans for the source\n' + ' - list-customers : List the customers in the source\n' + ' - list-subscriptions : List the subscriptions in the ' + 'source\n' + ' - create-plan ' + ' : Create a plan in the given source\n') def test_account_info_command(self) -> None: with self.mock_config_info({'api_key': 'TEST'}): @@ -91,3 +96,9 @@ class TestBaremetricsBot(BotTestCase): patch('requests.get'): with self.mock_http_conversation('test_key_error'): self.verify_reply('list-plans TEST', 'Invalid Response From API.') + + def test_create_plan_command(self) -> None: + with self.mock_config_info({'api_key': 'TEST'}), \ + patch('requests.get'): + with self.mock_http_conversation('create_plan'): + self.verify_reply('create-plan TEST 1 TEST USD 123 TEST 123', 'Plan Created.')