Improve style, add more tests

2 files changed, 6 insertions, 13 deletions

diff --git a/lib/api/helpers.rb b/lib/api/helpers.rb
index d6526ec4fdc..2c593dbb4ea 100644
--- a/lib/api/helpers.rb
+++ b/lib/api/helpers.rb
@@ -85,20 +85,11 @@ module API
       end
     end
 
-    def project_service
-      @project_service ||= user_project.find_or_initialize_service(params[:service_slug].underscore)
+    def project_service(project = user_project)
+      @project_service ||= project.find_or_initialize_service(params[:service_slug].underscore)
       @project_service || not_found!("Service")
     end
 
-    def service_by_slug(project, slug)
-      underscored_service = slug.underscore
-
-      not_found!('Service') unless Service.available_services_names.include?(underscored_service)
-      service_method = "#{underscored_service}_service"
-
-      service = project.public_send(service_method)
-    end
-
     def service_attributes
       @service_attributes ||= project_service.fields.inject([]) do |arr, hash|
         arr << hash[:name].to_sym
diff --git a/lib/api/services.rb b/lib/api/services.rb
index 163187d450d..e3c6a998631 100644
--- a/lib/api/services.rb
+++ b/lib/api/services.rb
@@ -67,7 +67,9 @@ module API
       post ':id/services/:service_slug/trigger' do
         project = Project.find_with_namespace(params[:id]) || Project.find_by(id: params[:id])
 
-        service = service_by_slug(project, params[:service_slug])
+        not_found! unless project
+
+        service = project_service(project)
 
         result = service.try(:active?) && service.try(:trigger, params)
 
@@ -75,7 +77,7 @@ module API
           status result[:status] || 200
           present result
         else
-          not_found!('Service')
+          not_found!
         end
       end
     end