def setUpClass(cls):
"""
Set up read-only test data to test GET requests.
"""
super(TestSavedSearchDelete, cls).setUpClass()
with app.app_context():
cls._person_0 = Person(
name="Aniki",
email="aniki_baniki_{}@test.com".format(random()),
)
db.session.add(cls._person_0)
db.session.flush()
saved_search = SavedSearch(
name="test_ss_1",
object_type="Assessment",
user=cls._person_0,
search_type="GlobalSearch"
)
cls._person_0.saved_searches.append(saved_search)
db.session.flush()
cls._user_role = setup_user_role(cls._person_0)
db.session.commit()
cls._client, cls._headers = get_client_and_headers(
app, cls._person_0,
)
cls._saved_search_id = saved_search.id
def setUpClass(cls):
"""
Set up read-only test data to test GET requests.
"""
super(TestSavedSearchPost, cls).setUpClass()
cls._valid_query = [{
"object_name": "Assessment",
"filters": {
"expression": {}
},
"limit": [0, 10],
"order_by": [{
"name": "updated_at",
"desc": True
}]
}]
email_0 = "aniki_baniki_{}@test.com".format(random())
cls._person_0 = Person(name="Aniki", email=email_0)
db.session.add(cls._person_0)
db.session.flush()
cls._user_role = setup_user_role(cls._person_0)
db.session.commit()
cls._client, cls._headers = get_client_and_headers(
app,
cls._person_0,
)
def get_full_user_json():
"""Get the full current user"""
with benchmark("Get full user JSON"):
from ggrc.models.person import Person
current_user = get_current_user()
person = Person.eager_query().filter_by(id=current_user.id).one()
result = publish_representation(publish(person, (), inclusion_filter))
return as_json(result)
def get_full_user_json():
"""Get the full current user"""
with benchmark("Get full user JSON"):
from ggrc.models.person import Person
current_user = get_current_user()
person = Person.eager_query().filter_by(id=current_user.id).one()
result = publish_representation(publish(person, (), inclusion_filter))
return as_json(result)
def create_user(email, **kwargs):
"""Create User
attr:
email (string) required
"""
user = Person(email=email, **kwargs)
db.session.add(user)
db.session.flush()
log_event(db.session, user, user.id)
db.session.commit()
return user
def create_user(email, **kwargs):
"""Creates a user.
Args:
email: (string) mandatory user email
"""
user = Person(email=email, **kwargs)
db.session.add(user)
db.session.flush()
log_event(db.session, user, user.id)
db.session.commit()
return user
def setUpClass(cls):
"""Set up read-only test data to test GET requests."""
super(TestSavedSearchPost, cls).setUpClass()
email_0 = "aniki_baniki_{}@test.com".format(random())
cls._person_0 = Person(name="Aniki", email=email_0)
db.session.add(cls._person_0)
db.session.flush()
cls._user_role = setup_user_role(cls._person_0)
db.session.commit()
cls._client, cls._headers = get_client_and_headers(
app, cls._person_0,
)
def create_user(email, **kwargs):
user = Person(email=email, **kwargs)
db.session.add(user)
db.session.flush()
log_event(db.session, user, user.id)
user_context = Context(
name='Personal Context for {0}'.format(email),
description='',
related_object=user,
context_id=1,
)
db.session.add(user_context)
db.session.commit()
get_indexer().create_record(fts_record_for(user))
return user
def create_user(email, **kwargs):
"""Creates a user.
Args:
email: (string) mandatory user email
"""
if not hasattr(flask.g, "user_cache"):
flask.g.user_cache = {}
if email in flask.g.user_cache:
return flask.g.user_cache[email]
user = Person(email=email, **kwargs)
flask.g.user_cache[email] = user
db.session.add(user)
return user
def create_user(email, **kwargs):
"""Create User
attr:
email (string) required
"""
user = Person(email=email, **kwargs)
db.session.add(user)
db.session.flush()
log_event(db.session, user, user.id)
user_context = Context(
name='Personal Context for {0}'.format(email),
description='',
related_object=user,
context_id=1,
)
db.session.add(user_context)
db.session.commit()
return user
def setUpClass(cls):
"""
Set up read-only test data to test GET requests.
"""
super(TestSavedSearchDelete, cls).setUpClass()
cls._person_0 = Person(
name="Aniki",
email="aniki_baniki_{}@test.com".format(random()),
)
db.session.add(cls._person_0)
db.session.flush()
saved_search = SavedSearch(
name="test_ss_1",
query=[{
"object_name": "Assessment",
"filters": {
"expression": {}
},
"limit": [0, 10],
"order_by": [{
"name": "updated_at",
"desc": True
}]
}],
object_type="Assessment",
user=cls._person_0,
)
cls._person_0.saved_searches.append(saved_search)
db.session.flush()
cls._user_role = setup_user_role(cls._person_0)
db.session.commit()
cls._client, cls._headers = get_client_and_headers(
app,
cls._person_0,
)
cls._saved_search_id = saved_search.id
def setUpClass(cls):
"""
Set up read-only test data to test GET requests.
"""
super(TestSavedSearchGet, cls).setUpClass()
email_0 = "aniki_baniki_{}@test.com".format(random())
cls._person_0 = Person(name="Aniki", email=email_0)
email_1 = "baniki_aniki_{}@test.com".format(random())
cls._person_1 = Person(name="Baniki", email=email_1)
db.session.add(cls._person_0)
db.session.add(cls._person_1)
db.session.flush()
cls._valid_query = [{
"object_name": "Assessment",
"filters": {
"expression": {}
},
"limit": [0, 10],
"order_by": [{
"name": "updated_at",
"desc": True
}]
}]
locked_time = {
"year": 2025,
"month": 1,
"day": 25,
"minute": 0,
"second": 0,
}
with freeze_time(datetime(**locked_time)) as frozen_time:
for i in range(1, 5):
user = cls._person_0 if i != 4 else cls._person_1
saved_search = SavedSearch(
name="test_ss_{}".format(i),
query=cls._valid_query,
object_type="Assessment",
user=user,
)
user.saved_searches.append(saved_search)
db.session.flush()
#
# we need to test the order in which saved
# searches are returned in response to GET
# request (descending order by created_at)
# created_at precision is seconds
#
locked_time["second"] = i
frozen_time.move_to(datetime(**locked_time))
cls._user_role = setup_user_role(cls._person_0)
db.session.commit()
cls._client, cls._headers = get_client_and_headers(
app,
cls._person_0,
)
def setUpClass(cls):
"""
Set up read-only test data to test GET requests.
"""
super(TestSavedSearchGet, cls).setUpClass()
with app.app_context():
email_0 = "aniki_baniki_{}@test.com".format(random())
cls._person_0 = Person(name="Aniki", email=email_0)
email_1 = "baniki_aniki_{}@test.com".format(random())
cls._person_1 = Person(name="Baniki", email=email_1)
db.session.add(cls._person_0)
db.session.add(cls._person_1)
db.session.flush()
locked_time = {
"year": 2025,
"month": 1,
"day": 25,
"minute": 0,
"second": 0,
}
with freeze_time(datetime(**locked_time)) as frozen_time:
for i in range(1, 5):
user = cls._person_0 if i != 4 else cls._person_1
saved_search = SavedSearch(
name="test_ss_{}".format(i),
object_type="Assessment",
user=user,
search_type=cls.SAVED_SEARCH_TYPE)
user.saved_searches.append(saved_search)
db.session.flush()
#
# we need to test the order in which saved
# searches are returned in response to GET
# request (descending order by created_at)
# created_at precision is seconds
#
locked_time["second"] = i
frozen_time.move_to(datetime(**locked_time))
saved_search_program = SavedSearch(
name="test_program_ss",
object_type="Program",
user=cls._person_0,
search_type=cls.SAVED_SEARCH_TYPE)
cls._person_0.saved_searches.append(saved_search_program)
db.session.flush()
cls._user_role = setup_user_role(cls._person_0)
db.session.commit()
cls._client, cls._headers = get_client_and_headers(
app,
cls._person_0,
)
def create_user(email, **kwargs):
user = Person(email=email, **kwargs)
db.session.add(user)
db.session.commit()
return user
