build cancel check before adding stages

lib/travis/scheduler/jobs/limit/stages.rb

@@ -20,7 +20,7 @@ def queueable?(job)
           end
 
           def queueable(job)
-            Travis::Stages.build(jobs(job.source_id)).startable
+            Travis::Stages.build(jobs(job.source_id), job.source_id).startable
           end
 
           def jobs(build_id)

lib/travis/scheduler/model/stages.rb

@@ -56,13 +56,19 @@
 
 module Travis
   module Stages
-    def self.build(jobs)
+    def self.build(jobs, build_id = nil)
+      return [] if build_id && build_canceled?(build_id)
+
       jobs.each_with_object(Stage.new(nil, 0)) do |job, stage|
         job = Job.new(*job.values_at(:id, :state, :stage))
         stage << job unless job.finished?
       end
     end
 
+    def self.build_canceled?(id)
+      Build.find(id)&.canceled? || false
+    end
+
     class Stage
       attr_reader :parent, :num, :children
 

lib/travis/scheduler/record/build.rb

@@ -8,4 +8,8 @@ class Build < ActiveRecord::Base
   has_many :jobs, as: :source
 
   serialize :config
+
+  def canceled?
+    state&.to_sym == :canceled
+  end
 end

lib/travis/scheduler/record/job.rb

@@ -81,6 +81,10 @@ def finished?
     FINISHED_STATES.include?(state.try(:to_sym))
   end
 
+  def canceled?
+    state&.to_sym == :canceled
+  end
+
   def queueable=(value)
     if value
       unless queueable