def _run_events_and_compare(events):
current_search.flush_and_refresh('relationships')
for ev in events:
event = generate_payload(ev)
EventAPI.handle_event(event)
reindex_all_relationships()
assert_es_equals_db()
def test_identities(events, result_sets, db, es):
# NOTE: We assume that only on identifier scheme being used so just using
# identifier values is enough when comparing sets.
for ev in events:
EventAPI.handle_event(generate_payload(ev))
for rs in result_sets:
for v in rs:
id_ = Identifier.query.filter_by(value=v).one()
ids = set(i.value for i in id_.get_identities())
assert ids == rs
def test_simple_citations(test_case_name, events, results, db, es):
"""Test simple citation queries."""
for ev in events:
EventAPI.handle_event(generate_payload(ev))
es.indices.refresh()
for cited_id_value, (citation_result, relation_result) in results.items():
cited_id = (Identifier.query.filter_by(value=cited_id_value).one())
citations = set()
relations = set()
for citing_id, relation in RelationshipAPI.get_citations(cited_id):
citations |= {i.value for i in citing_id}
relations |= {(r.source.value, r.target.value) for r in relation}
assert citations == citation_result
assert relations == relation_result
def _process_events(events):
for e in events:
EventAPI.handle_event(generate_payload(e))
reindex_all_relationships()
def _process_events(events):
for e in events:
EventAPI.handle_event(generate_payload(e))
current_search.flush_and_refresh('relationships')
def _handle_events(events, no_index=False):
for ev in events:
EventAPI.handle_event(generate_payload(ev), no_index=no_index)
def test_grouping_query(test_case_name, events, results, db, es):
for ev in events:
EventAPI.handle_event(generate_payload(ev))
for cited_id_value, _ in results.items():
pass
