Check disabled_services when finding a service

app/models/project.rb

@@ -1088,9 +1088,7 @@ class Project < ActiveRecord::Base
       find_or_initialize_service(service_name)
     end
 
-    available_services.reject do |service|
-      disabled_services.include?(service.to_param)
-    end
+    available_services.compact
   end
 
   def disabled_services
@@ -1098,6 +1096,8 @@ class Project < ActiveRecord::Base
   end
 
   def find_or_initialize_service(name)
+    return if disabled_services.include?(name)
+
     service = find_service(services, name)
     return service if service
 

spec/models/project_spec.rb

@@ -4037,6 +4037,21 @@ describe Project do
         expect(result).to be_empty
       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
     subject { build(:project) }
@@ -4044,12 +4059,17 @@ describe Project do
     it 'avoids N+1 database queries' do
       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
 
-        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
+
+    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