From 8655fd19ce743ab4f28e5a9327913e4987d8ca3f Mon Sep 17 00:00:00 2001
From: Ana Rosas <a.rosas10@gmail.com>
Date: Mon, 30 May 2016 15:46:59 +0200
Subject: [PATCH] Refacto endpoint

---
 lib/travis/api/app/endpoint/builds.rb | 25 +++++++++++--------------
 1 file changed, 11 insertions(+), 14 deletions(-)

diff --git a/lib/travis/api/app/endpoint/builds.rb b/lib/travis/api/app/endpoint/builds.rb
index e04dc25f..3a4243a2 100644
--- a/lib/travis/api/app/endpoint/builds.rb
+++ b/lib/travis/api/app/endpoint/builds.rb
@@ -51,25 +51,22 @@ class Travis::Api::App
         Metriks.meter("api.request.restart_build").mark
         if Travis::Features.owner_active?(:enqueue_to_hub, current_user)
           service = Travis::Enqueue::Services::RestartModel.new(current_user, { build_id: params[:id] })
-          if !service.accept?
-            status 400
-            result = false
-          else
-            payload = {id: params[:id], user_id: current_user.id}
-            service.push("build:restart", payload)
-            status 202
-            result = true
-          end
         else
           service = self.service(:reset_model, build_id: params[:id])
-          if !service.accept?
-            status 400
-            result = false
+        end
+
+        if !service.accept?
+          status 400
+          result = false
+        else
+          if service.respond_to?(:push)
+            payload = {id: params[:id], user_id: current_user.id}
+            service.push("build:restart", payload)
           else
             Travis::Sidekiq::BuildRestart.perform_async(id: params[:id], user_id: current_user.id)
-            status 202
-            result = true
           end
+          status 202
+          result = true
         end
 
         respond_with(result: result, flash: service.messages)