Enqueue restart message on build restart v3 endpoint

This commit is contained in:
Ana Rosas 2016-06-14 16:46:29 +02:00
parent e097c8889e
commit bb7260bc9e
2 changed files with 136 additions and 2 deletions

View File

@ -25,8 +25,14 @@ module Travis::API::V3
def restart(user)
raise BuildAlreadyRunning if %w(received queued started).include? find.state
payload = { id: id, user_id: user.id, source: 'api' }
perform_async(:build_restart, payload)
if Travis::Features.owner_active?(:enqueue_to_hub, user)
service = Travis::Enqueue::Services::RestartModel.new(user, { build_id: id })
payload = { id: id, user_id: user.id }
service.push("build:restart", payload)
else
payload = { id: id, user_id: user.id, source: 'api' }
perform_async(:build_restart, payload)
end
payload
end
end

View File

@ -8,6 +8,7 @@ describe Travis::API::V3::Services::Build::Restart do
before do
Travis::Features.stubs(:owner_active?).returns(true)
Travis::Features.stubs(:owner_active?).with(:enqueue_to_hub, repo.owner).returns(false)
@original_sidekiq = Sidekiq::Client
Sidekiq.send(:remove_const, :Client) # to avoid a warning
Sidekiq::Client = []
@ -245,6 +246,133 @@ describe Travis::API::V3::Services::Build::Restart do
end
end
describe "existing repository, push access, build not already running, enqueues message for Hub" do
let(:params) {{}}
let(:token) { Travis::Api::App::AccessToken.create(user: repo.owner, app_id: 1) }
let(:headers) {{ 'HTTP_AUTHORIZATION' => "token #{token}" }}
before do
Travis::API::V3::Models::Permission.create(repository: repo, user: repo.owner, push: true)
Travis::Features.stubs(:owner_active?).with(:enqueue_to_hub, repo.owner).returns(true)
end
describe "errored state" do
before do
build.update_attribute(:state, "errored")
post("/v3/build/#{build.id}/restart", params, headers)
end
example { expect(last_response.status).to be == 202 }
example { expect(JSON.load(body).to_s).to include(
"@type",
"pending",
"build",
"@href",
"@representation",
"minimal",
"restart",
"id",
"state_change")
}
example { expect(sidekiq_payload).to be == {
"id" => "#{build.id}",
"user_id"=> repo.owner_id}
}
example { expect(Sidekiq::Client.last['queue']).to be == 'hub' }
example { expect(Sidekiq::Client.last['class']).to be == 'Travis::Hub::Sidekiq::Worker' }
end
describe "passed state" do
before { build.update_attribute(:state, "passed") }
before { post("/v3/build/#{build.id}/restart", params, headers) }
example { expect(last_response.status).to be == 202 }
example { expect(JSON.load(body).to_s).to include(
"@type",
"pending",
"build",
"@href",
"@representation",
"minimal",
"restart",
"id",
"state_change")
}
example { expect(sidekiq_payload).to be == {
"id" => "#{build.id}",
"user_id"=> repo.owner_id}
}
example { expect(Sidekiq::Client.last['queue']).to be == 'hub' }
example { expect(Sidekiq::Client.last['class']).to be == 'Travis::Hub::Sidekiq::Worker' }
end
describe "failed state" do
before { build.update_attribute(:state, "failed") }
before { post("/v3/build/#{build.id}/restart", params, headers) }
example { expect(last_response.status).to be == 202 }
example { expect(JSON.load(body).to_s).to include(
"@type",
"pending",
"build",
"@href",
"@representation",
"minimal",
"restart",
"id",
"state_change")
}
example { expect(sidekiq_payload).to be == {
"id" => "#{build.id}",
"user_id"=> repo.owner_id}
}
example { expect(Sidekiq::Client.last['queue']).to be == 'hub' }
example { expect(Sidekiq::Client.last['class']).to be == 'Travis::Hub::Sidekiq::Worker' }
end
describe "canceled state" do
before { build.update_attribute(:state, "canceled") }
before { post("/v3/build/#{build.id}/restart", params, headers) }
example { expect(last_response.status).to be == 202 }
example { expect(JSON.load(body).to_s).to include(
"@type",
"pending",
"build",
"@href",
"@representation",
"minimal",
"restart",
"id",
"state_change")
}
example { expect(sidekiq_payload).to be == {
"id" => "#{build.id}",
"user_id"=> repo.owner_id}
}
example { expect(Sidekiq::Client.last['queue']).to be == 'hub' }
example { expect(Sidekiq::Client.last['class']).to be == 'Travis::Hub::Sidekiq::Worker' }
end
describe "setting id has no effect" do
before { build.update_attribute(:state, "canceled") }
before { post("/v3/build/#{build.id}/restart", params, headers) }
let(:params) {{ id: 42 }}
example { expect(sidekiq_payload).to be == {
"id" => "#{build.id}",
"user_id"=> repo.owner_id}
}
end
end
# TODO decided to discuss further with rkh as this use case doesn't really exist at the moment
# and 'fixing' the query requires modifying workers that v2 uses, thereby running the risk of breaking v2,
# and also because in 6 months or so travis-hub will be able to cancel builds without using travis-core at all.