Commit c3389c80 authored by Stan Hu's avatar Stan Hu

Merge branch 'bw-fix-checking-disabled-services' into 'master'

Check disabled_services when finding a service

See merge request gitlab-org/gitlab-ce!22172
parents 2fce6d6f f934b216
...@@ -1088,9 +1088,7 @@ class Project < ActiveRecord::Base ...@@ -1088,9 +1088,7 @@ class Project < ActiveRecord::Base
find_or_initialize_service(service_name) find_or_initialize_service(service_name)
end end
available_services.reject do |service| available_services.compact
disabled_services.include?(service.to_param)
end
end end
def disabled_services def disabled_services
...@@ -1098,6 +1096,8 @@ class Project < ActiveRecord::Base ...@@ -1098,6 +1096,8 @@ class Project < ActiveRecord::Base
end end
def find_or_initialize_service(name) def find_or_initialize_service(name)
return if disabled_services.include?(name)
service = find_service(services, name) service = find_service(services, name)
return service if service return service if service
......
...@@ -4037,6 +4037,21 @@ describe Project do ...@@ -4037,6 +4037,21 @@ describe Project do
expect(result).to be_empty expect(result).to be_empty
end end
end end
end
describe "#find_or_initialize_services" do
subject { build(:project) }
it 'returns only enabled services' do
allow(Service).to receive(:available_services_names).and_return(%w(prometheus pushover))
allow(subject).to receive(:disabled_services).and_return(%w(prometheus))
services = subject.find_or_initialize_services
expect(services.count).to eq 1
expect(services).to include(PushoverService)
end
end
describe "#find_or_initialize_service" do describe "#find_or_initialize_service" do
subject { build(:project) } subject { build(:project) }
...@@ -4044,12 +4059,17 @@ describe Project do ...@@ -4044,12 +4059,17 @@ describe Project do
it 'avoids N+1 database queries' do it 'avoids N+1 database queries' do
allow(Service).to receive(:available_services_names).and_return(%w(prometheus pushover)) allow(Service).to receive(:available_services_names).and_return(%w(prometheus pushover))
control_count = ActiveRecord::QueryRecorder.new { project.find_or_initialize_service('prometheus') }.count control_count = ActiveRecord::QueryRecorder.new { subject.find_or_initialize_service('prometheus') }.count
allow(Service).to receive(:available_services_names).and_call_original allow(Service).to receive(:available_services_names).and_call_original
expect { project.find_or_initialize_service('prometheus') }.not_to exceed_query_limit(control_count) expect { subject.find_or_initialize_service('prometheus') }.not_to exceed_query_limit(control_count)
end end
it 'returns nil if service is disabled' do
allow(subject).to receive(:disabled_services).and_return(%w(prometheus))
expect(subject.find_or_initialize_service('prometheus')).to be_nil
end end
end end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment