nextcloud/tests/lib/connector/sabre/requesttest/requesttest.php

130 lines
3.4 KiB
PHP

<?php
/**
* Copyright (c) 2015 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\Connector\Sabre\RequestTest;
use OC\Connector\Sabre\Server;
use OC\Connector\Sabre\ServerFactory;
use OC\Files\Mount\MountPoint;
use OC\Files\Storage\StorageFactory;
use OC\Files\Storage\Temporary;
use OC\Files\View;
use OCP\IUser;
use Sabre\HTTP\Request;
use Test\TestCase;
use Test\Traits\MountProviderTrait;
use Test\Traits\UserTrait;
abstract class RequestTest extends TestCase {
use UserTrait;
use MountProviderTrait;
/**
* @var \OC\Connector\Sabre\ServerFactory
*/
protected $serverFactory;
protected function getStream($string) {
$stream = fopen('php://temp', 'r+');
fwrite($stream, $string);
fseek($stream, 0);
return $stream;
}
protected function setUp() {
parent::setUp();
$this->serverFactory = new ServerFactory(
\OC::$server->getConfig(),
\OC::$server->getLogger(),
\OC::$server->getDatabaseConnection(),
\OC::$server->getUserSession(),
\OC::$server->getMountManager(),
\OC::$server->getTagManager(),
\OC::$server->getEventDispatcher()
);
}
protected function setupUser($name, $password) {
$this->createUser($name, $password);
$tmpFolder = \OC::$server->getTempManager()->getTemporaryFolder();
$this->registerMount($name, '\OC\Files\Storage\Local', '/' . $name, ['datadir' => $tmpFolder]);
$this->loginAsUser($name);
return new View('/' . $name . '/files');
}
/**
* @param \OC\Files\View $view the view to run the webdav server against
* @param string $user
* @param string $password
* @param string $method
* @param string $url
* @param resource|string|null $body
* @param array|null $headers
* @return \Sabre\HTTP\Response
*/
protected function request($view, $user, $password, $method, $url, $body = null, $headers = null) {
if (is_string($body)) {
$body = $this->getStream($body);
}
$this->logout();
$exceptionPlugin = new ExceptionPlugin('webdav', null);
$server = $this->getSabreServer($view, $user, $password, $exceptionPlugin);
$request = new Request($method, $url, $headers, $body);
// since sabre catches all exceptions we need to save them and throw them from outside the sabre server
$originalServer = $_SERVER;
if (is_array($headers)) {
foreach ($headers as $header => $value) {
$_SERVER['HTTP_' . strtoupper(str_replace('-', '_', $header))] = $value;
}
}
$result = $this->makeRequest($server, $request);
foreach ($exceptionPlugin->getExceptions() as $exception) {
throw $exception;
}
$_SERVER = $originalServer;
return $result;
}
/**
* @param Server $server
* @param Request $request
* @return \Sabre\HTTP\Response
*/
protected function makeRequest(Server $server, Request $request) {
$sapi = new Sapi($request);
$server->sapi = $sapi;
$server->httpRequest = $request;
$server->exec();
return $sapi->getResponse();
}
/**
* @param View $view
* @param string $user
* @param string $password
* @param ExceptionPlugin $exceptionPlugin
* @return Server
*/
protected function getSabreServer(View $view, $user, $password, ExceptionPlugin $exceptionPlugin) {
$authBackend = new Auth($user, $password);
$server = $this->serverFactory->createServer('/', 'dummy', $authBackend, function () use ($view) {
return $view;
});
$server->addPlugin($exceptionPlugin);
return $server;
}
}