SilverStripe\AssetAdmin\Tests\GraphQL\ReadFileUsageQueryCreatorTest

Complexity

Total Complexity

9

Size/Duplication

Total Lines	84
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	testUsageCount()	0	4	1
A	setUp()	0	8	1
A	testUsageCountForNonExistentFile()	0	4	1
A	tearDown()	0	7	1
A	fileUsageDataProvider()	0	8	1
A	assertUsageCount()	0	11	2
A	getResultsForSearch()	0	5	2

<?php

namespace SilverStripe\AssetAdmin\Tests\GraphQL;

use SilverStripe\AssetAdmin\GraphQL\ReadFileQueryCreator;
use SilverStripe\AssetAdmin\GraphQL\ReadFileUsageQueryCreator;
use SilverStripe\AssetAdmin\Tests\Controller\AssetAdminTest\FileExtension;
use SilverStripe\AssetAdmin\Tests\Controller\AssetAdminTest\FolderExtension;
use SilverStripe\Assets\File;
use SilverStripe\Assets\Folder;
use SilverStripe\Dev\SapphireTest;
use GraphQL\Type\Definition\ResolveInfo;
use SilverStripe\GraphQL\Manager;
use Silverstripe\Assets\Dev\TestAssetStore;

/**
 * Most of the search functionality is covered in {@link FileFilterInputTypeCreatorTest}
 */
class ReadFileUsageQueryCreatorTest extends SapphireTest
{

    protected $usesDatabase = true;

    protected static $fixture_file = '../fixtures.yml';

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

        TestAssetStore::activate('AssetAdminTest');

        File::add_extension(FileExtension::class);
        Folder::add_extension(FolderExtension::class);
    }

    protected function tearDown() : void
    {
        File::remove_extension(FileExtension::class);
        Folder::remove_extension(FolderExtension::class);

        TestAssetStore::reset();
        parent::tearDown();
    }

    public function fileUsageDataProvider()
    {
        return [
            [File::class, 'rootfile', 2, 'rootfile file is used twice'],
            [File::class, 'file2', 0, 'file2 is not used'],
            [File::class, 'disallowCanView', false, 'disallowCanView should provide file usage data'],
            [Folder::class, 'folder1', 2, 'folder1 contains files used 2 times'],
            [Folder::class, 'folder1.1', 0, 'folder1.1 does not contained any used files'],
        ];
    }

    /**
     * @dataProvider fileUsageDataProvider
     * @param string $class
     * @param string $fixture
     * @param int|false $expectedCount
     * @param $message
     */
    public function testUsageCount(string $class, string $fixture, $expectedCount, $message)
    {
        $id = $this->idFromFixture($class, $fixture);
        $this->assertUsageCount($id, $expectedCount, $message);
    }

    public function testUsageCountForNonExistentFile()
    {
        $this->expectException('InvalidArgumentException');
        $this->assertUsageCount(999, false, 'File 999 should throw an exception because it does not exists.');
    }

    /**
     * @param array $args
     * @param array $context
     * @return array
     */
    protected function getResultsForSearch($args, $context = null)
    {
        $context = $context ? $context : ['currentUser' => null];
        $creator = new ReadFileUsageQueryCreator(new Manager());
        return $creator->resolve(null, $args, $context, new ResolveInfo([]));
    }

    /**
     * Assert the file usage of the provided file ID.
     * @param string $fixture
     * @param int|false $expectedCount $expectedCount or false, if no result should be returned
     */
    private function assertUsageCount($id, $expectedCount, $message)
    {
        $actual = $this->getResultsForSearch(['IDs' => [$id]]);
        if ($expectedCount === false) {
            $this->assertEmpty($actual, $message);
        } else {
            $expected = [[
                'id' => $id,
                'inUseCount' => $expectedCount
            ]];
            $this->assertEquals($expected, $actual, $message);
        }
    }
}