diff --git a/src/Context/FixtureContext.php b/src/Context/FixtureContext.php index 4c3b11c..d60dbca 100644 --- a/src/Context/FixtureContext.php +++ b/src/Context/FixtureContext.php @@ -12,19 +12,23 @@ use InvalidArgumentException; use SilverStripe\Assets\Folder; use SilverStripe\Assets\Storage\AssetStore; use SilverStripe\Core\ClassInfo; +use SilverStripe\Core\Extensible; +use SilverStripe\Core\Extension; use SilverStripe\Core\Injector\Injector; +use SilverStripe\Core\Manifest\ModuleLoader; +use SilverStripe\Core\Manifest\ModuleManifest; use SilverStripe\Dev\BehatFixtureFactory; use SilverStripe\Dev\FixtureBlueprint; use SilverStripe\Dev\FixtureFactory; -use SilverStripe\Dev\SapphireTest; use SilverStripe\Dev\YamlFixture; use SilverStripe\ORM\Connect\TempDatabase; -use SilverStripe\ORM\DB; +use SilverStripe\ORM\DataExtension; use SilverStripe\ORM\DataObject; -use SilverStripe\Versioned\Versioned; +use SilverStripe\ORM\DB; use SilverStripe\Security\Group; use SilverStripe\Security\Member; use SilverStripe\Security\Permission; +use SilverStripe\Versioned\Versioned; /** * Context used to create fixtures in the SilverStripe ORM. @@ -55,10 +59,15 @@ class FixtureContext implements Context protected $tempDatabase; /** - * @var String Tracks all files and folders created from fixtures, for later cleanup. + * @var string[] Tracks all files and folders created from fixtures, for later cleanup. */ protected $createdFilesPaths = array(); + /** + * @var string[] Tracks any config files that have been activated as part of a scenario + */ + protected $activatedConfigFiles = array(); + /** * @var array Stores the asset tuples. */ @@ -632,6 +641,66 @@ class FixtureContext implements Context $this->getMainContext()->getSession()->visit($this->getMainContext()->locatePath($link)); } + /** + * @param $extension + * @param $class + * + * @Given I add an extension :extension to the :class class + */ + public function iAddAnExtensionToTheClass($extension, $class) + { + // Validate the extension + assertTrue( + class_exists($extension) && is_subclass_of($extension, Extension::class), + 'Given extension does not extend Extension' + ); + + // Add the extension to the CLI context + /** @var Extensible $targetClass */ + $targetClass = $this->convertTypeToClass($class); + $targetClass::add_extension($extension); + + // Write config for this extension too... + $snakedExtension = strtolower(str_replace('\\', '-', $extension)); + $config = <<getDestinationConfigFolder($filename); + file_put_contents($destPath, $config); + + // Remember to cleanup... + $this->activatedConfigFiles[] = $destPath; + + // Flush website. We'll need to dev/build too if it's a DataExtension + if (is_subclass_of($extension, DataExtension::class)) { + $this->getMainContext()->visit('/dev/build?flush'); + } else { + $this->getMainContext()->visit('/?flush'); + } + } + + /** + * Get the destination folder for config and assert the given file name doesn't exist within in. + * + * @param $filename + * @return string + */ + protected function getDestinationConfigFolder($filename) + { + $project = ModuleManifest::config()->get('project') ?: 'mysite'; + $mysite = ModuleLoader::getModule($project); + assertNotNull($mysite, 'Project exists'); + $destPath = $mysite->getResource("_config/{$filename}")->getPath(); + assertFileNotExists($destPath, "Config file {$filename} hasn't aleady been loaded"); + return $destPath; + } /** * Checks that a file or folder exists in the webroot. @@ -714,6 +783,19 @@ class FixtureContext implements Context } } + /** + * Clean up all config files after scenario + * + * @AfterScenario + * @param AfterScenarioScope $event + */ + public function afterResetConfig(AfterScenarioScope $event) + { + $this->clearConfigFiles(); + // Flush + $this->getMainContext()->visit('/?flush'); + } + /** * Prepares a fixture for use * @@ -867,4 +949,27 @@ class FixtureContext implements Context } return join('/', $paths); } + + protected function clearConfigFiles() + { + // No files to cleanup + if (empty($this->activatedConfigFiles)) { + return; + } + + foreach ($this->activatedConfigFiles as $configFile) { + if (file_exists($configFile)) { + unlink($configFile); + } + } + $this->activatedConfigFiles = []; + } + + /** + * Catch situations where failed scenarios and early exiting would prevent cleanup. + */ + public function __destruct() + { + $this->clearConfigFiles(); + } }