[ticket/14168] Move image check and don't use trigger_error()

PHPBB3-14168

2 files changed, 154 insertions, 5 deletions

diff --git a/tests/attachment/fixtures/resync.xml b/tests/attachment/fixtures/resync.xml
index d15f6d17f9..6e2cc62f68 100644
--- a/tests/attachment/fixtures/resync.xml
+++ b/tests/attachment/fixtures/resync.xml
@@ -36,6 +36,40 @@
             <value>1</value>
         </row>
     </table>
+    <table name="phpbb_extensions">
+        <column>extension</column>
+        <column>group_id</column>
+        <row>
+            <value>jpg</value>
+            <value>1</value>
+        </row>
+        <row>
+            <value>png</value>
+            <value>1</value>
+        </row>
+    </table>
+    <table name="phpbb_extension_groups">
+        <column>cat_id</column>
+        <column>group_id</column>
+        <column>download_mode</column>
+        <column>upload_icon</column>
+        <column>max_filesize</column>
+        <column>allow_group</column>
+        <column>allow_in_pm</column>
+        <column>allowed_forums</column>
+        <column>group_name</column>
+        <row>
+            <value>1</value>
+            <value>1</value>
+            <value>1</value>
+            <value> </value>
+            <value>1000</value>
+            <value>1</value>
+            <value>1</value>
+            <value>a:1:{i:0;i:1;}</value>
+            <value>Images</value>
+        </row>
+    </table>
     <table name="phpbb_posts">
         <column>post_id</column>
         <column>post_text</column>
diff --git a/tests/attachment/upload_test.php b/tests/attachment/upload_test.php
index 03d688cc1f..d1007dbd77 100644
--- a/tests/attachment/upload_test.php
+++ b/tests/attachment/upload_test.php
@@ -74,7 +74,9 @@ class phpbb_attachment_upload_test extends \phpbb_database_test_case
 		parent::setUp();
 
 		$this->auth = new \phpbb\auth\auth();
-		$this->config = new \phpbb\config\config(array());
+		$this->config = new \phpbb\config\config(array(
+			'upload_path'	=> '../attachment/fixtures/',
+		));
 		$config = $this->config;
 		$this->db = $this->new_dbal();
 		$this->cache = new \phpbb\cache\service(new \phpbb\cache\driver\dummy(), $this->config, $this->db, $phpbb_root_path, $phpEx);
@@ -153,14 +155,33 @@ class phpbb_attachment_upload_test extends \phpbb_database_test_case
 	public function data_upload()
 	{
 		return array(
-			array('foobar', 1, false, array(
+			array('foobar', 1, false,
+				array(),
+				array(
 					'error' => array(
 						'Upload initiated but no valid file upload form found.',
 					),
 					'post_attach'	=> false,
 				)
 			),
-			array('foobar', 1, true, array(
+			array('foobar', 1, true,
+				array(
+					'realname'		=> 'foobar.jpg',
+					'type'			=> 'jpg',
+					'size'			=> 100,
+				),
+				array(
+					'error' => array(
+						'NOT_UPLOADED',
+						'The image file you tried to attach is invalid.',
+					),
+					'post_attach'	=> false,
+					'thumbnail'		=> 0,
+				)
+			),
+			array('foobar', 1, true,
+				array(),
+				array(
 					'error' => array(
 						'NOT_UPLOADED',
 					),
@@ -174,9 +195,9 @@ class phpbb_attachment_upload_test extends \phpbb_database_test_case
 	/**
 	 * @dataProvider data_upload
 	 */
-	public function test_upload($form_name, $forum_id, $local, $expected)
+	public function test_upload($form_name, $forum_id, $local, $filedata, $expected)
 	{
-		$filedata = $this->upload->upload($form_name, $forum_id, $local);
+		$filedata = $this->upload->upload($form_name, $forum_id, $local, '', false, $filedata);
 
 		$this->assertSame($expected, $filedata);
 	}
@@ -229,4 +250,98 @@ class phpbb_attachment_upload_test extends \phpbb_database_test_case
 			'post_attach'	=> false,
 		), $filedata);
 	}
+
+	public function data_image_not_image()
+	{
+		return array(
+			array(false),
+			array(true),
+		);
+	}
+
+	/**
+	 * @dataProvider data_image_not_image
+	 */
+	public function test_image_not_image($plupload_active)
+	{
+		$filespec = $this->getMock('\phpbb\files\filespec',
+			array(
+				'init_error',
+				'is_image',
+				'move_file',
+				'is_uploaded',
+			),
+			array(
+				$this->filesystem,
+				$this->language,
+				$this->php_ini,
+				new \FastImageSize\FastImageSize(),
+				$this->phpbb_root_path,
+				$this->mimetype_guesser,
+				$this->plupload
+			));
+		$filespec->set_upload_namespace($this->files_upload);
+		$filespec->expects($this->any())
+			->method('init_error')
+			->willReturn(false);
+		$filespec->expects($this->any())
+			->method('is_image')
+			->willReturn(false);
+		$filespec->expects($this->any())
+			->method('is_uploaded')
+			->willReturn(true);
+		$filespec->expects($this->any())
+			->method('move_file')
+			->willReturn(false);
+		$this->container->set('files.filespec', $filespec);
+		$factory_mock = $this->getMockBuilder('\phpbb\files\factory')
+			->disableOriginalConstructor()
+			->getMock();
+		$factory_mock->expects($this->any())
+			->method('get')
+			->willReturn($filespec);
+		$this->container->set('files.types.local', new \phpbb\files\types\local(
+			$factory_mock,
+			$this->language,
+			$this->php_ini,
+			$this->request
+		));
+
+		$plupload = $this->getMockBuilder('\phpbb\plupload\plupload')
+			->disableOriginalConstructor()
+			->getMock();
+		$plupload->expects($this->any())
+			->method('is_active')
+			->willReturn($plupload_active);
+		if ($plupload_active)
+		{
+			$plupload->expects($this->once())
+				->method('emit_error')
+				->with(104, 'ATTACHED_IMAGE_NOT_IMAGE')
+				->willReturn(false);
+		}
+		$this->upload = new \phpbb\attachment\upload(
+			$this->auth,
+			$this->cache,
+			$this->config,
+			$this->files_upload,
+			$this->language,
+			$this->mimetype_guesser,
+			$this->phpbb_dispatcher,
+			$plupload,
+			$this->user,
+			$this->phpbb_root_path
+		);
+
+		$filedata = $this->upload->upload('foobar', 1, true, '', false, array(
+			'realname'		=> 'foobar.jpg',
+			'type'			=> 'jpg',
+			'size'			=> 100,
+		));
+		$this->assertEquals(array(
+			'error'			=> array('The image file you tried to attach is invalid.'),
+			'post_attach'	=> false,
+			'thumbnail'		=> 0,
+		), $filedata);
+	}
 }