Merge pull request #66 from sungwoncho/test/add-create-destroy-specs

Add create destroy after hook specs

lib/has_friendship/friendship.rb

@@ -7,7 +7,7 @@ class Friendship < ActiveRecord::Base
     end
 
     after_destroy do |record|
-      friend.try(:on_friendship_destroyed, record)
+      friendable.try(:on_friendship_destroyed, record)
     end
 
     enum status: { pending: 0, requested: 1, accepted: 2, blocked: 3 } do

spec/has_friendship/friendable_spec.rb

@@ -1,9 +1,8 @@
 require 'rails_helper'
 
-describe User, focus: true do
-
-  let(:user){ User.create(name: 'Jessie') }
-  let(:friend){ User.create(name: 'Heisenberg') }
+describe User do
+  let(:user) { User.create(name: 'Jessie') }
+  let(:friend) { User.create(name: 'Heisenberg') }
 
   describe "association" do
     # TODO: find a way to test condition
@@ -297,5 +296,26 @@
       end
     end
 
+    describe '#on_friendship_destroyed' do
+      context 'when friendship is destroyed' do
+        it 'should be called' do
+          user.friend_request(friend)
+          friendship = find_friendship_record(user, friend)
+          expect(friendship.friendable).to receive(:on_friendship_destroyed)
+          friendship.destroy
+        end
+      end
+    end
+
+    describe '#on_friendship_created' do
+      context 'when friendship is created' do
+        it 'should be called' do
+          user.friend_request(friend) do
+            friendship = find_friendship_record(user, friend)
+            expect(friendship.try(:friendable)).to receive(:on_friendship_created)
+          end
+        end
+      end
+    end
   end
 end