def mock_request_handler(ip=None, proxy_username=None, auth_username=None, proxied_ip=None): handler_mock = mock_object() handler_mock.application = mock_object() handler_mock.application.identification = mock_object() handler_mock.application.identification.identify_for_audit = lambda x: auth_username handler_mock.request = mock_object() handler_mock.request.headers = {} if proxy_username: credentials = proxy_username + ':pwd' credentials_base64 = base64.encodebytes(credentials.encode('utf-8')) handler_mock.request.headers[ 'Authorization'] = 'Basic ' + credentials_base64.decode('utf-8') handler_mock.request.remote_ip = ip if proxied_ip: handler_mock.request.headers['X-Forwarded-For'] = proxied_ip return handler_mock
def mock_request_handler(ip=None, x_forwarded_for=None, x_real_ip=None, saved_token=None, user_header_name=None, user_header_name_value=None): handler_mock = mock_object() handler_mock.application = mock_object() handler_mock.application.auth = TornadoAuth(None) handler_mock.application.identification = IpBasedIdentification( ['127.0.0.1'], user_header_name) handler_mock.request = mock_object() handler_mock.request.headers = {} handler_mock.request.remote_ip = ip if x_forwarded_for: handler_mock.request.headers['X-Forwarded-For'] = x_forwarded_for if x_real_ip: handler_mock.request.headers['X-Real-IP'] = x_real_ip if user_header_name and user_header_name_value: handler_mock.request.headers[user_header_name] = user_header_name_value cookies = {COOKIE_KEY: saved_token} def get_secure_cookie(name): values = cookies[name] if values is not None: return values.encode('utf8') return None def set_secure_cookie(key, value, expires_days=30): cookies[key] = value def clear_cookie(key): if cookies[key] is not None: cookies[key] = None handler_mock.get_cookie = lambda key: cookies[key] handler_mock.get_secure_cookie = get_secure_cookie handler_mock.set_secure_cookie = set_secure_cookie handler_mock.clear_cookie = clear_cookie return handler_mock
def create_execution_service(self): file_download_feature = mock_object() file_download_feature.is_downloadable = lambda x: False execution_service = ExecutionService(self.id_generator) self.exec_services.append(execution_service) return execution_service
def mock_request_handler(ip=None, proxy_username=None, auth_username=None, proxied_ip=None): handler_mock = mock_object() handler_mock.application = mock_object() handler_mock.application.auth = mock_object() handler_mock.application.auth.get_username = lambda x: auth_username handler_mock.request = mock_object() handler_mock.request.headers = {} if proxy_username: credentials = proxy_username + ':pwd' credentials_base64 = base64.encodebytes(credentials.encode('utf-8')) handler_mock.request.headers['Authorization'] = 'Basic ' + credentials_base64.decode('utf-8') handler_mock.request.remote_ip = ip if proxied_ip: handler_mock.request.headers['X-Forwarded-For'] = proxied_ip return handler_mock
def _mock_request_handler(username, password): request_handler = mock_object() def get_argument(arg_name): if arg_name == 'username': return username elif arg_name == 'password': return password else: return None request_handler.get_argument = get_argument return request_handler
def mock_request_handler(code): handler_mock = mock_object() handler_mock.get_argument = lambda arg, default: code if arg == 'code' else None secure_cookies = {} handler_mock.get_secure_cookie = lambda cookie: secure_cookies.get(cookie) def set_secure_cookie(cookie, value): secure_cookies[cookie] = value def clear_secure_cookie(cookie): if cookie in secure_cookies: del secure_cookies[cookie] handler_mock.set_secure_cookie = set_secure_cookie handler_mock.clear_cookie = clear_secure_cookie return handler_mock
def run_initialize(self): module_mock = mock_object() module_mock.Curl = '' with mock.patch.dict(sys.modules, {'pycurl': module_mock}): tornado_client_config.initialize()