Unit tests

tests/RegistryTest.php

@@ -34,6 +34,26 @@ public function testPersist()
         $registry->persist();
     }
 
+
+    public function testPersistNull()
+    {
+        $firstEntity = null;
+
+        $persister = $this->getMock('\eBayEnterprise\Behat\RegistryExtension\Persister');
+        $persister->expects($this->at(0))
+            ->method('beginTransaction');
+
+        $persister->expects($this->never())
+            ->method('persist');
+
+        $persister->expects($this->at(1))
+            ->method('commitTransaction');
+
+        $registry = new Registry($persister);
+        $registry->append($firstEntity);
+        $registry->persist();
+    }
+
     public function testPersistWithoutPersister()
     {
         $this->setExpectedException('InvalidArgumentException');
@@ -78,6 +98,25 @@ public function testCleanWithoutPersister()
         $this->assertEquals(array(), $registry->getArrayCopy());
     }
 
+    public function testCleanNull()
+    {
+        $firstEntity = null;
+
+        $persister = $this->getMock('\eBayEnterprise\Behat\RegistryExtension\Persister');
+        $persister->expects($this->at(0))
+            ->method('beginTransaction');
+
+        $persister->expects($this->never())
+            ->method('remove');
+
+        $persister->expects($this->at(1))
+            ->method('commitTransaction');
+
+        $registry = new Registry($persister);
+        $registry->append($firstEntity);
+        $registry->reset();
+    }
+
     public function testReload()
     {
         $oldEntity = new \stdClass();
@@ -96,6 +135,19 @@ public function testReload()
         $this->assertSame($newEntity, $registry->findOne(get_class($oldEntity)));
     }
 
+    public function testReloadNull()
+    {
+        $oldEntity = null;
+
+        $persister = $this->getMock('\eBayEnterprise\Behat\RegistryExtension\Persister');
+        $persister->expects($this->never())
+            ->method('reload');
+
+        $registry = new Registry($persister);
+        $registry->append($oldEntity);
+        $registry->reload();
+    }
+
     public function testMergeWithArray()
     {
         $firstEntity = new \stdClass();