From 75460b242768febd0801900346419400399d9a61 Mon Sep 17 00:00:00 2001
From: Ana Rosas <a.rosas10@gmail.com>
Date: Mon, 13 Jun 2016 18:13:21 +0200
Subject: [PATCH] Use Travis::Enqueue::Services::CancelModel on v3 build cancel
 endpoint

---
 lib/travis/api/v3/queries/build.rb | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/lib/travis/api/v3/queries/build.rb b/lib/travis/api/v3/queries/build.rb
index 5314a7c9..efa90e6f 100644
--- a/lib/travis/api/v3/queries/build.rb
+++ b/lib/travis/api/v3/queries/build.rb
@@ -1,3 +1,6 @@
+require 'travis/api/enqueue/services/restart_model'
+require 'travis/api/enqueue/services/cancel_model'
+
 module Travis::API::V3
   class Queries::Build < Query
     params :id
@@ -10,7 +13,13 @@ module Travis::API::V3
     def cancel(user)
       raise BuildNotCancelable if %w(passed failed canceled errored).include? find.state
       payload = { id: id, user_id: user.id, source: 'api' }
-      perform_async(:build_cancellation, payload)
+      if Travis::Features.owner_active?(:enqueue_to_hub, user)
+        payload[:type] = :build
+        service = Travis::Enqueue::Services::CancelModel.new(user, payload)
+        service.push
+      else
+        perform_async(:build_cancellation, payload)
+      end
       payload
     end