1 files changed, 57 insertions, 0 deletions
diff --git a/tests/lib/Repair/RepairSharePropagationTest.php b/tests/lib/Repair/RepairSharePropagationTest.php
new file mode 100644
index 00000000000..07db3c84f28
--- /dev/null
+++ b/tests/lib/Repair/RepairSharePropagationTest.php
@@ -0,0 +1,57 @@
+<?php
+/**
+ * Copyright (c) 2016 Robin Appelman <icewind@owncloud.com>
+ * This file is licensed under the Affero General Public License version 3 or
+ * later.
+ * See the COPYING-README file.
+ */
+
+namespace Test\Repair;
+
+use OC\Repair\SharePropagation;
+use OCP\Migration\IOutput;
+
+class RepairSharePropagationTest extends \Test\TestCase {
+	public function keyProvider() {
+		return [
+			[['1', '2'], ['1', '2']],
+			[['1', '2', 'foo'], ['1', '2']],
+			[['foo'], []],
+		];
+	}
+
+	/**
+	 * @dataProvider keyProvider
+	 * @param array $startKeys
+	 * @param array $expectedRemovedKeys
+	 */
+	public function testRemovePropagationEntries(array $startKeys, array $expectedRemovedKeys) {
+		/** @var \PHPUnit_Framework_MockObject_MockObject|\OCP\IConfig $config */
+		$config = $this->getMock('\OCP\IConfig');
+		$config->expects($this->once())
+			->method('getAppKeys')
+			->with('files_sharing')
+			->will($this->returnValue($startKeys));
+
+		$removedKeys = [];
+
+		$config->expects($this->any())
+			->method('deleteAppValue')
+			->will($this->returnCallback(function ($app, $key) use (&$removedKeys) {
+				$removedKeys[] = $key;
+			}));
+
+		/** @var IOutput | \PHPUnit_Framework_MockObject_MockObject $outputMock */
+		$outputMock = $this->getMockBuilder('\OCP\Migration\IOutput')
+			->disableOriginalConstructor()
+			->getMock();
+
+		$step = new SharePropagation($config);
+		$step->run($outputMock);
+
+		sort($expectedRemovedKeys);
+		sort($removedKeys);
+
+		$this->assertEquals($expectedRemovedKeys, $removedKeys);
+	}
+}