def test_init(self): arguments = MockArgumentParser() client = RestBotClient("testrest", arguments) self.assertIsNotNone(client) self.assertIsNotNone(client.get_client_configuration()) self.assertIsInstance(client.get_client_configuration(), RestConfiguration) self.assertEquals([], client.api_keys) request = unittest.mock.Mock() response, code = client.process_request(request)
def test_verify_api_key_usage(self): arguments = MockArgumentParser() client = RestBotClient("testrest", 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 + "api_keys.txt" client.initialise() request = MockRequest() request.args['apikey'] = '11111111' self.assertTrue(client.api_keys.verify_api_key_usage(request))
def test_api_keys(self): arguments = MockArgumentParser() client = RestBotClient("testrest", 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 + "api_keys.txt" client.initialise() self.assertEqual(3, len(client.api_keys.api_keys)) self.assertTrue(client.api_keys.is_apikey_valid("11111111")) self.assertTrue(client.api_keys.is_apikey_valid("22222222")) self.assertTrue(client.api_keys.is_apikey_valid("33333333")) self.assertFalse(client.api_keys.is_apikey_valid("99999999"))
def __init__(self, id, argument_parser=None): RestBotClient.__init__(self, id, argument_parser) self.initialise() bot_name = "defaultbot" try: bot = self._bot_factory.select_bot() brain = bot._brain_factory.select_brain() bot_name = brain.configuration.bot_name YLogger.set_prefix(bot_name) except Exception: pass self._server_mode = True # if Not Server-Mode then change to comment line. YLogger.set_stdout(self.arguments.args.stdoutlog) YLogger.set_traceback(False)
def test_unsupported_authorisation(self): arguments = MockArgumentParser() client = RestBotClient("testrest", arguments) self.assertIsNotNone(client) client.configuration.client_configuration._authorization = "Sha" handler = RestAuthorizationHandler.load_authorisation(client) self.assertIsNone(handler)
def test_basic_authorisation(self): arguments = MockArgumentParser() client = RestBotClient("testrest", arguments) self.assertIsNotNone(client) client.license_keys.add_key('BASIC_AUTH_TOKEN', "1234567890") client.configuration.client_configuration._authorization = "Basic" handler = RestAuthorizationHandler.load_authorisation(client) self.assertIsNotNone(handler)
def test_init(self): config = RestConfiguration("test") self.assertIsNotNone(config) handler = MockRestBasicAuthorizationHandler(config) self.assertIsNotNone(handler) handler.auth_token = "1234567890" arguments = MockArgumentParser() client = RestBotClient("testrest", arguments) self.assertIsNotNone(client) client.license_keys.add_key('BASIC_AUTH_TOKEN', "1234567890") handler.initialise(client) self.assertEquals("1234567890", handler._basic_auth_token)
def test_authorise_invalid_header(self): config = RestConfiguration("test") self.assertIsNotNone(config) handler = MockRestBasicAuthorizationHandler(config) self.assertIsNotNone(handler) handler.auth_token = "1234567890" arguments = MockArgumentParser() client = RestBotClient("testrest", arguments) self.assertIsNotNone(client) client.license_keys.add_key('BASIC_AUTH_TOKEN', "1234567890") handler.initialise(client) request = MockRequest() request.headers['AUTHORISATION'] = 'Basic 0123456789' self.assertFalse(handler.authorise(request))
def test_api_keys(self): arguments = MockArgumentParser() client = RestBotClient("testrest", 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 + "api_keys.txt" client.load_api_keys() self.assertEquals(3, len(client.api_keys)) self.assertTrue(client.is_apikey_valid("11111111")) self.assertTrue(client.is_apikey_valid("22222222")) self.assertTrue(client.is_apikey_valid("33333333")) self.assertFalse(client.is_apikey_valid("99999999"))
def __init__(self, id, argument_parser=None): RestBotClient.__init__(self, id, argument_parser) self.initialise()
def __init__(self, id, argument_parser=None): self._description = "Programy AIML2.x Sanic REST Client" RestBotClient.__init__(self, id, argument_parser)
def __init__(self, id, argument_parser=None): RestBotClient.__init__(self, id, argument_parser) self.preprocesser = PreProcessor( fpath="../../storage/lookups/normal.txt")
def __init__(self, id, argument_parser=None): RestBotClient.__init__(self, id, argument_parser)
def __init__(self, id, argument_parser=None): c = MockArgumentParserGiseldo() RestBotClient.__init__(self, id, c) self.initialise()
def __init__(self, botid, argument_parser=None): RestBotClient.__init__(self, botid, argument_parser)