def generate_body(self, type, **kwargs):
"""Generate body of HTTP request based on JSON representation."""
if type == self._count:
return json.dumps([TemplateProvider.generate_template_as_dict(
json_tmpl_name=type, **kwargs)[self._count]])
else:
return json.dumps([TemplateProvider.generate_template_as_dict(
json_tmpl_name=type, **kwargs)])
def upgrade_template(template_name, title, title_postfix, **kwargs):
"""Return updated JSON template according requirements."""
if template_name == "relationship" or template_name == "object_owner":
return TemplateProvider.get_template_as_dict(
template_name, **kwargs)
else:
if title_postfix is not None:
title += title_postfix
return TemplateProvider.get_template_as_dict(template_name,
title=title,
**kwargs)
def generate_body(self, type_name, **kwargs):
"""Generate body of HTTP request based on JSON representation."""
body = TemplateProvider.generate_template_as_dict(
json_tmpl_name=type_name, **kwargs)
if not self.is_api:
body = body[type_name]
return json.dumps([body]).encode("string-escape")
def generate_body(self, type_name, **kwargs):
"""Generate body of HTTP request based on JSON representation."""
body = TemplateProvider.generate_template_as_dict(
json_tmpl_name=type_name, **kwargs)
if not self.is_api:
body = body[type_name]
return json.dumps([body]).encode("string-escape")
def add_user(url_origin):
"""If two processes log in as a new user at the same time,
the second login may fail because of a race condition in dev code
(user_generator.py).
We workaround this issue by creating a user not in parallel as this issue
is considered not worth to fix by developers.
"""
from lib.entities import entities_factory
import json
from lib.service.rest.template_provider import TemplateProvider
from lib.service.rest import session_pool
environment.app_url = url_origin
environment.app_url = urlparse.urljoin(environment.app_url, "/")
session = requests.Session()
test_utils.wait_for(
lambda: session.get(environment.app_url).status_code == OK_CODE,
constants.timeouts.TWO_MIN_USER_WAIT * 4)
test_utils.wait_for(lambda: session.get(url_module.Urls().gae_login(
users.FAKE_SUPER_USER)).status_code == OK_CODE)
test_utils.wait_for(
lambda: session.get(url_module.Urls().login).status_code == OK_CODE)
person_dict = entities_factory.PeopleFactory().create(
is_add_rest_attrs=True, **{}).__dict__
url = environment.app_url + "api/people"
body = json.dumps([
TemplateProvider.generate_template_as_dict(json_tmpl_name='person',
**person_dict)
]).encode("string-escape")
test_utils.wait_for(lambda: session.post(
url=url, data=body, headers=session_pool.BASIC_HEADERS).status_code ==
OK_CODE)
def upgrade_template(template_name, title_postfix=None, **kwargs):
"""Return updated template json with random title by template name"""
obj_title = append_random_string(template_name)
if title_postfix is not None:
obj_title += title_postfix
return TemplateProvider.get_template_as_dict(template_name,
title=obj_title,
**kwargs)
def update_body(body, **kwargs):
"""Update body of HTTP request based on JSON representation."""
return json.dumps(TemplateProvider.update_template_as_dict(
json_data_str=body, **kwargs)).encode("string-escape")
def update_body(body, **kwargs):
"""Update body of HTTP request based on JSON representation."""
return json.dumps(TemplateProvider.update_template_as_dict(
json_data_str=body, **kwargs)).encode("string-escape")
def update_body(body, **kwargs):
"""Update body of HTTP request based on JSON representation."""
return json.dumps(TemplateProvider.update_template_as_dict(
json_tmpl_name=body, **kwargs))
def update_body(body, **kwargs):
"""Update body of HTTP request based on JSON representation."""
return json.dumps(TemplateProvider.update_template_as_dict(
json_tmpl_name=body, **kwargs))
