def test_rest_client_init(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("flask", arguments)
self.assertIsNotNone(client)
self.assertFalse(client._render_callback())
self.assertIsInstance(client.renderer, TextRenderer)
def test_verify_api_key_usage_inactive(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
client.configuration.client_configuration._use_api_keys = False
request = unittest.mock.Mock()
self.assertEquals((None, None),client.verify_api_key_usage(request))
def test_verify_api_key_usage_inactive(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
client.configuration.client_configuration._use_api_keys = False
request = unittest.mock.Mock()
self.assertEqual((None, None),
client.verify_api_key_usage(request))
def test_get_api_key(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['apikey'] = '11111111'
self.assertEquals('11111111', client.get_api_key(request))
def test_get_api_key(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['apikey'] = '11111111'
self.assertEqual('11111111', client.get_api_key(request))
def test_format_success_response(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
response = client.format_success_response("1234567890", "Hello", "Hi")
self.assertIsNotNone(response)
self.assertEquals("1234567890", response['userid'])
self.assertEquals("Hello", response['question'])
self.assertEquals("Hi", response['answer'])
def test_get_userid(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['userid'] = '1234567890'
self.assertEquals("1234567890", client.get_userid(request))
def test_get_question(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['question'] = 'Hello'
self.assertEquals("Hello", client.get_question(request))
def test_get_question(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['question'] = 'Hello'
self.assertEqual("Hello", client.get_question(request))
def test_get_userid(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['userid'] = '1234567890'
self.assertEqual("1234567890", client.get_userid(request))
def test_format_success_response(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
response = client.format_success_response("1234567890", "Hello",
"Hi")
self.assertIsNotNone(response)
self.assertEqual("1234567890", response['userid'])
self.assertEqual("Hello", response['question'])
self.assertEqual("Hi", response['answer'])
def test_verify_api_key_usage_active(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
client.configuration.client_configuration._use_api_keys = True
client.configuration.client_configuration._api_key_file = os.path.dirname(__file__) + os.sep + ".." + os.sep + ".." + os.sep + "api_keys.txt"
client.load_api_keys()
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['apikey'] = '11111111'
self.assertEquals((None, None),client.verify_api_key_usage(request))
def test_format_error_response(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
response = client.format_error_response("1234567890", "Hello",
"Something Bad")
self.assertIsNotNone(response)
self.assertEquals("1234567890", response['userid'])
self.assertEquals("Hello", response['question'])
self.assertEquals("", response['answer'])
self.assertEquals("Something Bad", response['error'])
def test_verify_api_key_usage_active(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("flask", arguments)
self.assertIsNotNone(client)
client.configuration.client_configuration._use_api_keys = True
client.configuration.client_configuration._api_key_file = os.path.dirname(
__file__
) + os.sep + ".." + os.sep + ".." + os.sep + "api_keys.txt"
client.initialise()
request = unittest.mock.Mock()
request.args = {}
request.args['apikey'] = '11111111'
self.assertTrue(client.api_keys.verify_api_key_usage(request))
def test_verify_api_key_usage_active(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
client.configuration.client_configuration._use_api_keys = True
client.configuration.client_configuration._api_key_file = os.path.dirname(
__file__
) + os.sep + ".." + os.sep + ".." + os.sep + "api_keys.txt"
client.load_api_keys()
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['apikey'] = '11111111'
self.assertEquals((None, None),
client.verify_api_key_usage(request))
def test_format_success_response(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
request = unittest.mock.Mock()
request.raw_args = {}
request.raw_args['userid'] = '1234567890'
request.raw_args['question'] = 'Hello'
userInfo = UserInfo(client, request)
response = client.format_success_response("1234567890", "Hello",
"Hi", userInfo)
self.assertIsNotNone(response)
self.assertEqual("1234567890", response['userId'])
self.assertEqual("Hello", response['utterance'])
self.assertEqual("Hi", response['response'])
self.assertEqual("", response['topic'])
def __init__(self, argument_parser=None):
SanicRestBotClient.__init__(self, "sanic", argument_parser)
self.aborted = False
self.answer = None
self.ask_question_exception = False
def test_rest_client_init(self):
arguments = MockArgumentParser()
client = SanicRestBotClient("sanic", arguments)
self.assertIsNotNone(client)
def __init__(self, argument_parser=None):
SanicRestBotClient.__init__(self, "sanic", argument_parser)
self.aborted = False
self.answer = None
self.ask_question_exception = False
