Moved office mimetype update to a repair step

1 files changed, 107 insertions, 0 deletions

diff --git a/tests/lib/repair/repairmimetypes.php b/tests/lib/repair/repairmimetypes.php
new file mode 100644
index 00000000000..42af116b72d
--- /dev/null
+++ b/tests/lib/repair/repairmimetypes.php
@@ -0,0 +1,107 @@
+<?php
+/**
+ * Copyright (c) 2014 Vincent Petry <pvince81@owncloud.com>
+ * This file is licensed under the Affero General Public License version 3 or
+ * later.
+ * See the COPYING-README file.
+ */
+
+/**
+ * Tests for the converting of legacy storages to home storages.
+ *
+ * @see \OC\Repair\RepairMimeTypes
+ */
+class TestRepairMimeTypes extends PHPUnit_Framework_TestCase {
+
+	/** @var \OC\RepairStep */
+	private $repair;
+
+	private $storage;
+
+	public function setUp() {
+		$this->storage = new \OC\Files\Storage\Temporary(array());
+
+		$this->repair = new \OC\Repair\RepairMimeTypes();
+	}
+
+	public function tearDown() {
+		$this->storage->getCache()->clear();
+		$sql = 'DELETE FROM `*PREFIX*storages` WHERE id = ?';
+		\OC_DB::executeAudited($sql, array($this->storage->getId()));
+		$this->clearMimeTypes();
+
+		DummyFileCache::clearCachedMimeTypes();
+	}
+
+	private function clearMimeTypes() {
+		$sql = 'DELETE FROM `*PREFIX*mimetypes`';
+		\OC_DB::executeAudited($sql);
+	}
+
+	private function addEntries($entries) {
+		// create files for the different extensions, this
+		// will also automatically create the corresponding mime types
+		foreach ($entries as $entry) {
+			$this->storage->getCache()->put(
+				$entry[0],
+				array(
+					'size' => 0,
+					'mtime' => 0,
+					'mimetype' => $entry[1]
+				)
+			);
+		}
+
+	}
+
+	private function checkEntries($entries) {
+		foreach ($entries as $entry) {
+			$data = $this->storage->getCache()->get($entry[0]);
+			$this->assertEquals($entry[1], $data['mimetype']);
+		}
+	}
+
+	/**
+	 * Test renaming and splitting old office mime types
+	 */
+	public function testRenameOfficeMimeTypes() {
+		$this->addEntries(
+			array(
+				array('test.doc', 'application/msword'),
+				array('test.docx', 'application/msword'),
+				array('test.xls', 'application/msexcel'),
+				array('test.xlsx', 'application/msexcel'),
+				array('test.ppt', 'application/mspowerpoint'),
+				array('test.pptx', 'application/mspowerpoint'),
+			)
+		);
+
+		$this->repair->run();
+
+		// force mimetype reload
+		$this->storage->getCache()->loadMimeTypes();
+
+		$this->checkEntries(
+			array(
+				array('test.doc', 'application/msword'),
+				array('test.docx', 'application/vnd.openxmlformats-officedocument.wordprocessingml.document'),
+				array('test.xls', 'application/msexcel'),
+				array('test.xlsx', 'application/vnd.ms-excel'),
+				array('test.ppt', 'application/mspowerpoint'),
+				array('test.pptx', 'application/vnd.ms-powerpoint'),
+			)
+		);
+	}
+}
+
+/**
+ * Dummy class to access protected members
+ */
+class DummyFileCache extends \OC\Files\Cache\Cache {
+
+	public static function clearCachedMimeTypes() {
+		self::$mimetypeIds = array();
+		self::$mimetypes = array();
+	}
+}
+