From a467124a4654eb50f6e18783ef4ab5ff35c2edaf Mon Sep 17 00:00:00 2001
From: Justin Hileman <justin@justinhileman.info>
Date: Sat, 14 Dec 2013 12:50:51 -0800
Subject: [PATCH] Improve test coverage.

---
 .../Mustache/Test/Cache/AbstractCacheTest.php | 44 +++++++++++++++++++
 test/Mustache/Test/EngineTest.php             |  2 +
 2 files changed, 46 insertions(+)
 create mode 100644 test/Mustache/Test/Cache/AbstractCacheTest.php

diff --git a/test/Mustache/Test/Cache/AbstractCacheTest.php b/test/Mustache/Test/Cache/AbstractCacheTest.php
new file mode 100644
index 0000000..142c12e
--- /dev/null
+++ b/test/Mustache/Test/Cache/AbstractCacheTest.php
@@ -0,0 +1,44 @@
+<?php
+
+/*
+ * This file is part of Mustache.php.
+ *
+ * (c) 2013 Justin Hileman
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+class Mustache_Test_Cache_AbstractCacheTest extends PHPUnit_Framework_TestCase
+{
+    public function testGetSetLogger()
+    {
+         $cache  = new CacheStub();
+         $logger = new Mustache_Logger_StreamLogger('php://stdout');
+         $cache->setLogger($logger);
+         $this->assertSame($logger, $cache->getLogger());
+    }
+
+    /**
+     * @expectedException Mustache_Exception_InvalidArgumentException
+     */
+    public function testSetLoggerThrowsExceptions()
+    {
+        $cache  = new CacheStub();
+        $logger = new StdClass();
+        $cache->setLogger($logger);
+    }
+}
+
+class CacheStub extends Mustache_Cache_AbstractCache
+{
+    public function load($key)
+    {
+        // nada
+    }
+
+    public function cache($key, $value)
+    {
+        // nada
+    }
+}
diff --git a/test/Mustache/Test/EngineTest.php b/test/Mustache/Test/EngineTest.php
index 1c71158..1ab3078 100644
--- a/test/Mustache/Test/EngineTest.php
+++ b/test/Mustache/Test/EngineTest.php
@@ -45,6 +45,7 @@ class Mustache_Test_EngineTest extends PHPUnit_Framework_TestCase
                 'bar' => 'BAR',
             ),
             'escape'  => 'strtoupper',
+            'entity_flags' => ENT_QUOTES,
             'charset' => 'ISO-8859-1',
         ));
 
@@ -54,6 +55,7 @@ class Mustache_Test_EngineTest extends PHPUnit_Framework_TestCase
         $this->assertEquals('{{ foo }}', $partialsLoader->load('foo'));
         $this->assertContains('__whot__', $mustache->getTemplateClassName('{{ foo }}'));
         $this->assertEquals('strtoupper', $mustache->getEscape());
+        $this->assertEquals(ENT_QUOTES, $mustache->getEntityFlags());
         $this->assertEquals('ISO-8859-1', $mustache->getCharset());
         $this->assertTrue($mustache->hasHelper('foo'));
         $this->assertTrue($mustache->hasHelper('bar'));