Alter class to avoid conflict with model test class

[rails.git] / lib / redactable.rb

diff --git a/lib/redactable.rb b/lib/redactable.rb
index b994e8563ab31e2934286d054533b0e2a555f712..d827cfd7462aa318a11c4754434ccd311dab4a7f 100644 (file)
--- a/lib/redactable.rb
+++ b/lib/redactable.rb
@@ -1,15 +1,22 @@
-require 'osm'
+require "osm"
 
 module Redactable
+  def self.included(base)
+    # this is used to extend activerecord bases, as these aren't
+    # in scope for the module itself.
+    base.scope :unredacted, -> { base.where(:redaction_id => nil) }
+  end
+
   def redacted?
-    not self.redaction.nil?
+    !redaction.nil?
   end
 
   def redact!(redaction)
     # check that this version isn't the current version
-    raise OSM::APICannotRedactError.new if self.is_latest_version?
+    raise OSM::APICannotRedactError if is_latest_version?
 
     # make the change
     self.redaction = redaction
+    save!
   end
 end