Completed
Push — master ( 51bc1d...ebadd9 )
by Ingo
02:40
created

ReadFileQueryCreatorTest::getResultsForSearch()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 6
Code Lines 4

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
c 0
b 0
f 0
dl 0
loc 6
rs 9.4285
cc 2
eloc 4
nc 2
nop 2
1
<?php
2
3
namespace SilverStripe\AssetAdmin\Tests\GraphQL;
4
5
use SilverStripe\AssetAdmin\GraphQL\ReadFileQueryCreator;
6
use SilverStripe\AssetAdmin\Tests\Controller\AssetAdminTest\FileExtension;
7
use SilverStripe\AssetAdmin\Tests\Controller\AssetAdminTest\FolderExtension;
8
use SilverStripe\Assets\File;
9
use SilverStripe\Assets\Folder;
10
use SilverStripe\Dev\SapphireTest;
11
use GraphQL\Type\Definition\ResolveInfo;
12
use SilverStripe\GraphQL\Manager;
13
use SilverStripe\Assets\Tests\Storage\AssetStoreTest\TestAssetStore;
14
15
/**
16
 * Most of the search functionality is covered in {@link FileFilterInputTypeCreatorTest}
17
 */
18
class ReadFileQueryCreatorTest extends SapphireTest
19
{
20
21
    protected $usesDatabase = true;
22
23
    public function setUp()
24
    {
25
        parent::setUp();
26
27
        TestAssetStore::activate('AssetAdminTest');
28
29
        File::add_extension(FileExtension::class);
30
        Folder::add_extension(FolderExtension::class);
31
    }
32
33
    public function tearDown()
34
    {
35
        File::remove_extension(FileExtension::class);
36
        Folder::remove_extension(FolderExtension::class);
37
38
        TestAssetStore::reset();
39
        parent::tearDown();
40
    }
41
42
    /**
43
     * @expectedException \InvalidArgumentException
44
     * @expectedExceptionMessage view access not permitted
45
     */
46
    public function testItRestrictsParentByCanView()
47
    {
48
        $folder = new Folder(['Name' => 'disallowCanView']);
49
        $folder->write();
50
51
        $this->getResultsForSearch([
52
            'filter' => ['parentId' => $folder->ID],
53
        ]);
54
    }
55
56
    public function testItFiltersResultsByCanView()
57
    {
58
        $allowedFolder = new Folder(['Name' => 'allowedFolder']);
59
        $allowedFolder->write();
60
61
        $disallowedFolder = new Folder(['Name' => 'disallowCanView']);
62
        $disallowedFolder->write();
63
64
        $allowedFile = new File(['Name' => 'allowedFile']);
65
        $allowedFile->write();
66
67
        $disallowedFile = new File(['Name' => 'disallowCanView.txt']);
68
        $disallowedFile->write();
69
70
        $list = $this->getResultsForSearch([
71
            'filter' => ['parentId' => 0],
72
        ]);
73
74
        $this->assertEquals(
75
            [
76
                $allowedFile->Name,
77
                $allowedFolder->Name,
78
            ],
79
            $list->column('Name')
80
        );
81
    }
82
83
    /**
84
     * @param array $args
85
     * @param array $context
86
     * @return \SilverStripe\ORM\DataList|\SilverStripe\ORM\Filterable
87
     */
88
    protected function getResultsForSearch($args, $context = null)
89
    {
90
        $context = $context ? $context : ['currentUser' => null];
91
        $creator = new ReadFileQueryCreator(new Manager());
92
        return $creator->resolveConnection(null, $args, $context, new ResolveInfo([]));
93
    }
94
}
95