Skip to content

Commit 7bbf76e

Browse files
committed
[Test Runner] Updates wipe cluster
Adds wait_for_completion=false, build is timing out on deleting jobs and datafeeds. Adds force: true and timeout to avoid duplicated entities.
1 parent 59e3768 commit 7bbf76e

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

elasticsearch-api/api-spec-testing/wipe_cluster.rb

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -414,24 +414,24 @@ def clear_privileges(client)
414414
def clear_ml_jobs(client)
415415
client.ml.close_job(job_id: '_all', force: true)
416416
client.ml.get_jobs['jobs'].each do |d|
417-
client.ml.delete_job(job_id: d['job_id'])
417+
client.ml.delete_job(job_id: d['job_id'], wait_for_completion: false, force: true)
418418
end
419419
end
420420

421421
def clear_datafeeds(client)
422-
client.ml.stop_datafeed(datafeed_id: '_all', force: true)
423-
client.ml.get_datafeeds['datafeeds'].each do |d|
424-
client.ml.delete_datafeed(datafeed_id: d['datafeed_id'])
422+
Thread.new do
423+
client.ml.stop_datafeed(datafeed_id: '_all', force: true, timeout: '120s')
424+
client.ml.get_datafeeds['datafeeds'].each do |d|
425+
client.ml.delete_datafeed(datafeed_id: d['datafeed_id'], force: true)
426+
end
425427
end
426428
end
427429

428430
def clear_tasks(client)
429431
tasks = client.tasks.get['nodes'].values.first['tasks'].values.select do |d|
430432
d['cancellable']
431-
end.map do |d|
432-
"#{d['node']}:#{d['id']}"
433433
end
434-
tasks.each { |t| client.tasks.cancel task_id: t }
434+
tasks.each { |d| client.tasks.cancel(task_id: "#{d['node']}:#{d['id']}") }
435435
end
436436

437437
def clear_machine_learning_indices(client)
@@ -448,7 +448,7 @@ def clear_index_templates(client)
448448

449449
def clear_transforms(client)
450450
client.transform.get_transform(transform_id: '*')['transforms'].each do |transform|
451-
client.transform.delete_transform(transform_id: transform['id'])
451+
client.transform.delete_transform(transform_id: transform['id'], force: true)
452452
end
453453
end
454454

@@ -482,7 +482,7 @@ def delete_trained_models(client)
482482
return unless models['trained_model_configs']
483483

484484
models['trained_model_configs'].each do |model|
485-
client.ml.delete_trained_model(model_id: model['model_id'], force: true, ignore: 400)
485+
client.ml.delete_trained_model(model_id: model['model_id'], force: true, ignore: 400, timeout: '120s')
486486
end
487487
end
488488
end

0 commit comments

Comments
 (0)