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 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()