1 files changed, 220 insertions, 0 deletions
diff --git a/apps/dav/tests/unit/Files/BundledFileTest.php b/apps/dav/tests/unit/Files/BundledFileTest.php
new file mode 100644
index 00000000000..e1a65459b60
--- /dev/null
+++ b/apps/dav/tests/unit/Files/BundledFileTest.php
@@ -0,0 +1,220 @@
+<?php
+/**
+ * @author Piotr Mrowczynski <Piotr.Mrowczynski@owncloud.com>
+ * @author Louis Chemineau <louis@chmn.me>
+ *
+ * @copyright Copyright (c) 2016, ownCloud GmbH.
+ * @license AGPL-3.0
+ *
+ * This code is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License, version 3,
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License, version 3,
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>
+ *
+ */
+
+namespace OCA\DAV\Files;
+
+use OCP\Lock\ILockingProvider;
+
+/**
+ * Class File
+ *
+ * @group DB
+ *
+ * @package OCA\DAV\Tests\unit\Connector\Sabre
+ */
+class BundledFileTest extends \Test\TestCase {
+
+	/**
+	 * @var string
+	 */
+	private $user;
+
+	/* BASICS */
+
+	public function setUp() {
+		parent::setUp();
+
+		\OC_Hook::clear();
+
+		$this->user = $this->getUniqueID('user_');
+		$userManager = \OC::$server->getUserManager();
+		$userManager->createUser($this->user, 'pass');
+
+		$this->loginAsUser($this->user);
+	}
+
+	/* TESTS */
+
+	/**
+	 * Test basic successful bundled file PutFile
+	 */
+	public function testPutFile() {
+		$bodyContent = 'blabla';
+		$headers['oc-total-length'] = 6;
+		$headers['oc-path'] = '/foo.txt';
+		$headers['oc-mtime'] = '1473336321';
+		$headers['response'] = null;
+
+		//this part will have some arbitrary, correct headers
+		$bodyFull = "$bodyContent\r\n--boundary--";
+		$multipartContentsParser = $this->fillMultipartContentsParserStreamWithBody($bodyFull);
+
+		$this->doPutFIle($headers, $multipartContentsParser);
+	}
+
+	/**
+	 * Test basic successful bundled file PutFile
+	 *
+	 * @expectedException \Sabre\DAV\Exception\Forbidden
+	 * @expectedExceptionMessage File requires oc-total-length header to be read
+	 */
+	public function testPutFileNoLength() {
+		$bodyContent = 'blabla';
+		$headers['oc-path'] = '/foo.txt';
+		$headers['oc-mtime'] = '1473336321';
+		$headers['response'] = null;
+
+		//this part will have some arbitrary, correct headers
+		$bodyFull = "$bodyContent\r\n--boundary--";
+		$multipartContentsParser = $this->fillMultipartContentsParserStreamWithBody($bodyFull);
+
+		$this->doPutFIle($headers, $multipartContentsParser);
+	}
+
+	/**
+	 * Test putting a single file
+	 *
+	 * @expectedException \Sabre\DAV\Exception\Forbidden
+	 * @expectedExceptionMessage PUT method not supported for bundling
+	 */
+	public function testThrowIfPut() {
+		$fileContents = $this->getStream('test data');
+		$this->doPut('/foo.txt', $fileContents);
+	}
+
+	/* UTILITIES */
+
+	private function getMockStorage() {
+		$storage = $this->getMockBuilder('\OCP\Files\Storage')
+			->getMock();
+		$storage->expects($this->any())
+			->method('getId')
+			->will($this->returnValue('home::someuser'));
+		return $storage;
+	}
+
+	public function tearDown() {
+		$userManager = \OC::$server->getUserManager();
+		$userManager->get($this->user)->delete();
+		unset($_SERVER['HTTP_OC_CHUNKED']);
+
+		parent::tearDown();
+	}
+
+	/**
+	 * @param string $string
+	 */
+	private function getStream($string) {
+		$stream = fopen('php://temp', 'r+');
+		fwrite($stream, $string);
+		fseek($stream, 0);
+		return $stream;
+	}
+
+	/**
+	 * Do basic put for single bundled file
+	 */
+	private function doPutFIle($fileMetadata, $contentHandler, $view = null, $viewRoot = null) {
+		$path = $fileMetadata['oc-path'];
+
+		if(is_null($view)){
+			$view = \OC\Files\Filesystem::getView();
+		}
+		if (!is_null($viewRoot)) {
+			$view = new \OC\Files\View($viewRoot);
+		} else {
+			$viewRoot = '/' . $this->user . '/files';
+		}
+
+		$info = new \OC\Files\FileInfo(
+			$viewRoot . '/' . ltrim($path, '/'),
+			$this->getMockStorage(),
+			null,
+			['permissions' => \OCP\Constants::PERMISSION_ALL],
+			null
+		);
+
+		$file = new BundledFile($view, $info, $contentHandler);
+
+		// beforeMethod locks
+		$view->lockFile($path, ILockingProvider::LOCK_SHARED);
+
+		$result = $file->putFile($fileMetadata);
+
+		// afterMethod unlocks
+		$view->unlockFile($path, ILockingProvider::LOCK_SHARED);
+
+		return $result;
+	}
+
+	private function fillMultipartContentsParserStreamWithBody($bodyString){
+		$bodyStream = fopen('php://temp', 'r+');
+		fwrite($bodyStream, $bodyString);
+		rewind($bodyStream);
+		$request = $this->getMockBuilder('Sabre\HTTP\RequestInterface')
+			->disableOriginalConstructor()
+			->getMock();
+		$request->expects($this->any())
+			->method('getBody')
+			->willReturn($bodyStream);
+
+		$mcp = new \OCA\DAV\Files\MultipartContentsParser($request);
+		return $mcp;
+	}
+
+	/**
+	 * Simulate putting a file to the given path.
+	 *
+	 * @param string $path path to put the file into
+	 * @param string $viewRoot root to use for the view
+	 *
+	 * @return null|string of the PUT operaiton which is usually the etag
+	 */
+	private function doPut($path, $fileContents, $viewRoot = null) {
+		$view = \OC\Files\Filesystem::getView();
+		if (!is_null($viewRoot)) {
+			$view = new \OC\Files\View($viewRoot);
+		} else {
+			$viewRoot = '/' . $this->user . '/files';
+		}
+
+		$info = new \OC\Files\FileInfo(
+			$viewRoot . '/' . ltrim($path, '/'),
+			$this->getMockStorage(),
+			null,
+			['permissions' => \OCP\Constants::PERMISSION_ALL],
+			null
+		);
+
+		$file = new BundledFile($view, $info, null);
+
+		// beforeMethod locks
+		$view->lockFile($path, ILockingProvider::LOCK_SHARED);
+
+		$result = $file->put($fileContents);
+
+		// afterMethod unlocks
+		$view->unlockFile($path, ILockingProvider::LOCK_SHARED);
+
+		return $result;
+	}
+}