def api_host_matcher(request): if (request._request.headers.get('Authorization', None) != ACCESS_TOKEN): return requests_mock.create_response( request, status_code=401, reason='Unauthorized' ) path = '%s-%s' % (request._request.method, request.path.lower()) result = FAKER_DATA.get(path, None) if not result: return requests_mock.create_response( request, status_code=404, reason='Not Found' ) return requests_mock.create_response(request, json=result)
def requests_get_mock(*args, **kwargs): self.assertEquals(kwargs.get('timeout'), 5) with requests.sessions.Session() as session: from requests.packages import urllib3 urllib3.disable_warnings() return requests_mock.create_response( session.request('GET', args[0]), json={})
def auth_host_matcher(request): return requests_mock.create_response( request, json={ 'access_token': { 'id': ACCESS_TOKEN, 'ttl': 7200, 'userId': USER_ID }, 'token_type': 'Bearer' })
def match(req): conversation = getattr(m, "conversation", {}) it = iter(conversation.get(testname, [])) lu = make_comparable(req.text) for kind, envel in it: if kind == "request": cand = make_comparable(envel) if lu == cand: # reply with next response we find for kind, envel in it: if kind == "response": return requests_mock.create_response(req, text=envel) return None
def token_matcher(request): """Match a fetch token request.""" if request.url != ACCESS_TOKEN_URL: return None assert dict(parse_qsl(request.text)) == { 'client_id': 'TEST-CLIENT-ID', 'client_secret': 'TEST-CLIENT-SECRET', 'code': 'TEST-CODE', 'grant_type': 'authorization_code' } return rmock.create_response(request, json={ 'access_token': 'TEST-ACCESS-TOKEN' })
def get_by_id(self, request, id): obj = self.model.get_by_id(id) if obj: return create_response(request, json=obj) return create_response(request, status_code=404)
def create(self, request): obj = self._schema_draft().loads(request.body) data = self.model.add(obj) return create_response(request, json=data)
def _session_data(self, session_type, request): """ Callback for POST/GET to /rest/session """ LOG.debug(session_type) session_data = { "saml_alias": None, "csrf_token": "79945884c2e6f2339cbffbbaba01f17b", "user_lname": "Doe", "user_id": 1, "is_ldap": False, "is_saml": False, "orgs": [{ "city": "AnyCity", "addr": None, "zip": None, "has_available_twofactor": False, "perms": { "create_shared_layout": True, "administrator": False, "create_incs": True, "master_administrator": True, "observer": False }, "supports_ldap": False, "enabled": True, "twofactor_auth_domain": None, "attachments_enabled": True, "has_saml": False, "state": None, "addr2": None, "twofactor_cookie_lifetime_secs": 0, "require_saml": False, "tasks_private": False, "authorized_ldap_group": None, "id": 201, "name": self.org_name }, { "city": None, "addr": None, "zip": None, "has_available_twofactor": False, "perms": { "create_shared_layout": True, "administrator": False, "create_incs": True, "master_administrator": True, "observer": False }, "supports_ldap": False, "enabled": True, "twofactor_auth_domain": None, "attachments_enabled": True, "has_saml": False, "state": None, "addr2": None, "twofactor_cookie_lifetime_secs": 0, "require_saml": False, "tasks_private": False, "authorized_ldap_group": None, "id": 202, "name": "Mock Org" }], "session_ip": "192.168.56.1", "user_fname": "John", "user_email": self.email } cookies = {'JSESSIONID': 'FakeSessionId'} return requests_mock.create_response( request, status_code=200, cookies=requests.cookies.cookiejar_from_dict(cookies), json=session_data)
def _send_response(self, request, filename): file_path = os.path.join(str(self.datadir), filename) with open(file_path, mode='rb') as f: return requests_mock.create_response(request, content=f.read())
def org_get(self, request): """ Callback for GET to /orgs/<org_id> """ LOG.debug("org_get") return requests_mock.create_response(request, status_code=200, json={"actions_framework_enabled": True})
def create_response(self, **kwargs): req = requests_mock.adapter._RequestObjectProxy._create( 'post', fcmclient.api.FCM_URL, {}) return requests_mock.create_response(req, **kwargs)
def artifacts_get(self, request): """ GET the list of artifacts """ data = [value for id, value in self.artifacts.items()] return requests_mock.create_response(request, status_code=200, json=data)
def tasks_get_with_parameters(self, request): """ GET tasks with parameters text_content_output_format and handle_format """ return requests_mock.create_response(request, status_code=200, content=six.b(json.dumps(task.get_mocked_task())))
def match_all(request): return requests_mock.create_response(request, content=six.b('data'))
def action_fields_get(self, request): """ Callback for GET to /orgs/<org_id>/types/actioninvocation/fields """ LOG.debug("action_fields_get") return requests_mock.create_response(request, status_code=200, json={})
def incident_fields_get(self, request): """ Callback for GET to /orgs/<org_id>/types/incident/fields """ LOG.debug("incident_fields_get") return requests_mock.create_response(request, status_code=200, json={})
def get_by_key(self, request, key): obj = self.model.get_by_key(key) if obj: return create_response(request, json=obj) return create_response(request, status_code=404)
def incident_patch(self, request): """ Callback for PATCH to /orgs/<org_id>/incidents/<inc_id> """ LOG.info("incident_patch") return requests_mock.create_response(request, status_code=200, json=IncidentMock.incident_2)
def _validate_resource_version(self, request, obj): update_version = self._get_version_from_request(request) if update_version != obj["version"]: data = self._create_version_error_response(obj["version"]) return create_response(request, json=data, status_code=409)
def incident_types_get(self, request): """ Callback for GET to /orgs/<org_id>/types/incident """ LOG.info("incident_types_get") return requests_mock.create_response( request, status_code=200, json=IncidentMock.mock_incident_fields)
def artifacts_files_post(self, request): """ POST an artifact file""" data = [{"id": 51, "type": 29, "value": "test", "description": None}] return requests_mock.create_response(request, status_code=200, json=data)
def task_get(self, request): """ Callback for GET to /orgs/<org_id>/tasks/<inc_id> """ LOG.debug("task_get") return requests_mock.create_response(request, status_code=200, json=self.task)
def artifacts_files_post(self, request): """ POST an artifact file""" return requests_mock.create_response(request, status_code=200, json={})
def actions_get(self, request): """ Callback for GET to /orgs/<org_id>/actions """ LOG.debug("actions_get") return requests_mock.create_response(request, status_code=200, json={"entities": []})
def test_response(**kwargs): r = requests.Request(method='GET', url='http://localhost:5000').prepare() return requests_mock.create_response(r, **kwargs)
def create_commercetools_response(request, **kwargs): correlation_id = request.headers.get(HEADER_CORRELATION_ID, f"projects-{str(uuid.uuid4())}") headers = kwargs.pop("headers", {}) headers.update({HEADER_CORRELATION_ID: correlation_id}) return create_response(request, headers=headers, **kwargs)
def incident_get(self, request): """ Callback for GET to /orgs/<org_id>/incidents/<inc_id> """ LOG.debug("incident_get") return requests_mock.create_response(request, status_code=200, json=self.incident)
def get_by_key(self, request, key): obj = self.model.get_by_key(key) if obj: content = schemas.ProductProjectionSchema().dumps(obj) return create_response(request, text=content) return create_response(request, status_code=404)
def function_fields_get(self, request): """ Callback for GET to /orgs/<org_id>/types/__function/fields """ LOG.debug("function_fields_get") field_data = test_data("200_JSON_GET__types_function_fields.json") return requests_mock.create_response(request, status_code=200, json=field_data)
def attachment_contents_get(self, request): """ Callback for GET to attachment contents """ LOG.debug("attachment_contents_get") return requests_mock.create_response(request, status_code=200, json='"abcdef"')