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)