From df1a800d9466c5df298ea4fa4e3118648c6599ad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Juanjo=20Baz=C3=A1n?= <jjbazan@gmail.com>
Date: Sun, 19 Jul 2015 14:46:21 +0200
Subject: [PATCH] adds editable methods to Debate

---
 app/models/debate.rb       | 10 +++++++++-
 spec/models/debate_spec.rb | 39 +++++++++++++++++++++++++++++++++++++-
 2 files changed, 47 insertions(+), 2 deletions(-)

diff --git a/app/models/debate.rb b/app/models/debate.rb
index 01dfa07c3..5104bec97 100644
--- a/app/models/debate.rb
+++ b/app/models/debate.rb
@@ -3,7 +3,7 @@ class Debate < ActiveRecord::Base
   acts_as_votable
   acts_as_commentable
   acts_as_taggable
-  
+
   belongs_to :author, class_name: 'User', foreign_key: 'author_id'
 
   validates :title, presence: true
@@ -24,4 +24,12 @@ class Debate < ActiveRecord::Base
     votes_for.size
   end
 
+  def editable?
+    total_votes == 0
+  end
+
+  def editable_by?(user)
+    editable? && author == user
+  end
+
 end
\ No newline at end of file
diff --git a/spec/models/debate_spec.rb b/spec/models/debate_spec.rb
index f277f1d7b..91797cb71 100644
--- a/spec/models/debate_spec.rb
+++ b/spec/models/debate_spec.rb
@@ -29,5 +29,42 @@ describe Debate do
     @debate.terms_of_service = nil
     expect(@debate).to_not be_valid
   end
-  
+
+  describe "#editable?" do
+    before(:each) do
+      @debate = create(:debate)
+    end
+
+    it "should be true if debate has no votes yet" do
+      expect(@debate.total_votes).to eq(0)
+      expect(@debate.editable?).to be true
+    end
+
+    it "should be false if debate has votes" do
+      create(:vote, votable: @debate)
+      expect(@debate.total_votes).to eq(1)
+      expect(@debate.editable?).to be false
+    end
+  end
+
+  describe "#editable_by?" do
+    before(:each) do
+      @debate = create(:debate)
+    end
+
+    it "should be true if user is the author and debate is editable" do
+      expect(@debate.editable_by?(@debate.author)).to be true
+    end
+
+    it "should be false if debate is not editable" do
+      create(:vote, votable: @debate)
+      expect(@debate.editable_by?(@debate.author)).to be false
+    end
+
+    it "should be false if user is not the author" do
+      expect(@debate.editable_by?(create(:user))).to be false
+
+    end
+  end
+
 end