def datastore(datastore_connection, filestore):
ds = datastore_connection
try:
random_data.create_heuristics(ds, heuristics_count=10)
random_data.create_services(ds)
random_data.create_signatures(ds)
random_data.create_users(ds)
random_data.create_workflows(ds)
submissions = []
for _ in range(2):
submissions.append(random_data.create_submission(
ds, filestore))
random_data.create_alerts(ds,
alert_count=10,
submission_list=submissions)
yield ds
finally:
# Cleanup test data
random_data.wipe_alerts(ds)
random_data.wipe_heuristics(ds)
random_data.wipe_services(ds)
random_data.wipe_signatures(ds)
random_data.wipe_submissions(ds, filestore)
random_data.wipe_users(ds)
random_data.wipe_workflows(ds)
def purge_data(ds, fs):
wipe_alerts(ds)
wipe_heuristics(ds)
wipe_services(ds)
wipe_signatures(ds)
wipe_submissions(ds, fs)
wipe_users(ds)
wipe_workflows(ds)
def datastore(request, datastore_connection, fs):
wipe_alerts(datastore_connection)
wipe_submissions(datastore_connection, fs)
for _ in range(NUM_SUBMISSIONS):
all_submissions.append(create_submission(datastore_connection, fs))
create_alerts(datastore_connection,
alert_count=NUM_ALERTS,
submission_list=all_submissions)
for alert in datastore_connection.alert.stream_search("id:*", fl="*"):
all_alerts.append(alert)
try:
yield datastore_connection
finally:
wipe_alerts(datastore_connection)
wipe_submissions(datastore_connection, fs)
datastore_connection.alert.wipe()
def cleanup(ds, fs):
wipe_users(ds)
wipe_submissions(ds, fs)
wipe_alerts(ds)
def purge_alert(ds, fs):
wipe_alerts(ds)
wipe_submissions(ds, fs)