SilverStripe\GridFieldQueuedExport\Tests\GenerateCSVJobTest

Complexity

Total Complexity

8

Size/Duplication

Total Lines	127
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Importance

Changes	2
Bugs	0	Features	1

5 Methods

Rating	Name	Duplication	Size	Complexity
A	setUp()	0	8	1
A	tearDown()	0	7	2
B	testGenerateExport()	0	34	1
B	testGenerateExportOverMultipleSteps()	0	46	3
A	createJob()	0	9	1

<?php

namespace SilverStripe\GridFieldQueuedExport\Tests;

use SilverStripe\Assets\Filesystem;
use SilverStripe\Control\Director;
use SilverStripe\Core\Config\Config;
use SilverStripe\Dev\SapphireTest;
use SilverStripe\Forms\GridField\GridField;
use SilverStripe\GridfieldQueuedExport\Jobs\GenerateCSVJob;

class GenerateCSVJobTest extends SapphireTest
{

    protected static $fixture_file = 'GenerateCSVJobTest.yml';

    protected static $extra_dataobjects = [GenerateCSVJobTestRecord::class];

    protected static $extra_controllers = [GenerateCSVJobTestController::class];

    protected function setUp()
    {
        parent::setUp();

        Config::modify()->merge(Director::class, 'rules', [
            'jobtest//$Action/$ID/$OtherID' => GenerateCSVJobTestController::class
        ]);
    }

    protected $paths = [];

    protected function tearDown()
    {
        foreach ($this->paths as $path) {
            Filesystem::removeFolder(dirname($path));
        }
        parent::tearDown();
    }

    public function testGenerateExport()
    {
        // Build session
        $memberID = $this->logInWithPermission('ADMIN');
        $session = ['loggedInAs' => $memberID];

        // Build controller
        $controller = new GenerateCSVJobTestController();
        $form = $controller->Form();
        $gridfield = $form->Fields()->fieldByName('MyGridfield');

        // Build job
        $job = $this->createJob($gridfield, $session);
        $path = sprintf('%1$s/.exports/%2$s/%2$s.csv', ASSETS_PATH, $job->getSignature());
        $this->paths[] = $path; // Mark for cleanup later

        // Test that the job runs
        $this->assertFileNotExists($path);
        $job->setup();
        $job->process();
        $job->afterComplete();
        $this->assertFileExists($path);

        // Test that the output matches the expected
        $expected = [
            'Title,Content,"Publish On"',
            '"Record 1","<p>""Record 1"" Body</p>","2015-01-01 23:34:01"',
            '"Record 2","<p>""Record 2"" Body</p>","2015-01-02 23:34:01"',
            '"Record 3","<p>""Record 3"" Body</p>","2015-01-03 23:34:01"',
            '',
        ];
        $actual = file_get_contents($path);
        $this->assertEquals(implode("\r\n", $expected), $actual);
    }

    public function testGenerateExportOverMultipleSteps()
    {
        Config::modify()->set(GenerateCSVJob::class, 'chunk_size', 1);

        // Build session
        $memberID = $this->logInWithPermission('ADMIN');
        $session = ['loggedInAs' => $memberID];

        // Build controller
        $controller = new GenerateCSVJobTestController();
        $form = $controller->Form();
        /** @var GridField $gridfield */
        $gridfield = $form->Fields()->fieldByName('MyGridfield');

        // Build job
        $job = $this->createJob($gridfield, $session);
        $path = sprintf('%1$s/.exports/%2$s/%2$s.csv', ASSETS_PATH, $job->getSignature());
        $this->paths[] = $path; // Mark for cleanup later

        // Test that the job runs
        $this->assertFileNotExists($path);
        $count = 0;
        while (!$job->jobFinished()) {
            ++$count;
            if ($job->currentStep) {
                $job->prepareForRestart();
            } else {
                $job->setup();
            }
            $job->process();
        }
        $job->afterComplete();
        $this->assertFileExists($path);
        $this->assertEquals(3, $count);

        // Test that the output matches the expected
        $expected = [
            'Title,Content,"Publish On"',
            '"Record 1","<p>""Record 1"" Body</p>","2015-01-01 23:34:01"',
            '"Record 2","<p>""Record 2"" Body</p>","2015-01-02 23:34:01"',
            '"Record 3","<p>""Record 3"" Body</p>","2015-01-03 23:34:01"',
            '',
        ];
        $actual = file_get_contents($path);
        $this->assertEquals(implode("\r\n", $expected), $actual);
    }

    /**
     * Rough copy of GridFieldQueuedExportButton::startExport
     *
     * @param GridField $gridField
     * @param array $session
     * @return GenerateCSVJob
     */
    protected function createJob($gridField, $session)
    {
        $job = new GenerateCSVJob();
        $job->setGridField($gridField);
        $job->setSession($session);
        $job->setSeparator(',');
        $job->setIncludeHeader(true);
        return $job;
    }
}