style: fix DoctrineORMLoaderTest

bpolaszek · Nov 14, 2023 · 881372b · 881372b
1 parent 2493830
commit 881372b
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/tests/Unit/Loader/DoctrineORMLoaderTest.php b/tests/Unit/Loader/DoctrineORMLoaderTest.php
@@ -4,8 +4,6 @@
 
 namespace BenTools\ETL\Tests\Unit\Loader;
 
-use BenTools\ETL\EtlConfiguration;
-use BenTools\ETL\EtlExecutor;
 use BenTools\ETL\EtlState;
 use BenTools\ETL\Exception\LoadException;
 use BenTools\ETL\Loader\DoctrineORMLoader;
@@ -15,21 +13,18 @@
 use Mockery;
 use stdClass;
 
+use function BenTools\ETL\loadInto;
+
 it('works', function () {
     $registry = Mockery::mock(ManagerRegistry::class);
     $manager = Mockery::mock(ObjectManager::class);
     $registry->shouldReceive('getManagerForClass')->andReturn($manager);
     $manager->shouldReceive('persist')->twice();
     $manager->shouldReceive('flush')->once();
 
-    $executor = (new EtlExecutor(options: new EtlConfiguration(flushEvery: 10)))
-        ->transformWith(function (array $book) {
-            yield new Book($book['id'], $book['name']);
-        })
-        ->loadInto(new DoctrineORMLoader($registry));
-    $executor->process([
-        ['id' => 1, 'name' => 'Holy Bible'],
-        ['id' => 2, 'name' => 'Fifty Shades of Grey'],
+    loadInto(new DoctrineORMLoader($registry))->process([
+        new Book(id: 1, name: 'Holy Bible'),
+        new Book(id: 2, name: 'Fifty Shades of Grey'),
     ]);
 });