Merge pull request #47 from fvovan/unregister_error_handler

Add unregisterErrorHandler() method
fluent · Feb 15, 2017 · c5a19ba · c5a19ba
2 parents 9a9e10f + a4f2858
commit c5a19ba
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/FluentLogger.php b/src/FluentLogger.php
@@ -165,6 +165,16 @@ public function registerErrorHandler($callable)
         return true;
     }
 
+    /**
+     * unregister error handler
+     *
+     * @return void
+     */
+    public function unregisterErrorHandler()
+    {
+        $this->error_handler = null;
+    }
+
     /**
      * make a various style transport uri with specified host and port.
      * currently, in_forward uses tcp transport only.

diff --git a/tests/Fluent/Logger/BaseLoggerTest.php b/tests/Fluent/Logger/BaseLoggerTest.php
@@ -64,4 +64,15 @@ public function invalidErrorHandlerProvider()
             ),
         );
     }
+
+    public function testUnregisterErrorHandler()
+    {
+        $base = $this->getMockForAbstractClass('Fluent\Logger\FluentLogger');
+        $prop = new \ReflectionProperty($base, 'error_handler');
+        $prop->setAccessible(true);
+        $base->registerErrorHandler(function() {});
+        $this->assertNotNull($prop->getValue($base));
+        $base->unregisterErrorHandler();
+        $this->assertNull($prop->getValue($base));
+    }
 }