Commit f934b216 authored by Brett Walker's avatar Brett Walker

Check disabled_services when finding a service

parent 54c442af
...@@ -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,19 +4037,39 @@ describe Project do ...@@ -4037,19 +4037,39 @@ describe Project do
expect(result).to be_empty expect(result).to be_empty
end end
end end
end
describe "#find_or_initialize_service" do describe "#find_or_initialize_services" do
subject { build(:project) } subject { build(:project) }
it 'avoids N+1 database queries' do it 'returns only enabled services' 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))
allow(subject).to receive(:disabled_services).and_return(%w(prometheus))
control_count = ActiveRecord::QueryRecorder.new { project.find_or_initialize_service('prometheus') }.count services = subject.find_or_initialize_services
allow(Service).to receive(:available_services_names).and_call_original expect(services.count).to eq 1
expect(services).to include(PushoverService)
end
end
expect { project.find_or_initialize_service('prometheus') }.not_to exceed_query_limit(control_count) describe "#find_or_initialize_service" do
end subject { build(:project) }
it 'avoids N+1 database queries' do
allow(Service).to receive(:available_services_names).and_return(%w(prometheus pushover))
control_count = ActiveRecord::QueryRecorder.new { subject.find_or_initialize_service('prometheus') }.count
allow(Service).to receive(:available_services_names).and_call_original
expect { subject.find_or_initialize_service('prometheus') }.not_to exceed_query_limit(control_count)
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