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 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"')
