CreatePdfFileTest.php 1.94 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 *
 */
declare(strict_types=1);

namespace Magento\Framework\App\Filesystem;

use Magento\Framework\App\Filesystem\DirectoryList;
use Magento\Framework\App\Response\Http\FileFactory;
use Magento\Framework\Filesystem;
use Magento\TestFramework\Helper\Bootstrap;
use Zend\Http\Header\ContentType;

/**
 * Class CreatePdfFileTest
 *
 * Integration test for testing a file creation from string
 */
class CreatePdfFileTest extends \PHPUnit\Framework\TestCase
{
    /**
     * @throws \Exception
     */
    public function testGenerateFileFromString()
    {
        $objectManager = Bootstrap::getObjectManager();
        /** @var FileFactory $fileFactory */
        $fileFactory =  $objectManager->get(FileFactory::class);
        /** @var Filesystem $filesystem */
        $filesystem = $objectManager->get(Filesystem::class);
        $filename = 'test.pdf';
        $contentType = 'application/pdf';
        $fileContent = ['type' => 'string', 'value' => ''];
        $response = $fileFactory->create($filename, $fileContent, DirectoryList::VAR_DIR, $contentType);
        /** @var ContentType $contentTypeHeader */
        $contentTypeHeader = $response->getHeader('Content-type');

        /* Check the system returns the correct type */
        self::assertEquals("Content-Type: $contentType", $contentTypeHeader->toString());

        $varDirectory = $filesystem->getDirectoryRead(DirectoryList::VAR_DIR);
        $varDirectory->isFile($filename);

        /* Check the file is generated */
        self::assertTrue($varDirectory->isFile($filename));

        /* Check the file is removed after generation if the corresponding option is set */
        $fileContent = ['type' => 'string', 'value' => '', 'rm' => true];
        $fileFactory->create($filename, $fileContent, DirectoryList::VAR_DIR, $contentType);

        self::assertFalse($varDirectory->isFile($filename));
    }
}