123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273 |
- <?php
- namespace OCA\DAV\Tests\unit\DAV;
- use OCA\DAV\BulkUpload\MultipartRequestParser;
- use Psr\Log\LoggerInterface;
- use Test\TestCase;
- class MultipartRequestParserTest extends TestCase {
- protected LoggerInterface $logger;
- protected function setUp(): void {
- $this->logger = $this->createMock(LoggerInterface::class);
- }
- private function getValidBodyObject() {
- return [
- [
- 'headers' => [
- 'Content-Length' => 7,
- 'X-File-MD5' => '4f2377b4d911f7ec46325fe603c3af03',
- 'X-File-Path' => '/coucou.txt'
- ],
- 'content' => "Coucou\n"
- ]
- ];
- }
- private function getMultipartParser(array $parts, array $headers = [], string $boundary = 'boundary_azertyuiop'): MultipartRequestParser {
- $request = $this->getMockBuilder('Sabre\HTTP\RequestInterface')
- ->disableOriginalConstructor()
- ->getMock();
- $headers = array_merge(['Content-Type' => 'multipart/related; boundary=' . $boundary], $headers);
- $request->expects($this->any())
- ->method('getHeader')
- ->willReturnCallback(function (string $key) use (&$headers) {
- return $headers[$key];
- });
- $body = '';
- foreach ($parts as $part) {
- $body .= '--' . $boundary . "\r\n";
- foreach ($part['headers'] as $headerKey => $headerPart) {
- $body .= $headerKey . ': ' . $headerPart . "\r\n";
- }
- $body .= "\r\n";
- $body .= $part['content'] . "\r\n";
- }
- $body .= '--' . $boundary . '--';
- $stream = fopen('php://temp', 'r+');
- fwrite($stream, $body);
- rewind($stream);
- $request->expects($this->any())
- ->method('getBody')
- ->willReturn($stream);
- return new MultipartRequestParser($request, $this->logger);
- }
-
- public function testBodyTypeValidation(): void {
- $bodyStream = 'I am not a stream, but pretend to be';
- $request = $this->getMockBuilder('Sabre\HTTP\RequestInterface')
- ->disableOriginalConstructor()
- ->getMock();
- $request->expects($this->any())
- ->method('getBody')
- ->willReturn($bodyStream);
- $this->expectExceptionMessage('Body should be of type resource');
- new MultipartRequestParser($request, $this->logger);
- }
-
- public function testValidRequest(): void {
- $multipartParser = $this->getMultipartParser(
- $this->getValidBodyObject()
- );
- [$headers, $content] = $multipartParser->parseNextPart();
- $this->assertSame((int)$headers['content-length'], 7, 'Content-Length header should be the same as provided.');
- $this->assertSame($headers['x-file-md5'], '4f2377b4d911f7ec46325fe603c3af03', 'X-File-MD5 header should be the same as provided.');
- $this->assertSame($headers['x-file-path'], '/coucou.txt', 'X-File-Path header should be the same as provided.');
- $this->assertSame($content, "Coucou\n", 'Content should be the same');
- }
-
- public function testInvalidMd5Hash(): void {
- $bodyObject = $this->getValidBodyObject();
- $bodyObject['0']['headers']['X-File-MD5'] = 'f2377b4d911f7ec46325fe603c3af03';
- $multipartParser = $this->getMultipartParser(
- $bodyObject
- );
- $this->expectExceptionMessage('Computed md5 hash is incorrect.');
- $multipartParser->parseNextPart();
- }
-
- public function testNullMd5Hash(): void {
- $bodyObject = $this->getValidBodyObject();
- unset($bodyObject['0']['headers']['X-File-MD5']);
- $multipartParser = $this->getMultipartParser(
- $bodyObject
- );
- $this->expectExceptionMessage('The X-File-MD5 header must not be null.');
- $multipartParser->parseNextPart();
- }
-
- public function testNullContentLength(): void {
- $bodyObject = $this->getValidBodyObject();
- unset($bodyObject['0']['headers']['Content-Length']);
- $multipartParser = $this->getMultipartParser(
- $bodyObject
- );
- $this->expectExceptionMessage('The Content-Length header must not be null.');
- $multipartParser->parseNextPart();
- }
-
- public function testLowerContentLength(): void {
- $bodyObject = $this->getValidBodyObject();
- $bodyObject['0']['headers']['Content-Length'] = 6;
- $multipartParser = $this->getMultipartParser(
- $bodyObject
- );
- $this->expectExceptionMessage('Computed md5 hash is incorrect.');
- $multipartParser->parseNextPart();
- }
-
- public function testHigherContentLength(): void {
- $bodyObject = $this->getValidBodyObject();
- $bodyObject['0']['headers']['Content-Length'] = 8;
- $multipartParser = $this->getMultipartParser(
- $bodyObject
- );
- $this->expectExceptionMessage('Computed md5 hash is incorrect.');
- $multipartParser->parseNextPart();
- }
-
- public function testWrongBoundary(): void {
- $bodyObject = $this->getValidBodyObject();
- $multipartParser = $this->getMultipartParser(
- $bodyObject,
- ['Content-Type' => 'multipart/related; boundary=boundary_poiuytreza']
- );
- $this->expectExceptionMessage('Boundary not found where it should be.');
- $multipartParser->parseNextPart();
- }
-
- public function testNoBoundaryInHeader(): void {
- $bodyObject = $this->getValidBodyObject();
- $this->expectExceptionMessage('Error while parsing boundary in Content-Type header.');
- $this->getMultipartParser(
- $bodyObject,
- ['Content-Type' => 'multipart/related']
- );
- }
-
- public function testNoBoundaryInBody(): void {
- $bodyObject = $this->getValidBodyObject();
- $multipartParser = $this->getMultipartParser(
- $bodyObject,
- ['Content-Type' => 'multipart/related; boundary=boundary_azertyuiop'],
- ''
- );
- $this->expectExceptionMessage('Boundary not found where it should be.');
- $multipartParser->parseNextPart();
- }
-
- public function testBoundaryWithQuotes(): void {
- $bodyObject = $this->getValidBodyObject();
- $multipartParser = $this->getMultipartParser(
- $bodyObject,
- ['Content-Type' => 'multipart/related; boundary="boundary_azertyuiop"'],
- );
- $multipartParser->parseNextPart();
-
- $this->assertTrue(true);
- }
-
- public function testWrongContentType(): void {
- $bodyObject = $this->getValidBodyObject();
- $this->expectExceptionMessage('Content-Type must be multipart/related');
- $this->getMultipartParser(
- $bodyObject,
- ['Content-Type' => 'multipart/form-data; boundary="boundary_azertyuiop"'],
- );
- }
-
- public function testWrongKeyInContentType(): void {
- $bodyObject = $this->getValidBodyObject();
- $this->expectExceptionMessage('Boundary is invalid');
- $this->getMultipartParser(
- $bodyObject,
- ['Content-Type' => 'multipart/related; wrongkey="boundary_azertyuiop"'],
- );
- }
-
- public function testNullContentType(): void {
- $bodyObject = $this->getValidBodyObject();
- $this->expectExceptionMessage('Content-Type can not be null');
- $this->getMultipartParser(
- $bodyObject,
- ['Content-Type' => null],
- );
- }
- }
|