--- /dev/null
+<?php
+
+/*
+ * This file is part of the Monolog package.
+ *
+ * (c) Jordi Boggiano <j.boggiano@seld.be>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Monolog\Handler;
+
+use Monolog\TestCase;
+use Monolog\Logger;
+use InvalidArgumentException;
+
+function mail($to, $subject, $message, $additional_headers = null, $additional_parameters = null)
+{
+ $GLOBALS['mail'][] = func_get_args();
+}
+
+class NativeMailerHandlerTest extends TestCase
+{
+ protected function setUp()
+ {
+ $GLOBALS['mail'] = array();
+ }
+
+ /**
+ * @expectedException InvalidArgumentException
+ */
+ public function testConstructorHeaderInjection()
+ {
+ $mailer = new NativeMailerHandler('spammer@example.org', 'dear victim', "receiver@example.org\r\nFrom: faked@attacker.org");
+ }
+
+ /**
+ * @expectedException InvalidArgumentException
+ */
+ public function testSetterHeaderInjection()
+ {
+ $mailer = new NativeMailerHandler('spammer@example.org', 'dear victim', 'receiver@example.org');
+ $mailer->addHeader("Content-Type: text/html\r\nFrom: faked@attacker.org");
+ }
+
+ /**
+ * @expectedException InvalidArgumentException
+ */
+ public function testSetterArrayHeaderInjection()
+ {
+ $mailer = new NativeMailerHandler('spammer@example.org', 'dear victim', 'receiver@example.org');
+ $mailer->addHeader(array("Content-Type: text/html\r\nFrom: faked@attacker.org"));
+ }
+
+ /**
+ * @expectedException InvalidArgumentException
+ */
+ public function testSetterContentTypeInjection()
+ {
+ $mailer = new NativeMailerHandler('spammer@example.org', 'dear victim', 'receiver@example.org');
+ $mailer->setContentType("text/html\r\nFrom: faked@attacker.org");
+ }
+
+ /**
+ * @expectedException InvalidArgumentException
+ */
+ public function testSetterEncodingInjection()
+ {
+ $mailer = new NativeMailerHandler('spammer@example.org', 'dear victim', 'receiver@example.org');
+ $mailer->setEncoding("utf-8\r\nFrom: faked@attacker.org");
+ }
+
+ public function testSend()
+ {
+ $to = 'spammer@example.org';
+ $subject = 'dear victim';
+ $from = 'receiver@example.org';
+
+ $mailer = new NativeMailerHandler($to, $subject, $from);
+ $mailer->handleBatch(array());
+
+ // batch is empty, nothing sent
+ $this->assertEmpty($GLOBALS['mail']);
+
+ // non-empty batch
+ $mailer->handle($this->getRecord(Logger::ERROR, "Foo\nBar\r\n\r\nBaz"));
+ $this->assertNotEmpty($GLOBALS['mail']);
+ $this->assertInternalType('array', $GLOBALS['mail']);
+ $this->assertArrayHasKey('0', $GLOBALS['mail']);
+ $params = $GLOBALS['mail'][0];
+ $this->assertCount(5, $params);
+ $this->assertSame($to, $params[0]);
+ $this->assertSame($subject, $params[1]);
+ $this->assertStringEndsWith(" test.ERROR: Foo Bar Baz [] []\n", $params[2]);
+ $this->assertSame("From: $from\r\nContent-type: text/plain; charset=utf-8\r\n", $params[3]);
+ $this->assertSame('', $params[4]);
+ }
+
+ public function testMessageSubjectFormatting()
+ {
+ $mailer = new NativeMailerHandler('to@example.org', 'Alert: %level_name% %message%', 'from@example.org');
+ $mailer->handle($this->getRecord(Logger::ERROR, "Foo\nBar\r\n\r\nBaz"));
+ $this->assertNotEmpty($GLOBALS['mail']);
+ $this->assertInternalType('array', $GLOBALS['mail']);
+ $this->assertArrayHasKey('0', $GLOBALS['mail']);
+ $params = $GLOBALS['mail'][0];
+ $this->assertCount(5, $params);
+ $this->assertSame('Alert: ERROR Foo Bar Baz', $params[1]);
+ }
+}