Update Encouragements Controller specs.

Author: valdeci-dev

app/controllers/encouragements_controller.rb

@@ -1,8 +1,11 @@
 class EncouragementsController < ApplicationController
 	before_action :authenticate_user
+	before_action :authors_are_not_allowed
+	before_action :only_one_encouragement
 
-  def new
 
+  def new
+    @encouragement = Encouragement.new
   end
 
 	def authenticate_user
@@ -11,5 +14,17 @@ def authenticate_user
     unless current_user
       redirect_to achievement_path(@achievement), alert: 'You must be logged in to encourage people'
     end
-  end
+	end
+
+	def authors_are_not_allowed
+		if current_user.id == @achievement.user_id
+			redirect_to achievement_path(@achievement), alert: "You can't encourage yourself."
+		end
+	end
+
+	def only_one_encouragement
+		if Encouragement.exists?(user: current_user, achievement: @achievement)
+			redirect_to achievement_path(@achievement), alert: "You already encouraged it. You can't be so generous."
+		end
+	end
 end

app/views/encouragements/new.html renamed to app/views/encouragements/new.html.erb

@@ -33,22 +33,32 @@
 	end
 
 	context 'achievement author' do
+		before {sign_in(author)}
 		it 'is redirected back to achievement page' do
-
+			get :new, params: {achievement_id: achievement.id}
+			expect(response).to redirect_to(achievement_path(achievement))
 		end
 
 		it 'assigns flash message' do
-
+			get :new, params: {achievement_id: achievement.id}
+			expect(flash[:alert]).to eq("You can't encourage yourself.")
 		end
 	end
 
 	context 'user who already left encouragement for this achievement' do
-		it 'is redirected back to achievement page' do
+		before do
+			sign_in(user)
+			FactoryGirl.create(:encouragement, user:user, achievement: achievement )
+		end
 
+		it 'is redirected back to achievement page' do
+			get :new, params: {achievement_id: achievement.id}
+			expect(response).to redirect_to(achievement_path(achievement))
 		end
 
 		it 'assigns flash message' do
-
+			get :new, params: {achievement_id: achievement.id}
+			expect(flash[:alert]).to eq("You already encouraged it. You can't be so generous.")
 		end
 	end
 end

spec/controllers/encouragements_controller_spec.rb

@@ -1,7 +1,7 @@
 FactoryGirl.define do
   factory :encouragement do
-    user nil
-    achievement nil
-    message "MyString"
+    user
+    achievement
+    message "My Message"
   end
 end