Commit 6b589235 by Ryan Bigg

Merge pull request #21 from chrislwade/test-coverage-hm-hmt

Add test coverage for has_many and has_many :through behavior.
parents f69132e2 d2a21517
...@@ -13,6 +13,11 @@ ActiveRecord::Base.connection.execute 'CREATE TABLE paranoid_models (id INTEGER ...@@ -13,6 +13,11 @@ ActiveRecord::Base.connection.execute 'CREATE TABLE paranoid_models (id INTEGER
ActiveRecord::Base.connection.execute 'CREATE TABLE featureful_models (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME, name VARCHAR(32))' ActiveRecord::Base.connection.execute 'CREATE TABLE featureful_models (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME, name VARCHAR(32))'
ActiveRecord::Base.connection.execute 'CREATE TABLE plain_models (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME)' ActiveRecord::Base.connection.execute 'CREATE TABLE plain_models (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME)'
ActiveRecord::Base.connection.execute 'CREATE TABLE callback_models (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME)' ActiveRecord::Base.connection.execute 'CREATE TABLE callback_models (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME)'
ActiveRecord::Base.connection.execute 'CREATE TABLE parent_models (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME)'
ActiveRecord::Base.connection.execute 'CREATE TABLE related_models (id INTEGER NOT NULL PRIMARY KEY, parent_model_id INTEGER NOT NULL, deleted_at DATETIME)'
ActiveRecord::Base.connection.execute 'CREATE TABLE employers (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME)'
ActiveRecord::Base.connection.execute 'CREATE TABLE employees (id INTEGER NOT NULL PRIMARY KEY, deleted_at DATETIME)'
ActiveRecord::Base.connection.execute 'CREATE TABLE jobs (id INTEGER NOT NULL PRIMARY KEY, employer_id INTEGER NOT NULL, employee_id INTEGER NOT NULL, deleted_at DATETIME)'
class ParanoiaTest < Test::Unit::TestCase class ParanoiaTest < Test::Unit::TestCase
def test_plain_model_class_is_not_paranoid def test_plain_model_class_is_not_paranoid
...@@ -68,7 +73,6 @@ class ParanoiaTest < Test::Unit::TestCase ...@@ -68,7 +73,6 @@ class ParanoiaTest < Test::Unit::TestCase
assert_equal 0, model.class.count assert_equal 0, model.class.count
assert_equal 1, model.class.unscoped.count assert_equal 1, model.class.unscoped.count
end end
def test_scoping_behavior_for_paranoid_models def test_scoping_behavior_for_paranoid_models
...@@ -115,48 +119,89 @@ class ParanoiaTest < Test::Unit::TestCase ...@@ -115,48 +119,89 @@ class ParanoiaTest < Test::Unit::TestCase
assert_equal model, ParanoidModel.only_deleted.last assert_equal model, ParanoidModel.only_deleted.last
assert_equal false, ParanoidModel.only_deleted.include?(model2) assert_equal false, ParanoidModel.only_deleted.include?(model2)
end end
def test_default_scope_for_has_many_relationships
parent = ParentModel.create
assert_equal 0, parent.related_models.count
child = parent.related_models.create
assert_equal 1, parent.related_models.count
child.destroy
assert_equal false, child.deleted_at.nil?
assert_equal 0, parent.related_models.count
assert_equal 1, parent.related_models.unscoped.count
end
def test_default_scope_for_has_many_through_relationships
employer = Employer.create
employee = Employee.create
assert_equal 0, employer.jobs.count
assert_equal 0, employer.employees.count
assert_equal 0, employee.jobs.count
assert_equal 0, employee.employers.count
job = Job.create :employer => employer, :employee => employee
assert_equal 1, employer.jobs.count
assert_equal 1, employer.employees.count
assert_equal 1, employee.jobs.count
assert_equal 1, employee.employers.count
employee2 = Employee.create
job2 = Job.create :employer => employer, :employee => employee2
employee2.destroy
assert_equal 2, employer.jobs.count
assert_equal 1, employer.employees.count
job.destroy
assert_equal 1, employer.jobs.count
assert_equal 0, employer.employees.count
assert_equal 0, employee.jobs.count
assert_equal 0, employee.employers.count
end
def test_delete_behavior_for_callbacks def test_delete_behavior_for_callbacks
model = CallbackModel.new model = CallbackModel.new
model.save model.save
model.delete model.delete
assert_equal nil, model.instance_variable_get(:@callback_called) assert_equal nil, model.instance_variable_get(:@callback_called)
end end
def test_destroy_behavior_for_callbacks def test_destroy_behavior_for_callbacks
model = CallbackModel.new model = CallbackModel.new
model.save model.save
model.destroy model.destroy
assert model.instance_variable_get(:@callback_called) assert model.instance_variable_get(:@callback_called)
end end
def test_restore def test_restore
model = ParanoidModel.new model = ParanoidModel.new
model.save model.save
id = model.id id = model.id
model.destroy model.destroy
assert model.destroyed? assert model.destroyed?
model = ParanoidModel.only_deleted.find(id) model = ParanoidModel.only_deleted.find(id)
model.restore! model.restore!
assert_equal false, model.destroyed? assert_equal false, model.destroyed?
end end
def test_real_destroy def test_real_destroy
model = ParanoidModel.new model = ParanoidModel.new
model.save model.save
model.destroy! model.destroy!
assert_equal false, ParanoidModel.unscoped.exists?(model.id) assert_equal false, ParanoidModel.unscoped.exists?(model.id)
end end
def test_real_delete def test_real_delete
model = ParanoidModel.new model = ParanoidModel.new
model.save model.save
model.delete! model.delete!
assert_equal false, ParanoidModel.unscoped.exists?(model.id) assert_equal false, ParanoidModel.unscoped.exists?(model.id)
end end
...@@ -189,3 +234,31 @@ class CallbackModel < ActiveRecord::Base ...@@ -189,3 +234,31 @@ class CallbackModel < ActiveRecord::Base
acts_as_paranoid acts_as_paranoid
before_destroy {|model| model.instance_variable_set :@callback_called, true } before_destroy {|model| model.instance_variable_set :@callback_called, true }
end end
class ParentModel < ActiveRecord::Base
acts_as_paranoid
has_many :related_models
end
class RelatedModel < ActiveRecord::Base
acts_as_paranoid
belongs_to :parent_model
end
class Employer < ActiveRecord::Base
acts_as_paranoid
has_many :jobs
has_many :employees, :through => :jobs
end
class Employee < ActiveRecord::Base
acts_as_paranoid
has_many :jobs
has_many :employers, :through => :jobs
end
class Job < ActiveRecord::Base
acts_as_paranoid
belongs_to :employer
belongs_to :employee
end
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment