def test_find_events_smoke():
num = 50
rs, e_desc, data_keys = setup_syn()
all_data = syn_data(data_keys, num)
mdsc.bulk_insert_events(e_desc, all_data, validate=False)
mdsc.insert_run_stop(rs, ttime.time(), uid=str(uuid.uuid4()))
mdsc.clear_process_cache()
# make sure the uid works
next(mdsc.find_events(descriptor=e_desc))
mdsc.clear_process_cache()
descriptor, = mdsc.find_descriptors(uid=e_desc)
mdsc.clear_process_cache()
# make sure that searching by descriptor document works
next(mdsc.find_events(descriptor=descriptor))
def test_find_events_smoke():
num = 50
rs, e_desc, data_keys = setup_syn()
all_data = syn_data(data_keys, num)
mdsc.bulk_insert_events(e_desc, all_data, validate=False)
mdsc.insert_run_stop(rs, ttime.time(), uid=str(uuid.uuid4()))
mdsc.clear_process_cache()
# make sure the uid works
next(mdsc.find_events(descriptor=e_desc))
mdsc.clear_process_cache()
descriptor, = mdsc.find_descriptors(uid=e_desc)
mdsc.clear_process_cache()
# make sure that searching by descriptor document works
next(mdsc.find_events(descriptor=descriptor))
def test_event_descriptor_insertion():
# format some data keys for insertion
data_keys = {
"some_value": {"source": "PV:pv1", "shape": [1, 2], "dtype": "array"},
"some_other_val": {"source": "PV:pv2", "shape": [], "dtype": "number"},
"data_key3": {"source": "PV:pv1", "shape": [], "dtype": "number", "external": "FS:foobar"},
}
time = ttime.time()
# test insert
ev_desc_uid = mdsc.insert_descriptor(run_start_uid, data_keys, time, str(uuid.uuid4()))
ev_desc_mds, = mdsc.find_descriptors(uid=ev_desc_uid)
# make sure the sanitized event descriptor has no uid
check_for_id(ev_desc_mds)
# make sure the event descriptor is pointing to the correct run start
referenced_run_start = ev_desc_mds["run_start"]
assert_equal(referenced_run_start.uid, run_start_uid)
assert_equal(ev_desc_mds["time"], time)
for k in data_keys:
for ik in data_keys[k]:
assert_equal(ev_desc_mds.data_keys[k][ik], data_keys[k][ik])
def test_event_descriptor_insertion():
# format some data keys for insertion
data_keys = {
'some_value': {
'source': 'PV:pv1',
'shape': [1, 2],
'dtype': 'array'
},
'some_other_val': {
'source': 'PV:pv2',
'shape': [],
'dtype': 'number'
},
'data_key3': {
'source': 'PV:pv1',
'shape': [],
'dtype': 'number',
'external': 'FS:foobar'
}
}
time = ttime.time()
# test insert
ev_desc_uid = mdsc.insert_descriptor(run_start_uid, data_keys, time,
str(uuid.uuid4()))
ev_desc_mds, = mdsc.find_descriptors(uid=ev_desc_uid)
# make sure the sanitized event descriptor has no uid
check_for_id(ev_desc_mds)
# make sure the event descriptor is pointing to the correct run start
referenced_run_start = ev_desc_mds['run_start']
assert_equal(referenced_run_start.uid, run_start_uid)
assert_equal(ev_desc_mds['time'], time)
for k in data_keys:
for ik in data_keys[k]:
assert_equal(ev_desc_mds.data_keys[k][ik], data_keys[k][ik])
