def test_db_alert_dates(): al = Alert(query = 'test data', user = get_test_user(), label = 'test query', checking_interval = 5) db.session.add(al) db.session.commit() assert al.due_to_check() al.mark_as_checked() db.session.add(al) db.session.commit() assert not al.due_to_check() al.checked_at = datetime.datetime.utcnow() - datetime.timedelta(days=6) db.session.add(al) db.session.commit() assert al.due_to_check() al.checked_at = datetime.datetime.utcnow() - datetime.timedelta(days=4) db.session.add(al) db.session.commit() assert not al.due_to_check() db.session.delete(al) db.session.commit()
def create(): # also handles update data = request.get_json() print(data) if 'query_text' not in data: return jsonify({'status': 'invalid'}) authz.require(authz.logged_in()) if data.get('alert_id', None): # UPDATE alert_id = int(data['alert_id']) alert = obj_or_404(Alert.by_id(alert_id)) authz.require(alert.role_id == request.auth_role.id) alert.query_text = data['query_text'] alert.custom_label = data.get('custom_label' '') or data['query_text'] alert.checking_interval=int(data.get('checking_interval', 9)) else: # CREATE alert = Alert( role_id = request.auth_role.id, query_text=data['query_text'], custom_label=data.get('custom_label' '') or data['query_text'], checking_interval=int(data.get('checking_interval', 9)) ) db.session.add(alert) db.session.commit() return view(alert.id)
def setUp(self): super(EntitiesTestCase, self).setUp() self.rolex = self.create_user(foreign_id='user_3') self.col = Collection() self.col.label = 'Original Collection' self.col.foreign_id = 'test_coll_entities' db.session.add(self.col) self.col_other = Collection() self.col_other.label = 'Other Collection' self.col_other.foreign_id = 'test_coll_entities_other' db.session.add(self.col_other) db.session.flush() self.ent = Entity.save( { 'name': 'Winnie the Pooh', 'jurisdiction_code': 'pa', 'summary': 'a fictional teddy bear created by author A. A. Milne', 'identifiers': [{ 'scheme': 'wikipedia', 'identifier': 'en:Winnie-the-Pooh' }], 'other_names': [{ 'name': u'Puh der Bär' }, { 'name': 'Pooh Bear' }] }, [self.col]) db.session.add(self.ent) db.session.flush() self.other = Entity.save( { 'name': 'Pu der Bär', 'jurisdiction_code': 'de', 'description': 'he is a bear', 'identifiers': [{ 'scheme': 'wikipedia', 'identifier': 'en:Winnie-the-Pooh' }, { 'scheme': 'animals', 'identifier': 'bears.winnie.pooh' }], 'other_names': [{ 'name': u'Puh der Bär' }] }, [self.col_other]) db.session.add(self.other) self.alert = Alert() self.alert.entity = self.other db.session.add(self.alert) db.session.commit()
def create(): authz.require(authz.logged_in()) data = request.get_json() validate(data, alerts_schema) alert = Alert(user_id=current_user.id, query=data['query'], label=data.get('custom_label', data['query']), checking_interval=int(data.get('checking_interval', 9))) db.session.add(alert) db.session.commit() return view(alert.id)
def setUp(self): super(EntitiesTestCase, self).setUp() self.rolex = self.create_user(foreign_id='user_3') self.col = Collection() self.col.label = 'Original Collection' self.col.foreign_id = 'test_coll_entities' db.session.add(self.col) self.col_other = Collection() self.col_other.label = 'Other Collection' self.col_other.foreign_id = 'test_coll_entities_other' db.session.add(self.col_other) db.session.flush() self.ent = Entity.save( { 'schema': 'LegalEntity', 'name': 'Winnie the Pooh', 'data': { 'country': 'pa', 'summary': 'a fictional teddy bear created by A. A. Milne', 'alias': [u'Puh der Bär', 'Pooh Bear'] } }, self.col) self.other = Entity.save( { 'schema': 'LegalEntity', 'name': 'Pu der Bär', 'data': { 'country': 'de', 'description': 'he is a bear', 'alias': [u'Puh der Bär'] } }, self.col) self.alert = Alert() self.alert.entity = self.other db.session.add(self.alert) db.session.commit() index_entity(self.ent) index_entity(self.other)
def test_search(): al = Alert(query = 'China', user = get_test_user(), label = 'test query: China', checking_interval = 5) res = alerts.run_alert(al)