148 lines
4.2 KiB
PHP
148 lines
4.2 KiB
PHP
<?php
|
|
/**
|
|
* @copyright Copyright (c) 2016 Robin Appelman <robin@icewind.nl>
|
|
*
|
|
* @license GNU AGPL version 3 or any later version
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* 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
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
namespace Test\Files\ObjectStore;
|
|
|
|
use Icewind\Streams\Wrapper;
|
|
use Icewind\Streams\CallbackWrapper;
|
|
use OC\Files\ObjectStore\S3;
|
|
|
|
use Aws\S3\S3Client;
|
|
use Aws\S3\Exception\S3MultipartUploadException;
|
|
|
|
class MultiPartUploadS3 extends S3
|
|
{
|
|
public function writeObject($urn, $stream, string $mimetype = null)
|
|
{
|
|
$this->getConnection()->upload($this->bucket, $urn, $stream, 'private', [
|
|
'mup_threshold' => 1,
|
|
]);
|
|
}
|
|
}
|
|
|
|
class NonSeekableStream extends Wrapper
|
|
{
|
|
public static function wrap($source)
|
|
{
|
|
$context = stream_context_create([
|
|
'nonseek' => [
|
|
'source' => $source,
|
|
],
|
|
]);
|
|
return Wrapper::wrapSource($source, $context, 'nonseek', self::class);
|
|
}
|
|
|
|
public function dir_opendir($path, $options)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
public function stream_open($path, $mode, $options, &$opened_path)
|
|
{
|
|
$this->loadContext('nonseek');
|
|
return true;
|
|
}
|
|
|
|
public function stream_seek($offset, $whence = SEEK_SET)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @group PRIMARY-s3
|
|
*/
|
|
class S3Test extends ObjectStoreTest
|
|
{
|
|
protected function getInstance()
|
|
{
|
|
$config = \OC::$server->getConfig()->getSystemValue('objectstore');
|
|
if (!is_array($config) || $config['class'] !== '\\OC\\Files\\ObjectStore\\S3') {
|
|
$this->markTestSkipped('objectstore not configured for s3');
|
|
}
|
|
|
|
return new S3($config['arguments']);
|
|
}
|
|
|
|
public function testUploadNonSeekable()
|
|
{
|
|
$s3 = $this->getInstance();
|
|
|
|
$s3->writeObject('multiparttest', NonSeekableStream::wrap(fopen(__FILE__, 'r')));
|
|
|
|
$result = $s3->readObject('multiparttest');
|
|
|
|
$this->assertEquals(file_get_contents(__FILE__), stream_get_contents($result));
|
|
|
|
$s3->deleteObject('multiparttest');
|
|
}
|
|
|
|
public function testSeek()
|
|
{
|
|
$data = file_get_contents(__FILE__);
|
|
|
|
$instance = $this->getInstance();
|
|
$instance->writeObject('seek', $this->stringToStream($data));
|
|
|
|
$read = $instance->readObject('seek');
|
|
$this->assertEquals(substr($data, 0, 100), fread($read, 100));
|
|
|
|
fseek($read, 10);
|
|
$this->assertEquals(substr($data, 10, 100), fread($read, 100));
|
|
|
|
fseek($read, 100, SEEK_CUR);
|
|
$this->assertEquals(substr($data, 210, 100), fread($read, 100));
|
|
|
|
$instance->deleteObject('seek');
|
|
}
|
|
|
|
public function assertNoUpload($objectUrn)
|
|
{
|
|
$s3 = $this->getInstance();
|
|
$s3client = $s3->getConnection();
|
|
$uploads = $s3client->listMultipartUploads([
|
|
'Bucket' => $s3->getBucket(),
|
|
'Prefix' => $objectUrn,
|
|
]);
|
|
//fwrite(STDERR, print_r($uploads, TRUE));
|
|
$this->assertArrayNotHasKey('Uploads', $uploads);
|
|
}
|
|
|
|
public function testEmptyUpload()
|
|
{
|
|
//$this->expectException(S3MultipartUploadException::class);
|
|
$s3 = $this->getInstance();
|
|
|
|
// create an empty stream and check that it fits to the
|
|
// pre-conditions in writeObject for the empty case
|
|
$emptyStream = fopen("php://memory", "r");
|
|
fwrite($emptyStream, null);
|
|
|
|
$s3->writeObject('emptystream', $emptyStream);
|
|
|
|
// this method intendedly produces an S3Exception
|
|
$this->assertNoUpload('emptystream');
|
|
$this->assertTrue($s3->objectExists('emptystream'));
|
|
|
|
$s3->deleteObject('emptystream');
|
|
}
|
|
}
|