spec/support/arb.rb in sidekiq-apriori-0.0.1 vs spec/support/arb.rb in sidekiq-apriori-0.0.2

@@ -1,19 +1,27 @@
 require 'active_record'
+require 'sidekiq-apriori/arb'
 require 'sqlite3'
 
 ActiveRecord::Base.establish_connection(
-  :adapter => :sqlite3,
-  :database => 'spec/support/test.db'
+  :adapter   => :sqlite3,
+  :database  => 'spec/support/test.db'
 )
 
 ActiveRecord::Schema.define do
-  drop_table :arbs
-
-  create_table :arbs do |t|
-    t.column :priority, :string
-  end
+  drop_table(:arbs) rescue nil
+  create_table(:arbs) { |t| t.column(:priority, :string) }
 end
 
 class Arb < ActiveRecord::Base
-  include Sidekiq::Apriori
+  include Sidekiq::Apriori::Arb
+end
+
+class PrioritizedUsingMethod < Arb
+  prioritize using: :some_method
+end
+
+class PrioritizedUsingCallable < Arb
+  prioritize do
+    self.some_other_method
+  end
 end