Define state machine for Friendship. Change the update calls for the …

…state methods

lib/has_friendship/friendable.rb

@@ -51,8 +51,7 @@ def friend_request(friend)
 
       def accept_request(friend)
         on_relation_with(friend) do |one, other|
-          HasFriendship::Friendship.find_unblocked_friendship(one, other)
-                                   .update(status: 2)
+          HasFriendship::Friendship.find_unblocked_friendship(one, other).accept!
         end
       end
 
@@ -66,8 +65,7 @@ def decline_request(friend)
 
       def block_friend(friend)
         on_relation_with(friend) do |one, other|
-          HasFriendship::Friendship.find_unblocked_friendship(one, other)
-                                   .update(status: 3, blocker_id: self.id)
+          HasFriendship::Friendship.find_unblocked_friendship(one, other).block!
         end
       end
 

lib/has_friendship/friendship.rb

@@ -1,5 +1,20 @@
 module HasFriendship
   class Friendship < ActiveRecord::Base
+
+    enum status: { pending: 0, requested: 1, accepted: 2, blocked: 3 } do
+      event :accept do
+        transition [:pending, :requested] => :accepted
+      end
+
+      event :block do
+        before do
+          self.blocker_id = self.friendable.id
+        end
+
+        transition all - [:blocked] => :blocked
+      end
+    end
+
     def self.relation_attributes(one, other, status: nil)
       attr = {
         friendable_id: one.id,

spec/rails_helper.rb

@@ -7,6 +7,7 @@
 
 require 'shoulda'
 require 'support/friendship_macro'
+require 'stateful_enum'
 
 load 'internal/db/schema.rb'