nextcloud/apps/files_encryption/tests/stream.php

<?php
/**
 * Copyright (c) 2012 Robin Appelman <icewind@owncloud.com>
 * This file is licensed under the Affero General Public License version 3 or
 * later.
 * See the COPYING-README file.
 */

class Test_CryptStream extends UnitTestCase {
	private $tmpFiles=array();

	function testStream() {
		$stream=$this->getStream('test1', 'w', strlen('foobar'));
		fwrite($stream, 'foobar');
		fclose($stream);

		$stream=$this->getStream('test1', 'r', strlen('foobar'));
		$data=fread($stream, 6);
		fclose($stream);
		$this->assertEqual('foobar', $data);

		$file=OC::$SERVERROOT.'/3rdparty/MDB2.php';
		$source=fopen($file, 'r');
		$target=$this->getStream('test2', 'w', 0);
		OCP\Files::streamCopy($source, $target);
		fclose($target);
		fclose($source);

		$stream=$this->getStream('test2', 'r', filesize($file));
		$data=stream_get_contents($stream);
		$original=file_get_contents($file);
		$this->assertEqual(strlen($original), strlen($data));
		$this->assertEqual($original, $data);
	}

	/**
	 * get a cryptstream to a temporary file
	 * @param string $id
	 * @param string $mode
	 * @param int size
	 * @return resource
	 */
	function getStream($id, $mode, $size) {
		if($id==='') {
			$id=uniqid();
		}
		if(!isset($this->tmpFiles[$id])) {
			$file=OCP\Files::tmpFile();
			$this->tmpFiles[$id]=$file;
		}else{
			$file=$this->tmpFiles[$id];
		}
		$stream=fopen($file, $mode);
		OC_CryptStream::$sourceStreams[$id]=array('path'=>'dummy'.$id, 'stream'=>$stream, 'size'=>$size);
		return fopen('crypt://streams/'.$id, $mode);
	}

	function testBinary() {
		$file=__DIR__.'/binary';
		$source=file_get_contents($file);

		$stream=$this->getStream('test', 'w', strlen($source));
		fwrite($stream, $source);
		fclose($stream);

		$stream=$this->getStream('test', 'r', strlen($source));
		$data=stream_get_contents($stream);
		fclose($stream);
		$this->assertEqual(strlen($data), strlen($source));
		$this->assertEqual($source, $data);

		$file=__DIR__.'/zeros';
		$source=file_get_contents($file);

		$stream=$this->getStream('test2', 'w', strlen($source));
		fwrite($stream, $source);
		fclose($stream);

		$stream=$this->getStream('test2', 'r', strlen($source));
		$data=stream_get_contents($stream);
		fclose($stream);
		$this->assertEqual(strlen($data), strlen($source));
		$this->assertEqual($source, $data);
	}
}