def test_verify_api_key_usage_inactive(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", 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_process_debug_request_not_server_mode(self): home_dir = os.path.dirname(__file__) + os.sep + "testdata" tmp_dir = home_dir + os.sep + "tmp" if os.path.exists(tmp_dir): shutil.rmtree(tmp_dir) self.assertFalse(os.path.exists(tmp_dir)) config_file = home_dir + os.sep + "config.yaml" arguments = MockArgumentParser(config=config_file) client = FlaskYadlanClient("testrest", arguments) client._server_mode = False self.assertIsNotNone(client) request = unittest.mock.Mock() request.data = b'{"userId": "1234567890", "utterance": "Hello"}' client.process_request(request) request = unittest.mock.Mock() request.data = b'{"userId": "1234567890", "variables": [{"type": "name", "key": "testname", "value": "value1"}]}' debugInfo, status = client.process_debug_request(request) self.assertEqual(4, len(debugInfo)) self.assertEqual(200, status) if os.path.exists(tmp_dir): shutil.rmtree(tmp_dir) self.assertFalse(os.path.exists(tmp_dir))
def test_get_api_key(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) request = unittest.mock.Mock() request.args = {} request.args['apikey'] = '11111111' self.assertEqual('11111111', client.get_api_key(request))
def test_get_userid(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) self.assertIsNotNone(client) request = unittest.mock.Mock() request.data = b'{"userId": "1234567890"}' self.assertEqual("1234567890", client.get_userid(request))
def test_process_debug_request_invalid_variables_parameter(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) self.assertIsNotNone(client) request = unittest.mock.Mock() request.data = b'{"userId": "1234567890", "variables": [{"var": "test", "value": "value1"}]}' debugInfo, status = client.process_debug_request(request) self.assertEqual('Invalid variables list format', debugInfo['error']) self.assertEqual(400, status)
def test_process_debug_request_no_conversation(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) self.assertIsNotNone(client) request = unittest.mock.Mock() request.data = b'{"userId": "1234567890", "variables": [{"type": "name", "key": "testname", "value": "value1"}]}' debugInfo, status = client.process_debug_request(request) self.assertEqual(0, len(debugInfo)) self.assertEqual(200, status)
def test_verify_api_key_usage_active(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", 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.args = {} request.args['apikey'] = '11111111' self.assertEqual(({'error': 'Unauthorized access'}, 401), client.verify_api_key_usage(request))
def test_process_request_no_question(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) self.assertIsNotNone(client) request = unittest.mock.Mock() request.args = {} request.data = b'{"userId": "1234567890"}' response, status = client.process_request(request) self.assertIsNotNone(response) self.assertEqual(status, 400)
def test_process_debug_request_with_variables(self): home_dir = os.path.dirname(__file__) + os.sep + "testdata" tmp_dir = home_dir + os.sep + "tmp" errors_file = tmp_dir + os.sep + "errors.txt" duplicates_file = tmp_dir + os.sep + "duplicates.txt" conversation_file = tmp_dir + os.sep + "testrest_testUser.conv" logs_file = tmp_dir + os.sep + "testrest_testUser.log" if os.path.exists(tmp_dir): shutil.rmtree(tmp_dir) self.assertFalse(os.path.exists(tmp_dir)) config_file = home_dir + os.sep + "config.yaml" arguments = MockArgumentParser(config=config_file) client = FlaskYadlanClient("testrest", arguments) self.assertIsNotNone(client) request = unittest.mock.Mock() request.data = b'{"userId": "testUser", "utterance": "VARIABLES"}' response, _ = client.process_request(request) self.assertIsNotNone(response) self.assertEqual("testUser", response['userId']) self.assertEqual("VARIABLES.", response['utterance']) self.assertEqual("Variable datas.", response['response']) self.assertTrue(os.path.exists(errors_file)) self.assertTrue(os.path.exists(duplicates_file)) self.assertTrue(os.path.exists(conversation_file)) self.assertTrue(os.path.exists(logs_file)) request = unittest.mock.Mock() request.data = b'{"userId": "testUser"}' debug_info, _ = client.process_debug_request(request) self.assertTrue('errors' in debug_info) self.assertTrue('duplicates' in debug_info) self.assertTrue('conversations' in debug_info) self.assertTrue('logs' in debug_info) self.assertTrue('current_conversation' in debug_info) self.assertEqual("name_value", debug_info['conversations']['properties']['name_key']) self.assertEqual( "data_value", debug_info['conversations']['data_properties']['data_key']) question = debug_info['conversations']['questions'][0] self.assertEqual("var_value", question['var_properties']['var_key']) if os.path.exists(tmp_dir): shutil.rmtree(tmp_dir) self.assertFalse(os.path.exists(tmp_dir))
def test_ask_question(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) self.assertIsNotNone(client) client.configuration.client_configuration._debug = True request = unittest.mock.Mock() request.data = b'{"userId": "1234567890", "utterance": "Hello"}' response, _ = client.process_request(request) self.assertIsNotNone(response) self.assertEqual("1234567890", response['userId']) self.assertEqual("Hello.", response['utterance']) self.assertEqual("", response['response'])
def test_format_success_response(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) self.assertIsNotNone(client) request = unittest.mock.Mock() request.data = b'{"userId": "1234567890"}' 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("None", response['topic'])
def test_process_debug_request_no_userid(self): home_dir = os.path.dirname(__file__) + os.sep + "testdata" tmp_dir = home_dir + os.sep + "tmp" errors_file = tmp_dir + os.sep + "errors.txt" duplicates_file = tmp_dir + os.sep + "duplicates.txt" conversation_file = tmp_dir + os.sep + "testrest_testUser.conv" logs_file = tmp_dir + os.sep + "testrest_testUser.log" if os.path.exists(tmp_dir): shutil.rmtree(tmp_dir) self.assertFalse(os.path.exists(tmp_dir)) config_file = home_dir + os.sep + "config.yaml" arguments = MockArgumentParser(config=config_file) client = FlaskYadlanClient("testrest", arguments) self.assertIsNotNone(client) request = unittest.mock.Mock() request.data = b'{"userId": "testUser", "utterance": "Hello"}' response, _ = client.process_request(request) self.assertIsNotNone(response) self.assertEqual("testUser", response['userId']) self.assertEqual("Hello.", response['utterance']) self.assertEqual("HELLO, WORLD.", response['response']) self.assertTrue(os.path.exists(errors_file)) self.assertTrue(os.path.exists(duplicates_file)) self.assertTrue(os.path.exists(conversation_file)) self.assertTrue(os.path.exists(logs_file)) request = unittest.mock.Mock() request.data = b'{}' debug_info, _ = client.process_debug_request(request) self.assertTrue('errors' in debug_info) self.assertTrue('duplicates' in debug_info) self.assertFalse('conversations' in debug_info) self.assertFalse('logs' in debug_info) self.assertFalse('current_conversation' in debug_info) if os.path.exists(tmp_dir): shutil.rmtree(tmp_dir) self.assertFalse(os.path.exists(tmp_dir))
def test_dump_request(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) self.assertIsNotNone(client) client.configuration.client_configuration._debug = True request = unittest.mock.Mock() response_data = "hello" latency = 1.0 request.data = b'{"userId": "1234567890", "utterance": "Hello"}' client.dump_request_response(request, response_data, latency) request.data = b'' client.dump_request_response(request, response_data, latency) request.data = b'test' client.dump_request_response(request, response_data, latency)
def test_rest_client_init(self): arguments = MockArgumentParser() client = FlaskYadlanClient("yadlan", arguments) self.assertIsNotNone(client)