Implemented PoolOptimizer

tests/Composer/Test/DependencyResolver/PoolOptimizerTest.php

@@ -0,0 +1,197 @@
+<?php
+
+/*
+ * This file is part of Composer.
+ *
+ * (c) Nils Adermann <naderman@naderman.de>
+ *     Jordi Boggiano <j.boggiano@seld.be>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Composer\Test\DependencyResolver;
+
+use Composer\DependencyResolver\DefaultPolicy;
+use Composer\DependencyResolver\Pool;
+use Composer\DependencyResolver\PoolOptimizer;
+use Composer\DependencyResolver\Request;
+use Composer\Json\JsonFile;
+use Composer\Package\AliasPackage;
+use Composer\Package\BasePackage;
+use Composer\Package\Loader\ArrayLoader;
+use Composer\Package\Version\VersionParser;
+use Composer\Repository\LockArrayRepository;
+use Composer\Test\TestCase;
+
+class PoolOptimizerTest extends TestCase
+{
+    /**
+     * @dataProvider provideIntegrationTests
+     * @param mixed[] $requestData
+     * @param BasePackage[] $packagesBefore
+     * @param BasePackage[] $expectedPackages
+     * @param string $message
+     */
+    public function testPoolOptimizer(array $requestData, array $packagesBefore, array $expectedPackages, $message)
+    {
+        $lockedRepo = new LockArrayRepository();
+
+        $request = new Request($lockedRepo);
+        $parser = new VersionParser();
+
+        if (isset($requestData['locked'])) {
+            foreach ($requestData['locked'] as $package) {
+                $request->lockPackage($this->loadPackage($package));
+            }
+        }
+        if (isset($requestData['fixed'])) {
+            foreach ($requestData['fixed'] as $package) {
+                $request->fixPackage($this->loadPackage($package));
+            }
+        }
+
+        foreach ($requestData['require'] as $package => $constraint) {
+            $request->requireName($package, $parser->parseConstraints($constraint));
+        }
+
+        $preferStable = isset($requestData['preferStable']) ? $requestData['preferStable'] : false;
+        $preferLowest = isset($requestData['preferLowest']) ? $requestData['preferLowest'] : false;
+
+        $pool = new Pool($packagesBefore);
+        $poolOptimizer = new PoolOptimizer(new DefaultPolicy($preferStable, $preferLowest));
+
+        $pool = $poolOptimizer->optimize($request, $pool);
+
+        $this->assertSame(
+            $this->reducePackagesInfoForComparison($expectedPackages),
+            $this->reducePackagesInfoForComparison($pool->getPackages()),
+            $message
+        );
+    }
+
+    public function provideIntegrationTests()
+    {
+        $fixturesDir = realpath(__DIR__.'/Fixtures/pooloptimizer/');
+        $tests = array();
+        foreach (new \RecursiveIteratorIterator(new \RecursiveDirectoryIterator($fixturesDir), \RecursiveIteratorIterator::LEAVES_ONLY) as $file) {
+            if (!preg_match('/\.test$/', $file)) {
+                continue;
+            }
+
+            try {
+                $testData = $this->readTestFile($file, $fixturesDir);
+                $message = $testData['TEST'];
+                $requestData = JsonFile::parseJson($testData['REQUEST']);
+                $packagesBefore = $this->loadPackages(JsonFile::parseJson($testData['POOL-BEFORE']));
+                $expectedPackages = $this->loadPackages(JsonFile::parseJson($testData['POOL-AFTER']));
+
+            } catch (\Exception $e) {
+                die(sprintf('Test "%s" is not valid: '.$e->getMessage(), str_replace($fixturesDir.'/', '', $file)));
+            }
+
+            $tests[basename($file)] = array($requestData, $packagesBefore, $expectedPackages, $message);
+        }
+
+        return $tests;
+    }
+
+    /**
+     * @param  string $fixturesDir
+     * @return mixed[]
+     */
+    protected function readTestFile(\SplFileInfo $file, $fixturesDir)
+    {
+        $tokens = preg_split('#(?:^|\n*)--([A-Z-]+)--\n#', file_get_contents($file->getRealPath()), -1, PREG_SPLIT_DELIM_CAPTURE);
+
+        /** @var array<string, bool> $sectionInfo */
+        $sectionInfo = array(
+            'TEST' => true,
+            'REQUEST' => true,
+            'POOL-BEFORE' => true,
+            'POOL-AFTER' => true,
+        );
+
+        $section = null;
+        $data = array();
+        foreach ($tokens as $i => $token) {
+            if (null === $section && empty($token)) {
+                continue; // skip leading blank
+            }
+
+            if (null === $section) {
+                if (!isset($sectionInfo[$token])) {
+                    throw new \RuntimeException(sprintf(
+                        'The test file "%s" must not contain a section named "%s".',
+                        str_replace($fixturesDir.'/', '', $file),
+                        $token
+                    ));
+                }
+                $section = $token;
+                continue;
+            }
+
+            $sectionData = $token;
+
+            $data[$section] = $sectionData;
+            $section = $sectionData = null;
+        }
+
+        foreach ($sectionInfo as $section => $required) {
+            if ($required && !isset($data[$section])) {
+                throw new \RuntimeException(sprintf(
+                    'The test file "%s" must have a section named "%s".',
+                    str_replace($fixturesDir.'/', '', $file),
+                    $section
+                ));
+            }
+        }
+
+        return $data;
+    }
+
+    /**
+     * @param BasePackage[] $packages
+     * @return string[]
+     */
+    private function reducePackagesInfoForComparison(array $packages)
+    {
+        $packagesInfo = array();
+
+        foreach ($packages as $package) {
+            $packagesInfo[] = $package->getName() . '@' . $package->getVersion() . ($package instanceof AliasPackage ? ' (alias of '.$package->getAliasOf()->getVersion().')' : '');
+        }
+
+        sort($packagesInfo);
+
+        return $packagesInfo;
+    }
+
+    /**
+     * @param mixed[][] $packagesData
+     * @return BasePackage[]
+     */
+    private function loadPackages(array $packagesData)
+    {
+        $packages = array();
+
+        foreach ($packagesData as $packageData) {
+            $packages[] = $package = $this->loadPackage($packageData);
+            if ($package instanceof AliasPackage) {
+                $packages[] = $package->getAliasOf();
+            }
+        }
+
+        return $packages;
+    }
+
+    /**
+     * @param mixed[] $packageData
+     * @return BasePackage
+     */
+    private function loadPackage(array $packageData)
+    {
+        $loader = new ArrayLoader();
+        return $loader->load($packageData);
+    }
+}