Rework the tests

Author: theofidry

bin/console

@@ -8,8 +8,10 @@ require __DIR__.'/../vendor/autoload.php';
 use Symfony\Bundle\FrameworkBundle\Console\Application;
 use Webmozarts\Console\Parallelization\ImportMoviesCommand;
 use Webmozarts\Console\Parallelization\Kernel;
+use Webmozarts\Console\Parallelization\NoSubProcessCommand;
 
 $application = new Application(new Kernel());
 $application->add(new ImportMoviesCommand());
+$application->add(new NoSubProcessCommand());
 
 $application->run();

src/Parallelization.php

@@ -31,6 +31,7 @@
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Helper\ProgressBar;
 use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Logger\ConsoleLogger;
 use Symfony\Component\Console\Output\OutputInterface;
 use Symfony\Component\Console\Terminal;
 use Symfony\Component\DependencyInjection\ContainerInterface;
@@ -348,7 +349,8 @@ function () use ($input) {
                 $this->getEnvironmentVariables($this->getContainer()),
                 $numberOfProcesses,
                 $segmentSize,
-                $this->getContainer()->get('logger', ContainerInterface::NULL_ON_INVALID_REFERENCE),
+                // TODO: offer a way to create the process launcher in a different manner
+                new ConsoleLogger($output),
                 function (string $type, string $buffer) use ($progressBar, $output, $terminalWidth) {
                     $this->processChildOutput($buffer, $progressBar, $output, $terminalWidth);
                 }

tests/ImportMoviesCommand.php

@@ -13,6 +13,9 @@
 
 namespace Webmozarts\Console\Parallelization;
 
+use function file_get_contents;
+use function json_decode;
+use const JSON_THROW_ON_ERROR;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
 
@@ -22,9 +25,18 @@ final class ImportMoviesCommand extends ContainerAwareCommand
 
     protected static $defaultName = 'import:movies';
 
-    public function __construct(?string $name = null)
+    private TestLogger $logger;
+
+    /**
+     * @var array<string, string>
+     */
+    private array $batchMovies;
+
+    public function __construct()
     {
-        parent::__construct($name);
+        parent::__construct(self::$defaultName);
+
+        $this->logger = new TestLogger();
     }
 
     protected function configure(): void
@@ -50,27 +62,66 @@ protected function getSegmentSize(): int
 
     protected function runBeforeFirstCommand(InputInterface $input, OutputInterface $output): void
     {
+        $this->logger->recordFirstCommand();
     }
 
     protected function runBeforeBatch(
         InputInterface $input,
         OutputInterface $output,
-        array $items
+        array $movieFileNames
     ): void {
+        $this->logger->recordBeforeBatch();
+
+        $this->batchMovies = self::fetchMovieTitles($movieFileNames);
     }
 
-    protected function runSingleCommand(string $item, InputInterface $input, OutputInterface $output): void
+    protected function runSingleCommand(string $movieFileName, InputInterface $input, OutputInterface $output): void
     {
-        // insert into the database
+        $this->logger->recordSingleCommand(
+            $movieFileName,
+            $this->batchMovies[$movieFileName],
+        );
     }
 
     protected function runAfterBatch(InputInterface $input, OutputInterface $output, array $items): void
     {
-        // flush the database and clear the entity manager
+        $this->logger->recordAfterBatch();
+
+        $this->batchMovies = null;
+    }
+
+    protected function runAfterLastCommand(InputInterface $input, OutputInterface $output): void
+    {
+        $this->logger->recordLastCommand();
     }
 
     protected function getItemName(int $count): string
     {
         return 1 === $count ? 'movie' : 'movies';
     }
+
+    /**
+     * @param list<string> $movieFileNames
+     *
+     * @return array<string, string>
+     */
+    private static function fetchMovieTitles(array $movieFileNames): array
+    {
+        $movies = [];
+
+        foreach ($movieFileNames as $movieFileName) {
+            $moviePath = __DIR__.'/movies/'.$movieFileName;
+
+            $decodedContent = json_decode(
+                file_get_contents($moviePath),
+                null,
+                512,
+                JSON_THROW_ON_ERROR,
+            );
+
+            $movies[$movieFileName] = $decodedContent->title;
+        }
+
+        return $movies;
+    }
 }

tests/NoSubProcessCommand.php

@@ -0,0 +1,65 @@
+<?php
+
+/*
+ * This file is part of the Webmozarts Console Parallelization package.
+ *
+ * (c) Webmozarts GmbH <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+declare(strict_types=1);
+
+namespace Webmozarts\Console\Parallelization;
+
+use DomainException;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+
+final class NoSubProcessCommand extends ContainerAwareCommand
+{
+    use Parallelization;
+
+    protected static $defaultName = 'test:no-subprocess';
+
+    private bool $mainProcess = false;
+
+    protected function configure(): void
+    {
+        self::configureParallelization($this);
+    }
+
+    protected function fetchItems(InputInterface $input): array
+    {
+        return [
+            'item1',
+            'item2',
+            'item3',
+            'item4',
+            'item5',
+        ];
+    }
+
+    protected function getSegmentSize(): int
+    {
+        return 2;
+    }
+
+    protected function runBeforeFirstCommand(InputInterface $input, OutputInterface $output): void
+    {
+        $this->mainProcess = true;
+    }
+
+    protected function runSingleCommand(string $item, InputInterface $input, OutputInterface $output): void
+    {
+        if (!$this->mainProcess) {
+            throw new DomainException('Expected to be executed within the main process.');
+        }
+    }
+
+    protected function getItemName(int $count): string
+    {
+        return 0 === $count ? 'item' : 'items';
+    }
+}