Override the count method to apply deletionTimestamp to the query (#19)

* Overload the count method and apply the deletionTimestamp to the query

* Add test for overloaded count method

Author: rrigby

src/Mapping.php

@@ -82,6 +82,20 @@ public function findAll()
         return $this->map(parent::findAll());
     }
 
+    /**
+     * Counts all records.
+     *
+     * @return int
+     */
+    public function count(string $column = '*')
+    {
+        if ($this->definition->getDeletionTimestamp()) {
+            $this->isNull($this->prefixTableNameTo($this->definition->getDeletionTimestamp()));
+        }
+  
+        return parent::count($column);
+    }
+
     /**
      * Maps the provided array into the database.
      *

tests/MappingTest.php

@@ -72,6 +72,15 @@ public function testFindOne()
         $this->assertEquals(400, $customer['orders'][0]['items'][1]['amount']);
     }
 
+    public function testCount()
+    {
+        $this->assertSame(2, $this->getMapping()->count());
+
+        $this->getMapping()->eq('id', 1)->remove();
+
+        $this->assertSame(1, $this->getMapping()->count());
+    }
+
     public function testInsert()
     {
         $customer = [