From 10f5e3c74612e1cf97e2d54fec52e66ad38e3b6c Mon Sep 17 00:00:00 2001
From: Christophe Coevoet <stof@notk.org>
Date: Fri, 6 May 2011 21:24:27 +0200
Subject: [PATCH] Added support of ALERT and CRITICAL in TestHandler

---
 src/Monolog/Handler/TestHandler.php | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/Monolog/Handler/TestHandler.php b/src/Monolog/Handler/TestHandler.php
index dd7e8d94..5667645a 100644
--- a/src/Monolog/Handler/TestHandler.php
+++ b/src/Monolog/Handler/TestHandler.php
@@ -30,6 +30,16 @@ class TestHandler extends AbstractHandler
         return $this->records;
     }
 
+    public function hasAlert($record)
+    {
+        return $this->hasRecord($record, Logger::ALERT);
+    }
+
+    public function hasCritical($record)
+    {
+        return $this->hasRecord($record, Logger::CRITICAL);
+    }
+
     public function hasError($record)
     {
         return $this->hasRecord($record, Logger::ERROR);
@@ -50,6 +60,16 @@ class TestHandler extends AbstractHandler
         return $this->hasRecord($record, Logger::DEBUG);
     }
 
+    public function hasAlertRecords()
+    {
+        return isset($this->recordsByLevel[Logger::ALERT]);
+    }
+
+    public function hasCriticalRecords()
+    {
+        return isset($this->recordsByLevel[Logger::CRITICAL]);
+    }
+
     public function hasErrorRecords()
     {
         return isset($this->recordsByLevel[Logger::ERROR]);