@@ -4,3 +4,4 @@ coverage .svn/ pkg *.swp +tags .gitignore @@ -65,8 +65,47 @@ module Shoulda # :nodoc: end end + # Macro that creates a test asserting that a record of the given class was + # created. + # + # Example: + # + # context "creating a post" do + # setup { Post.create(post_attributes) } + # should_create :post + # end + def should_create(class_name) + should_change_record_count_of(class_name, 1, 'create') + end + + # Macro that creates a test asserting that a record of the given class was + # destroyed. + # + # Example: + # + # context "destroying a post" do + # setup { Post.first.destroy } + # should_destroy :post + # end + def should_destroy(class_name) + should_change_record_count_of(class_name, -1, 'destroy') + end + private + def should_change_record_count_of(class_name, amount, action) # :nodoc: + klass = class_name.to_s.camelize.constantize + before = lambda do + @_before_change_record_count = klass.count + end + human_name = class_name.to_s.humanize.downcase + should "#{action} a #{human_name}", :before => before do + assert_equal @_before_change_record_count + amount, + klass.count, + "Expected to #{action} a #{human_name}" + end + end + include Shoulda::Private end end lib/shoulda/macros.rb @@ -238,4 +238,68 @@ class HelpersTest < Test::Unit::TestCase # :nodoc: end end end + + context "given one treat exists and one post exists" do + setup do + Treat.create! + Post.create!(:title => 'title', :body => 'body', :user_id => 1) + end + + teardown do + Treat.delete_all + Post.delete_all + end + + context "creating a treat" do + setup do + Treat.create! + end + + should_create :treat + should_fail do + should_create :post + end + end + + context "creating a treat and a post" do + setup do + Treat.create! + Post.create!(:title => 'title 2', :body => 'body', :user_id => 1) + end + + should_create :treat + should_create :post + end + + context "destroying a treat" do + setup do + Treat.first.destroy + end + + should_destroy :treat + should_fail do + should_destroy :post + end + end + + context "destroying a treat and a post" do + setup do + Treat.first.destroy + Post.first.destroy + end + + should_destroy :treat + should_destroy :post + end + + context "doing nothing" do + should_fail do + should_create :treat + end + + should_fail do + should_destroy :treat + end + end + end end test/other/helpers_test.rb 712a02615a2d120a46262b845d7cb338d9203c5e Joe Ferris joe.r.ferris@gmail.com http://github.com/rmm5t/shoulda/commit/673ffa66974502e1f12c70fab32a19dc91d365e1 673ffa66974502e1f12c70fab32a19dc91d365e1 2009-05-05T18:16:12-07:00 2009-05-05T18:16:12-07:00 Added should_create and should_destroy [#190 state:resolved] 3c3e054e6f0d9902901b5b98dff4515e5527f233 Joe Ferris joe.r.ferris@gmail.com