diff --git a/test/ImageResizeTest.php b/test/ImageResizeTest.php
index b0a7740..7f7f159 100644
--- a/test/ImageResizeTest.php
+++ b/test/ImageResizeTest.php
@@ -21,6 +21,11 @@
 class ImageResizeTest extends \PHPUnit_Framework_TestCase
 {
 
+    public function setUp()
+    {
+        $_SERVER["DOCUMENT_ROOT"] = __DIR__ . "/../example/";
+    }
+
     public function testShouldBeTrue()
     {
         $this->assertTrue(true);
@@ -116,4 +121,50 @@ public function testImagesShouldNotBeAllowed()
             "size" => "100x200",
             "extension" => "pdf")));
     }
+
+    public function testShouldReturnImage()
+    {
+
+        \Slim\Environment::mock(array(
+            "SCRIPT_NAME" => "/index.php",
+            "PATH_INFO" => "/images/viper-200x200.jpg"
+        ));
+        $app = new \Slim\Slim();
+        $app->get("/foo", function () {
+            echo "Success";
+        });
+
+        $middleware = new \Slim\Middleware\ImageResize(array(
+        ));
+
+        $middleware->setApplication($app);
+        $middleware->setNextMiddleware($app);
+        $middleware->call();
+
+        $this->assertEquals(200, $app->response()->status());
+        $this->assertEquals("image/jpeg", $app->response()->header("Content-Type"));
+    }
+
+    public function testShouldReturnHtml()
+    {
+
+        \Slim\Environment::mock(array(
+            "SCRIPT_NAME" => "/index.php",
+            "PATH_INFO" => "/foo"
+        ));
+        $app = new \Slim\Slim();
+        $app->get("/foo", function () {
+            echo "Success";
+        });
+
+        $middleware = new \Slim\Middleware\ImageResize(array(
+        ));
+
+        $middleware->setApplication($app);
+        $middleware->setNextMiddleware($app);
+        $middleware->call();
+
+        $this->assertEquals(200, $app->response()->status());
+        $this->assertEquals("text/html", $app->response()->header("Content-Type"));
+    }
 }