From 203d7eb1d396940c6141b313de553320a003fce4 Mon Sep 17 00:00:00 2001 From: Roeland Jago Douma Date: Thu, 2 Apr 2020 14:07:40 +0200 Subject: Add AppFramework GZip middleware to gzip responses Signed-off-by: Roeland Jago Douma --- .../Middleware/CompressionMiddlewareTest.php | 163 +++++++++++++++++++++ 1 file changed, 163 insertions(+) create mode 100644 tests/lib/AppFramework/Middleware/CompressionMiddlewareTest.php (limited to 'tests/lib/AppFramework') diff --git a/tests/lib/AppFramework/Middleware/CompressionMiddlewareTest.php b/tests/lib/AppFramework/Middleware/CompressionMiddlewareTest.php new file mode 100644 index 00000000000..3eeac3ef103 --- /dev/null +++ b/tests/lib/AppFramework/Middleware/CompressionMiddlewareTest.php @@ -0,0 +1,163 @@ + + * + * @author Roeland Jago Douma + * + * @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 . + * + */ + +namespace Test\AppFramework\Middleware; + +use OC\AppFramework\Middleware\CompressionMiddleware; +use OC\AppFramework\OCS\V1Response; +use OC\AppFramework\OCS\V2Response; +use OCP\AppFramework\Controller; +use OCP\AppFramework\Http; +use OCP\AppFramework\Http\DataResponse; +use OCP\AppFramework\Http\JSONResponse; +use OCP\IRequest; + +class CompressionMiddlewareTest extends \Test\TestCase { + + /** @var IRequest */ + private $request; + /** @var Controller */ + private $controller; + /** @var CompressionMiddleware */ + private $middleWare; + + protected function setUp(): void { + parent::setUp(); + + $this->request = $this->createMock(IRequest::class); + $this->middleWare = new CompressionMiddleware( + $this->request + ); + + $this->controller = $this->createMock(Controller::class); + } + + public function testGzipOCSV1() { + $this->request->method('getHeader') + ->with('Accept-Encoding') + ->willReturn('gzip'); + + $response = $this->createMock(V1Response::class); + $response->expects($this->once()) + ->method('addHeader') + ->with('Content-Encoding', 'gzip'); + + $response->method('getStatus') + ->willReturn(Http::STATUS_OK); + + $this->middleWare->beforeController($this->controller, 'myMethod'); + $this->middleWare->afterController($this->controller,'myMethod', $response); + + $output = 'myoutput'; + $result = $this->middleWare->beforeOutput($this->controller, 'myMethod', $output); + + $this->assertSame($output, gzdecode($result)); + } + + public function testGzipOCSV2() { + $this->request->method('getHeader') + ->with('Accept-Encoding') + ->willReturn('gzip'); + + $response = $this->createMock(V2Response::class); + $response->expects($this->once()) + ->method('addHeader') + ->with('Content-Encoding', 'gzip'); + + $response->method('getStatus') + ->willReturn(Http::STATUS_OK); + + $this->middleWare->beforeController($this->controller, 'myMethod'); + $this->middleWare->afterController($this->controller,'myMethod', $response); + + $output = 'myoutput'; + $result = $this->middleWare->beforeOutput($this->controller, 'myMethod', $output); + + $this->assertSame($output, gzdecode($result)); + } + + public function testGzipJSONResponse() { + $this->request->method('getHeader') + ->with('Accept-Encoding') + ->willReturn('gzip'); + + $response = $this->createMock(JSONResponse::class); + $response->expects($this->once()) + ->method('addHeader') + ->with('Content-Encoding', 'gzip'); + + $response->method('getStatus') + ->willReturn(Http::STATUS_OK); + + $this->middleWare->beforeController($this->controller, 'myMethod'); + $this->middleWare->afterController($this->controller,'myMethod', $response); + + $output = 'myoutput'; + $result = $this->middleWare->beforeOutput($this->controller, 'myMethod', $output); + + $this->assertSame($output, gzdecode($result)); + } + + public function testNoGzipDataResponse() { + $this->request->method('getHeader') + ->with('Accept-Encoding') + ->willReturn('gzip'); + + $response = $this->createMock(DataResponse::class); + $response->expects($this->never()) + ->method('addHeader'); + + $response->method('getStatus') + ->willReturn(Http::STATUS_OK); + $this->middleWare->beforeController($this->controller, 'myMethod'); + $this->middleWare->afterController($this->controller,'myMethod', $response); + + $output = 'myoutput'; + $result = $this->middleWare->beforeOutput($this->controller, 'myMethod', $output); + + $this->assertSame($output, $result); + } + + public function testNoGzipNo200() { + $this->request->method('getHeader') + ->with('Accept-Encoding') + ->willReturn('gzip'); + + $response = $this->createMock(JSONResponse::class); + $response->expects($this->never()) + ->method('addHeader'); + + $response->method('getStatus') + ->willReturn(Http::STATUS_NOT_FOUND); + + $this->middleWare->beforeController($this->controller, 'myMethod'); + $this->middleWare->afterController($this->controller,'myMethod', $response); + + $output = 'myoutput'; + $result = $this->middleWare->beforeOutput($this->controller, 'myMethod', $output); + + $this->assertSame($output, $result); + } +} -- cgit v1.2.3