silverstripe-cms/tests/behat/features/bootstrap/SilverStripe/Cms/Test/Behaviour/ThemeContext.php

<?php

namespace SilverStripe\Cms\Test\Behaviour;

use Behat\Behat\Context\ClosuredContextInterface;
use Behat\Behat\Context\TranslatedContextInterface;
use Behat\Behat\Context\BehatContext;
use Behat\Behat\Context\Step;
use Behat\Behat\Event\StepEvent;
use Behat\Behat\Exception\PendingException;
use Behat\Mink\Driver\Selenium2Driver;
use Behat\Gherkin\Node\PyStringNode;
use Behat\Gherkin\Node\TableNode;

/**
 * Context used to create fixtures in the SilverStripe ORM.
 */
class ThemeContext extends BehatContext
{

    protected $restoreFiles = array();
    protected $restoreDirectories = array();

    /**
     * Create a test theme
     *
     * @Given /^a theme "(?<theme>[^"]+)"/
     */
    public function stepCreateTheme($theme)
    {
        if (!preg_match('/^[0-9a-zA-Z_-]+$/', $theme)) {
            throw new \InvalidArgumentException("Bad theme '$theme'");
        }

        $this->requireDir(BASE_PATH . '/themes');
        $this->requireDir(BASE_PATH . '/themes/' . $theme);
        $this->requireDir(BASE_PATH . '/themes/' . $theme . '/templates');
    }

    /**
     * Create a template within a test theme
     *
     * @Given /^a template "(?<template>[^"]+)" in theme "(?<theme>[^"]+)" with content "(?<content>[^"]+)"/
     */
    public function stepCreateTemplate($template, $theme, $content)
    {
        if (!preg_match('/^[0-9a-zA-Z_-]+$/', $theme)) {
            throw new \InvalidArgumentException("Bad theme '$theme'");
        }
        if (!preg_match('/^(Layout\/)?[0-9a-zA-Z_-]+\.ss$/', $template)) {
            throw new \InvalidArgumentException("Bad template '$template'");
        }

        $this->stepCreateTheme($theme);
        $this->requireFile(BASE_PATH . '/themes/' . $theme . '/templates/' . $template, $content);
    }

    protected function requireFile($filename, $content)
    {
        // Already exists
        if (file_exists($filename)) {
            // If the content is different, remember old content for restoration
            $origContent = file_get_contents($filename);
            if ($origContent != $content) {
                file_put_contents($filename, $content);
                $this->restoreFiles[$filename] = $origContent;
            }
        // Doesn't exist, mark it for deletion after test
        } else {
            file_put_contents($filename, $content);
            $this->restoreFiles[$filename] = null;
        }
    }

    protected function requireDir($dirname)
    {
        // Directory doesn't exist, create it and mark it for deletion
        if (!file_exists($dirname)) {
            mkdir($dirname);
            $this->restoreDirectories[] = $dirname;
        }
    }

    /**
     * Clean up any theme manipulation
     *
     * @AfterScenario
     */
    public function cleanThemesAfterScenario()
    {
        // Restore any created/modified files.
        //  - If modified, revert then to original contnet
        //  - If created, delete them
        if ($this->restoreFiles) {
            foreach ($this->restoreFiles as $file => $origContent) {
                if ($origContent === null) {
                    unlink($file);
                } else {
                    file_put_contents($file, $origContent);
                }
            }

            $this->restoreFiles = array();
        }

        // Restore any created directories: that is, delete them
        if ($this->restoreDirectories) {
            // Flip the order so that nested direcotires are unlinked() first
            $this->restoreDirectories = array_reverse($this->restoreDirectories);
            foreach ($this->restoreDirectories as $dir) {
                rmdir($dir);
            }

            $this->restoreDirectories = array();
        }
    }
}