1 files changed, 127 insertions, 0 deletions
diff --git a/tests/attachment/delete_test.php b/tests/attachment/delete_test.php
new file mode 100644
index 0000000000..f1835dd37a
--- /dev/null
+++ b/tests/attachment/delete_test.php
@@ -0,0 +1,127 @@
+<?php
+/**
+*
+* This file is part of the phpBB Forum Software package.
+*
+* @copyright (c) phpBB Limited <https://www.phpbb.com>
+* @license GNU General Public License, version 2 (GPL-2.0)
+*
+* For full copyright and license information, please see
+* the docs/CREDITS.txt file.
+*
+*/
+
+require_once(dirname(__FILE__) . '/../../phpBB/includes/functions_admin.php');
+
+class phpbb_attachment_delete_test extends \phpbb_database_test_case
+{
+	/** @var \phpbb\config\config */
+	protected $config;
+
+	/** @var \phpbb\db\driver\driver_interface */
+	protected $db;
+
+	/** @var \phpbb\filesystem\filesystem */
+	protected $filesystem;
+
+	/** @var \phpbb\attachment\resync */
+	protected $resync;
+
+	/** @var \phpbb\attachment\delete */
+	protected $attachment_delete;
+
+	protected $phpbb_root_path;
+
+	public function getDataSet()
+	{
+		return $this->createXMLDataSet(dirname(__FILE__) . '/fixtures/resync.xml');
+	}
+
+	public function setUp()
+	{
+		global $db, $phpbb_root_path;
+
+		parent::setUp();
+
+		$this->config = new \phpbb\config\config(array());
+		$this->db = $this->new_dbal();
+		$db = $this->db;
+		$this->resync = new \phpbb\attachment\resync($this->db);
+		$this->filesystem = $this->getMock('\phpbb\filesystem\filesystem', array('remove', 'exists'));
+		$this->filesystem->expects($this->any())
+			->method('remove')
+			->willReturn(false);
+		$this->filesystem->expects($this->any())
+			->method('exists')
+			->willReturn(true);
+		$this->phpbb_root_path = $phpbb_root_path;
+		$this->dispatcher = new \phpbb_mock_event_dispatcher();
+		$this->attachment_delete = new \phpbb\attachment\delete($this->config, $this->db, $this->dispatcher, $this->filesystem, $this->resync, $phpbb_root_path);
+	}
+
+	public function data_attachment_delete()
+	{
+		return array(
+			array('attach', '', false, false),
+			array('meh', 5, false, 0),
+			array('attach', array(5), false, 0),
+			array('attach', array(1,2), false, 2),
+			array('attach', array(1,2), true, 2),
+			array('post', 5, false, 0),
+			array('topic', 5, false, 0),
+			array('topic', 1, true, 3),
+			array('user', 1, false, 0),
+		);
+	}
+
+	/**
+	 * @dataProvider data_attachment_delete
+	 */
+	public function test_attachment_delete($mode, $ids, $resync, $expected)
+	{
+		// We need to reset the attachment ID sequence to properly test this
+		if ($this->db->get_sql_layer() === 'postgres')
+		{
+			$sql = 'ALTER SEQUENCE phpbb_attachments_seq RESTART WITH 1';
+			$this->db->sql_query($sql);
+		}
+
+		$this->assertSame($expected, $this->attachment_delete->delete($mode, $ids, $resync));
+	}
+
+	public function data_attachment_unlink()
+	{
+		return array(
+			array(true, true, true),
+			array(true, false, false),
+			array(true, true, false, true),
+		);
+	}
+
+	/**
+	 * @dataProvider data_attachment_unlink
+	 */
+	public function test_attachment_delete_success($remove_success, $exists_success, $expected, $throw_exception = false)
+	{
+		$this->filesystem = $this->getMock('\phpbb\filesystem\filesystem', array('remove', 'exists'));
+		if ($throw_exception)
+		{
+			$this->filesystem->expects($this->any())
+				->method('remove')
+				->willThrowException(new \phpbb\filesystem\exception\filesystem_exception);;
+		}
+		else
+		{
+			$this->filesystem->expects($this->any())
+				->method('remove')
+				->willReturn($remove_success);
+		}
+
+		$this->filesystem->expects($this->any())
+			->method('exists')
+			->willReturn($exists_success);
+
+		$this->attachment_delete = new \phpbb\attachment\delete($this->config, $this->db, $this->dispatcher, $this->filesystem, $this->resync, $this->phpbb_root_path);
+		$this->assertSame($expected, $this->attachment_delete->unlink_attachment('foobar'));
+	}
+}