From 52f478b289b3edd9c612ab5997f41204bc8e0d76 Mon Sep 17 00:00:00 2001
From: Jordi Boggiano <j.boggiano@seld.be>
Date: Sun, 28 Jul 2013 23:19:20 +0200
Subject: [PATCH] Fix tests

---
 src/Monolog/Handler/NewRelicHandler.php       |  2 ++
 tests/Monolog/Handler/NewRelicHandlerTest.php | 12 ++++++------
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/Monolog/Handler/NewRelicHandler.php b/src/Monolog/Handler/NewRelicHandler.php
index d647ba21..0b83b968 100644
--- a/src/Monolog/Handler/NewRelicHandler.php
+++ b/src/Monolog/Handler/NewRelicHandler.php
@@ -11,6 +11,8 @@
 
 namespace Monolog\Handler;
 
+use Monolog\Logger;
+
 /**
  * Class to record a log on a NewRelic application
  *
diff --git a/tests/Monolog/Handler/NewRelicHandlerTest.php b/tests/Monolog/Handler/NewRelicHandlerTest.php
index 249fbad1..583c164d 100644
--- a/tests/Monolog/Handler/NewRelicHandlerTest.php
+++ b/tests/Monolog/Handler/NewRelicHandlerTest.php
@@ -15,26 +15,26 @@ use Monolog\TestCase;
 use Monolog\Logger;
 
 class NewRelicHandlerTest extends TestCase
-{    
+{
     /**
      * @expectedException Monolog\Handler\MissingExtensionException
      */
     public function testThehandlerThrowsAnExceptionIfTheNRExtensionIsNotLoaded()
     {
         $handler = new StubNewRelicHandlerWithoutExtension();
-        $handler->handle($this->getRecord());
+        $handler->handle($this->getRecord(Logger::ERROR));
     }
-    
+
     public function testThehandlerCanHandleTheRecord()
     {
         $handler = new StubNewRelicHandler();
-        $handler->handle($this->getRecord());
+        $handler->handle($this->getRecord(Logger::ERROR));
     }
-    
+
     public function testThehandlerCanAddParamsToTheNewRelicTrace()
     {
         $handler = new StubNewRelicHandler();
-        $handler->handle($this->getRecord(100, 'log message', array('a' => 'b')));
+        $handler->handle($this->getRecord(Logger::ERROR, 'log message', array('a' => 'b')));
     }
 }