lib/troo/actions/set_default.rb in troo-0.0.3 vs lib/troo/actions/set_default.rb in troo-0.0.4

@@ -1,6 +1,12 @@
 module Troo
+  class NullEntity
+    def default?
+      true
+    end
+  end
+
   class SetDefault
     class << self
       def for(entity)
         new(entity).set_default!
       end
@@ -28,8 +34,13 @@
       entity.class.update(default: false)
     end
 
     def already_default?
       entity.default?
+    end
+
+    def entity
+      return NullEntity.new if @entity.nil?
+      @entity
     end
   end
 end