fix html writer test

apioo · Oct 11, 2014 · e79e328 · e79e328
1 parent 6a7e484
commit e79e328
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tests/PSX/Data/Writer/HtmlTest.php b/tests/PSX/Data/Writer/HtmlTest.php
@@ -251,15 +251,25 @@ public function testSetExistingTemplateFile()
 
 	public function testIsContentTypeSupported()
 	{
-		$writer = new Html();
+		$template = $this->getMock('PSX\TemplateInterface');
+		$router   = $this->getMockBuilder('PSX\Loader\ReverseRouter')
+			->disableOriginalConstructor()
+			->getMock();
+
+		$writer = new Html($template, $router);
 
 		$this->assertTrue($writer->isContentTypeSupported('text/html'));
 		$this->assertFalse($writer->isContentTypeSupported('application/xml'));
 	}
 
 	public function testGetContentType()
 	{
-		$writer = new Html();
+		$template = $this->getMock('PSX\TemplateInterface');
+		$router   = $this->getMockBuilder('PSX\Loader\ReverseRouter')
+			->disableOriginalConstructor()
+			->getMock();
+
+		$writer = new Html($template, $router);
 
 		$this->assertEquals('text/html', $writer->getContentType());
 	}