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