undestroy 0.0.2 → 0.1.0

data/test/unit/binding/active_record/restorable_test.rb

@@ -0,0 +1,132 @@
+require 'assert'
+
+module Undestroy::Binding::ActiveRecord::Restorable::Test
+
+  class Base < Undestroy::Test::Base
+    desc 'Binding::ActiveRecord::Restorable module'
+    subject { Undestroy::Binding::ActiveRecord::Restorable }
+
+    setup do
+      @ar = Undestroy::Test::ARMain
+      @receiver = Class.new(@ar) do
+        cattr_accessor :calls
+        self.calls = []
+
+        def destroy
+          calls << [:destroy]
+          self.freeze
+        end
+      end
+      @receiver.table_name = "foobar"
+      @receiver.class_attribute :undestroy_model_binding, :instance_writer => false
+      @receiver.undestroy_model_binding = Undestroy::Binding::ActiveRecord.new(@receiver)
+      @receiver.connection.create_table :foobar do |t|
+        t.string :name
+      end
+
+      @restore = Class.new do
+        cattr_accessor :calls
+        self.calls = []
+        def initialize(*args)
+          calls << [:init, args]
+        end
+        def run
+          calls << [:run]
+        end
+      end
+      @receiver.undestroy_model_binding.config.internals[:restore] = @restore
+    end
+
+    teardown do
+      @receiver.connection.drop_table :foobar
+    end
+
+    def assert_ran_restore(object)
+      assert_equal [:init, [{ :target => object, :config => @receiver.undestroy_model_binding.config }]],
+                    @restore.calls[0]
+      assert_equal [:run], @restore.calls[1]
+      assert_equal 2, @restore.calls.size
+
+    end
+  end
+
+  class RestoreBaseMethods < Base
+    desc 'restore and restore__copy methods'
+
+    setup do
+      @receiver.send :include, subject
+    end
+
+    should "be instance methods" do
+      assert_respond_to :restore_copy, @receiver.new
+      assert_respond_to :restore, @receiver.new
+    end
+
+    should "create a config[:restore] instance passing self and the binding's config and call run on restore_copy" do
+      object = @receiver.new
+      object.restore_copy
+
+      assert_ran_restore(object)
+
+      # No Destroy
+      assert_equal 0, @receiver.calls.size
+    end
+
+    should "run restore_copy and then destroy when restore called" do
+      object = @receiver.new
+      object.restore
+
+      assert_ran_restore(object)
+
+      assert_equal [:destroy], @receiver.calls[0]
+    end
+  end
+
+  class RestoreRelationMethods < Base
+    desc 'restore_all method for relation'
+
+    setup do
+      @receiver.send :include, subject
+    end
+
+    should "include RelationExtensions on the singleton_class for @relation" do
+      assert_respond_to :restore_all, @receiver.send(:relation)
+      assert_includes subject::RelationExtensions, @receiver.send(:relation).singleton_class.ancestors
+    end
+
+    should "call restore on all items in the query" do
+      query = [@receiver.new, @receiver.new, @receiver.new]
+      relation = @receiver.send(:relation)
+      relation.instance_variable_set(:@records, query)
+      relation.instance_variable_set(:@loaded, true)
+      relation.restore_all
+      assert_equal [[:destroy], [:destroy], [:destroy]], @receiver.calls
+    end
+
+    should "reset the relation" do
+      relation = @receiver.send(:relation)
+      relation.instance_variable_set(:@records, [@receiver.new])
+      relation.instance_variable_set(:@loaded, true)
+      relation.restore_all
+      assert_equal [], relation.instance_variable_get(:@records)
+    end
+
+    should "return an array of restored records" do
+      records = [@receiver.new]
+      relation = @receiver.send(:relation)
+      relation.instance_variable_set(:@records, records)
+      relation.instance_variable_set(:@loaded, true)
+      assert_equal records, relation.restore_all
+    end
+
+    should "not be visible on other Relation instances" do
+      model = Class.new(@ar)
+      model.table_name = 'test'
+      @ar.connection.create_table :test
+      assert_not_respond_to :restore_all, model.send(:relation)
+      @ar.connection.drop_table :test
+    end
+  end
+
+end
+

data/test/unit/binding/active_record_test.rb

@@ -25,7 +25,15 @@ module Undestroy::Binding::ActiveRecord::Test
     desc 'basic instance'
     subject { Undestroy::Binding::ActiveRecord.new @model }
 
-    should have_accessors :config, :model
+    should have_accessors :config, :model, :active
+
+    should "have instance method :active? as alias for :active" do
+      obj = subject
+      obj.active = false
+      assert_equal false, obj.active?
+      obj.active = true
+      assert_equal true, obj.active?
+    end
   end
 
   class AddClassMethod < Base
@@ -36,20 +44,13 @@ module Undestroy::Binding::ActiveRecord::Test
         undef_method :undestroy_model_binding if respond_to?(:undestroy_model_binding)
         remove_possible_method :undestroy_model_binding=
         class << self
-          undef_method :undestroy_model_binding
           undef_method :undestroy
         end
       end
       @model._destroy_callbacks = []
     end
 
-    should "add class_attr called `undestroy_model_binding`" do
-      subject.add @model
-      assert_respond_to :undestroy_model_binding, @model
-      assert_respond_to :undestroy_model_binding=, @model
-    end
-
-    should "add undestroy class method to AR::Base initializing this binding" do
+    should "add undestroy class method to klass initializing this binding" do
       subject.add @model
       assert_respond_to :undestroy, @model
 
@@ -59,8 +60,31 @@ module Undestroy::Binding::ActiveRecord::Test
       assert_equal Hash.new, @model.undestroy_model_binding.config.fields
     end
 
-    should "add before_destroy callback when undestroy called" do
+    should "allow adding to other classes" do
+      new_model = Class.new(@model)
+      subject.add(new_model)
+      assert_respond_to :undestroy, new_model
+      assert_not_respond_to :undestroy, @model
+    end
+
+  end
+
+  class UndestroyExtensionMethod < AddClassMethod
+    desc 'added undestroy method'
+
+    setup do
       subject.add @model
+    end
+
+    should "add class_attr called `undestroy_model_binding`" do
+      assert_not_respond_to :undestroy_model_binding, @model
+      assert_not_respond_to :undestroy_model_binding=, @model
+      @model.undestroy
+      assert_respond_to :undestroy_model_binding, @model
+      assert_respond_to :undestroy_model_binding=, @model
+    end
+
+    should "add before_destroy callback when undestroy called" do
       archive_class = Undestroy::Test::Fixtures::Archive
       @model.undestroy :internals => { :archive => archive_class }
 
@@ -75,20 +99,94 @@ module Undestroy::Binding::ActiveRecord::Test
     end
 
     should "only add callbacks once" do
-      subject.add @model
       @model.undestroy
       @model.undestroy
       assert_equal 1, @model._destroy_callbacks.size
     end
 
-    should "allow adding to other classes" do
-      new_model = Class.new(@model)
-      subject.add(new_model)
-      assert_respond_to :undestroy_model_binding, new_model
-      assert_not_respond_to :undestroy_model_binding, @model
+    should "add archived class method" do
+      assert_not_respond_to :archived, @model
+      @model.undestroy
+      assert_respond_to :archived, @model
+    end
+
+    should "add archived method that returns configured target_class after undestroy configured" do
+      @model.undestroy
+      assert_equal @model.undestroy_model_binding.config.target_class, @model.archived
+    end
+
+    should "add restore method" do
+      assert_not_respond_to :restore, @model
+      @model.undestroy
+      assert_respond_to :restore, @model
+    end
+
+    should "add restore method that passes args to target_class find and then calls 'restore' on each record" do
+      @model.undestroy
+      fixture = Class.new do
+        cattr_accessor :calls, :return_val
+        self.calls = []
+        self.return_val = nil
+
+        def self.find(*args)
+          self.calls << [:find, args]
+          self.return_val
+        end
+
+        def restore
+          self.calls << [:restore]
+        end
+      end
+
+      @model.undestroy_model_binding.config.target_class = fixture
+
+      fixture.return_val = fixture.new
+      @model.restore(:foo, :bar => :baz)
+
+      assert_equal [:find, [:foo, { :bar => :baz }]], fixture.calls[0]
+      assert_equal [:restore], fixture.calls[1]
+
+      fixture.calls = []
+
+      fixture.return_val = [fixture.new, fixture.new]
+      @model.restore
+
+      assert_equal [:restore], fixture.calls[1]
+      assert_equal [:restore], fixture.calls[2]
+      assert_nil fixture.calls[3]
+    end
+
+    should "add destroy! instance method that calls destroy without archival" do
+      @model.class_eval do
+        cattr_accessor :calls
+        self.calls = []
+
+        def destroy
+          calls << [:destroy, undestroy_model_binding.active]
+        end
+      end
+      @model.undestroy
+      assert_respond_to :destroy!, @model.new
+
+      @model.new.destroy!
+
+      assert_equal [:destroy, false], @model.calls[0]
     end
   end
 
+  class LoadModelsClassMethod < Base
+    desc 'load_models class method'
+    include Undestroy::Test::Helpers::ModelLoading
+
+    should "force all models to load recursively in supplied path" do
+      path = Undestroy::Test.fixtures_path('load_test', 'models1')
+      assert_loads_models(path) do
+        subject.load_models(path)
+      end
+    end
+
+  end
+
   class InitMethod < Base
     desc 'init method'
 
@@ -111,24 +209,36 @@ module Undestroy::Binding::ActiveRecord::Test
       assert_instance_of Undestroy::Config, binding.config
     end
 
+    should "default :active attr to true" do
+      assert subject.new(@model).active, "Should be active by default"
+    end
+
     should "merge config options onto global config" do
       Undestroy::Config.configure do |config|
         config.fields = {}
       end
       assert_equal Hash.new, subject.new(@model).config.fields
       assert_not_equal subject.new(@model).config.object_id, Undestroy::Config.config.object_id
+      assert_not_equal subject.new(@model).config.fields.object_id, Undestroy::Config.config.fields.object_id
       assert_equal({ :foo => :bar }, subject.new(@model, :fields => { :foo => :bar }).config.fields)
     end
 
+    should "accept block and pass config object to it" do
+      binding = subject.new @model, {} do |config|
+        config.fields = {}
+      end
+      assert_equal Hash.new, binding.config.fields
+    end
+
     should "set config.source_class to value of model" do
       binding = subject.new(@model)
       assert_equal @model, binding.config.source_class
     end
 
-    should "default :table_name to 'archive_{source.table_name}'" do
+    should "default :table_name to '{config.prefix}{source.table_name}'" do
       @model.table_name = :foobar
-      binding = subject.new(@model)
-      assert_equal 'archive_foobar', binding.config.table_name
+      binding = subject.new(@model, :prefix => "prefix_archive_")
+      assert_equal 'prefix_archive_foobar', binding.config.table_name
     end
 
     should "create a target_class if none provided" do
@@ -162,6 +272,24 @@ module Undestroy::Binding::ActiveRecord::Test
       assert_equal 'tmp/alt.db', binding.config.target_class.connection_config[:database]
     end
 
+    should "create class_attribute undestroy_model_binding with no instance writer" do
+      binding = subject.new(@model)
+      assert_respond_to :undestroy_model_binding, binding.config.target_class
+      assert_respond_to :undestroy_model_binding=, binding.config.target_class
+      assert_respond_to :undestroy_model_binding, binding.config.target_class.new
+      assert_not_respond_to :undestroy_model_binding=, binding.config.target_class.new
+    end
+
+    should "set self to the undestroy_model_binding on the target_class" do
+      binding = subject.new(@model)
+      assert_equal binding, binding.config.target_class.undestroy_model_binding
+    end
+
+    should "include Restorable mixin" do
+      binding = subject.new(@model)
+      assert_includes Undestroy::Binding::ActiveRecord::Restorable, binding.config.target_class.ancestors
+    end
+
   end
 
   class BeforeDestroy < Base
@@ -184,5 +312,45 @@ module Undestroy::Binding::ActiveRecord::Test
     end
   end
 
+  class PrefixTableNameMethod < Base
+    desc 'prefix_table_name method'
+    subject { @binding ||= Undestroy::Binding::ActiveRecord.new(@model) }
+
+    should "return {config.prefix}{source.table_name}" do
+      subject.config.prefix = "archive_prefix_"
+      assert_equal "archive_prefix_foo", subject.prefix_table_name("foo")
+    end
+  end
+
+  class DeactivatedMethod < Base
+    desc 'deactivated method'
+    subject { @binding ||= Undestroy::Binding::ActiveRecord.new(@model) }
+
+    should "set activated to false and yield to block" do
+      called = false
+      block = proc {
+        called = true
+        assert !subject.active, "Binding should be deactivated"
+      }
+      subject.deactivated(&block)
+      assert called, "Block should be called"
+    end
+
+    should "always set activated back to original value" do
+      block = proc {}
+      raise_block = proc { raise "FOO" }
+
+      subject.deactivated(&block)
+      assert subject.active, "Should be activated"
+
+      assert_raises { subject.deactivated(&raise_block) }
+      assert subject.active, "Should be activated"
+
+      subject.active = false
+      subject.deactivated(&block)
+      assert !subject.active, "Should not be activated"
+    end
+  end
+
 end
 

data/test/unit/config/field_test.rb

@@ -0,0 +1,86 @@
+require 'assert'
+
+class Undestroy::Config::Field::Test
+
+  class Base < Undestroy::Test::Base
+    desc 'Undestroy::Config::Field class'
+    subject { @subject_class }
+
+    setup do
+      @subject_class = Undestroy::Config::Field
+    end
+  end
+
+  class BasicInstance < Base
+    desc 'basic instance'
+    subject { @subject_class.new :foo, :foo, 'foo' }
+
+    should have_accessors(:name, :type, :raw_value)
+
+  end
+
+  class InitMethod < Base
+    desc 'initialize method'
+
+    should "have 2 required params" do
+      assert_raises(ArgumentError) { subject.new }
+      assert_not_raises { subject.new :foo, :bar, 'val' }
+      assert_not_raises { subject.new(:foo, :bar) { 'val' } }
+    end
+
+    should "store name in name attr" do
+      obj = subject.new :field, :string, 'val'
+      assert_equal :field, obj.name
+    end
+
+    should "store type in type attr" do
+      obj = subject.new :field, :string, 'val'
+      assert_equal :string, obj.type
+    end
+
+    should "store value in raw_value attr if present" do
+      obj = subject.new :field, :string, 'val'
+      assert_equal 'val', obj.raw_value
+    end
+
+    should "store optional block in raw_value" do
+      block = proc { "foo" }
+      obj = subject.new :field, :string, 'val', &block
+      assert_equal block, obj.raw_value
+    end
+
+    should "raise if a block and a value are not passed" do
+      assert_raises(ArgumentError) { subject.new :field, :string }
+    end
+  end
+
+  class ValueMethod < Base
+    desc 'value method'
+
+    should "return raw_value if not callable" do
+      obj = subject.new :field, :string, 'val'
+      assert_equal 'val', obj.value
+    end
+
+    should "evaluate raw_value with passed args" do
+      block = proc { |args| args }
+      block2 = proc { |arg1, arg2| [arg1, arg2] }
+      obj = subject.new :field, :string, &block
+      obj2 = subject.new :field, :string, &block2
+      assert_equal 1, obj.value(1)
+      assert_equal [1, 2], obj2.value(1, 2)
+    end
+  end
+
+  class SortOperator < Base
+    desc 'sort operator'
+
+    should "sort by alphanumeric field name" do
+      names = ["orange", :apple, "banana", "orange2"]
+      fields = names.collect { |name| Undestroy::Config::Field.new(name, :type, 'val') }
+      assert_equal names.collect(&:to_s).sort, fields.sort.collect { |f| f.name.to_s }
+    end
+  end
+
+end
+