add votesDiff method

mohamednagy · Jan 22, 2018 · d6e91dc · d6e91dc
1 parent 81c2313
commit d6e91dc
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/Traits/Votable.php b/src/Traits/Votable.php
@@ -25,4 +25,9 @@ public function downVotesCount()
     {
         return $this->votes()->where('value', 0)->count();
     }
+
+    public function votesDiff()
+    {
+        return $this->upVotesCount() - $this->downVotesCount();
+    }
 }
diff --git a/tests/VotingTest.php b/tests/VotingTest.php
@@ -76,4 +76,17 @@ public function it_returns_total_votes_count()
 
         $this->assertTrue($post->votesCount() == 2);
     }
+
+    /** @test */
+    public function it_returns_votes_diff()
+    {
+        $user = User::create(['name' => 'test']);
+        $user2 = User::create(['name' => 'test2']);
+        $post = Post::create(['name' => 'test post']);
+
+        $user->upVote($post);
+        $user2->downVote($post);
+
+        $this->assertTrue($post->votesDiff() == 0);
+    }
 }