return Delayed::Priority from add/subtract, add #to_d

Betterment · Apr 5, 2024 · 8436715 · 8436715
1 parent f3aac3e
commit 8436715
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lib/delayed/priority.rb b/lib/delayed/priority.rb
@@ -174,11 +174,17 @@ def <=>(other)
     end
 
     def -(other)
-      to_i - other.to_i
+      other = other.to_i if other.is_a?(self.class)
+      self.class.new(to_i - other)
     end
 
     def +(other)
-      to_i + other.to_i
+      other = other.to_i if other.is_a?(self.class)
+      self.class.new(to_i + other)
+    end
+
+    def to_d
+      to_i.to_d
     end
 
     private