nextcloud_server/apps/dav/tests/unit/Upload/AssemblyStreamTest.php

<?php

/**
 * SPDX-FileCopyrightText: 2016-2024 Nextcloud GmbH and Nextcloud contributors
 * SPDX-FileCopyrightText: 2016 ownCloud, Inc.
 * SPDX-License-Identifier: AGPL-3.0-only
 */
namespace OCA\DAV\Tests\unit\Upload;

use OCA\DAV\Upload\AssemblyStream;
use Sabre\DAV\File;

class AssemblyStreamTest extends \Test\TestCase {

	/**
	 * @dataProvider providesNodes()
	 */
	public function testGetContents($expected, $nodes): void {
		$stream = AssemblyStream::wrap($nodes);
		$content = stream_get_contents($stream);

		$this->assertEquals($expected, $content);
	}

	/**
	 * @dataProvider providesNodes()
	 */
	public function testGetContentsFread($expected, $nodes, $chunkLength = 3): void {
		$stream = AssemblyStream::wrap($nodes);

		$content = '';
		while (!feof($stream)) {
			$chunk = fread($stream, $chunkLength);
			$content .= $chunk;
			if ($chunkLength !== 3) {
				$this->assertEquals($chunkLength, strlen($chunk));
			}
		}

		$this->assertEquals($expected, $content);
	}

	/**
	 * @dataProvider providesNodes()
	 */
	public function testSeek($expected, $nodes): void {
		$stream = AssemblyStream::wrap($nodes);

		$offset = floor(strlen($expected) * 0.6);
		if (fseek($stream, $offset) === -1) {
			$this->fail('fseek failed');
		}

		$content = stream_get_contents($stream);
		$this->assertEquals(substr($expected, $offset), $content);
	}

	public function providesNodes() {
		$data8k = $this->makeData(8192);
		$dataLess8k = $this->makeData(8191);

		$tonofnodes = [];
		$tonofdata = '';
		for ($i = 0; $i < 101; $i++) {
			$thisdata = random_int(0, 100); // variable length and content
			$tonofdata .= $thisdata;
			$tonofnodes[] = $this->buildNode((string)$i, (string)$thisdata);
		}

		return[
			'one node zero bytes' => [
				'', [
					$this->buildNode('0', '')
				]],
			'one node only' => [
				'1234567890', [
					$this->buildNode('0', '1234567890')
				]],
			'one node buffer boundary' => [
				$data8k, [
					$this->buildNode('0', $data8k)
				]],
			'two nodes' => [
				'1234567890', [
					$this->buildNode('1', '67890'),
					$this->buildNode('0', '12345')
				]],
			'two nodes end on buffer boundary' => [
				$data8k . $data8k, [
					$this->buildNode('1', $data8k),
					$this->buildNode('0', $data8k)
				]],
			'two nodes with one on buffer boundary' => [
				$data8k . $dataLess8k, [
					$this->buildNode('1', $dataLess8k),
					$this->buildNode('0', $data8k)
				]],
			'two nodes on buffer boundary plus one byte' => [
				$data8k . 'X' . $data8k, [
					$this->buildNode('1', $data8k),
					$this->buildNode('0', $data8k . 'X')
				]],
			'two nodes on buffer boundary plus one byte at the end' => [
				$data8k . $data8k . 'X', [
					$this->buildNode('1', $data8k . 'X'),
					$this->buildNode('0', $data8k)
				]],
			'a ton of nodes' => [
				$tonofdata, $tonofnodes
			],
			'one read over multiple nodes' => [
				'1234567890', [
					$this->buildNode('0', '1234'),
					$this->buildNode('1', '5678'),
					$this->buildNode('2', '90'),
				], 10],
			'two reads over multiple nodes' => [
				'1234567890', [
					$this->buildNode('0', '1234'),
					$this->buildNode('1', '5678'),
					$this->buildNode('2', '90'),
				], 5],
		];
	}

	private function makeData($count) {
		$data = '';
		$base = '1234567890';
		$j = 0;
		for ($i = 0; $i < $count; $i++) {
			$data .= $base[$j];
			$j++;
			if (!isset($base[$j])) {
				$j = 0;
			}
		}
		return $data;
	}

	private function buildNode($name, $data) {
		$node = $this->getMockBuilder(File::class)
			->setMethods(['getName', 'get', 'getSize'])
			->getMockForAbstractClass();

		$node->expects($this->any())
			->method('getName')
			->willReturn($name);

		$node->expects($this->any())
			->method('get')
			->willReturn($data);

		$node->expects($this->any())
			->method('getSize')
			->willReturn(strlen($data));

		return $node;
	}
}