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