-
-
Notifications
You must be signed in to change notification settings - Fork 88
/
DoctrineODMQuerySourceIteratorTest.php
92 lines (72 loc) · 2.61 KB
/
DoctrineODMQuerySourceIteratorTest.php
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
<?php
declare(strict_types=1);
/*
* This file is part of the Sonata Project package.
*
* (c) Thomas Rabaix <thomas.rabaix@sonata-project.org>
*
* For the full copyright and license information, please view the LICENSE
* file that was distributed with this source code.
*/
namespace Sonata\Exporter\Tests\Source;
use Doctrine\ODM\MongoDB\Configuration;
use Doctrine\ODM\MongoDB\DocumentManager;
use PHPUnit\Framework\TestCase;
use Sonata\Exporter\Source\DoctrineODMQuerySourceIterator;
use Sonata\Exporter\Tests\Source\Fixtures\Document;
final class DoctrineODMQuerySourceIteratorTest extends TestCase
{
private DocumentManager $dm;
protected function setUp(): void
{
if (!\extension_loaded('mongodb')) {
static::markTestSkipped('The mongodb extension is not available.');
}
$this->dm = DocumentManager::create(null, $this->createConfiguration());
$documentA = new Document();
$documentB = new Document();
$documentC = new Document();
$this->dm->persist($documentA);
$this->dm->persist($documentB);
$this->dm->persist($documentC);
$this->dm->flush();
}
protected function tearDown(): void
{
$this->dm->createQueryBuilder(Document::class)
->remove()
->getQuery()
->execute();
}
public function testHandler(): void
{
$query = $this->dm->createQueryBuilder(Document::class)->getQuery();
$iterator = new DoctrineODMQuerySourceIterator($query, ['id']);
static::assertCount(3, iterator_to_array($iterator));
}
/**
* @psalm-suppress InternalMethod
*/
public function testEntityManagerClear(): void
{
$query = $this->dm->createQueryBuilder(Document::class)->getQuery();
$batchSize = 2;
$iterator = new DoctrineODMQuerySourceIterator($query, ['id'], 'r', $batchSize);
foreach ($iterator as $i => $item) {
static::assertSame(0 === $i % $batchSize ? 0 : $i, $this->dm->getUnitOfWork()->size());
}
}
private function createConfiguration(): Configuration
{
$config = new Configuration();
$directory = sys_get_temp_dir().'/mongodb';
$config->setProxyDir($directory);
$config->setProxyNamespace('Proxies');
$config->setHydratorDir($directory);
$config->setHydratorNamespace('Hydrators');
$config->setPersistentCollectionDir($directory);
$config->setPersistentCollectionNamespace('PersistentCollections');
$config->setMetadataDriverImpl($config->newDefaultAnnotationDriver());
return $config;
}
}