[PhpUnit] Mock time on @group clockMock annotations

src/Symfony/Bridge/PhpUnit/ClockMock.php

@@ -0,0 +1,113 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Bridge\PhpUnit;
+
+/**
+ * @author Nicolas Grekas <p@tchwork.com>
+ */
+class ClockMock
+{
+    private static $now;
+
+    public static function withClockMock($enable = null)
+    {
+        if (null === $enable) {
+            return null !== self::$now;
+        }
+
+        self::$now = is_numeric($enable) ? (float) $enable : ($enable ? microtime(true) : null);
+    }
+
+    public static function time()
+    {
+        if (null === self::$now) {
+            return \time();
+        }
+
+        return (int) self::$now;
+    }
+
+    public static function sleep($s)
+    {
+        if (null === self::$now) {
+            return \sleep($s);
+        }
+
+        self::$now += (int) $s;
+
+        return 0;
+    }
+
+    public static function usleep($us)
+    {
+        if (null === self::$now) {
+            return \usleep($us);
+        }
+
+        self::$now += $us / 1000000;
+    }
+
+    public static function microtime($asFloat = false)
+    {
+        if (null === self::$now) {
+            return \microtime($asFloat);
+        }
+
+        if ($asFloat) {
+            return self::$now;
+        }
+
+        return sprintf("%0.6f %d\n", $now - (int) $now, (int) self::$now);
+    }
+
+    public static function register($class)
+    {
+        $self = get_called_class();
+
+        $mockedNs = array(substr($class, 0, strrpos($class, '\\')));
+        if (strpos($class, '\\Tests\\')) {
+            $ns = str_replace('\\Tests\\', '\\', $class);
+            $mockedNs[] = substr($ns, 0, strrpos($ns, '\\'));
+        }
+        foreach ($mockedNs as $ns) {
+            if (function_exists($ns.'\time')) {
+                continue;
+            }
+            eval(<<<EOPHP
+namespace $ns;
+
+function time()
+{
+    return \\$self::time();
+}
+
+function microtime(\$asFloat = false)
+{
+    return \\$self::microtime(\$asFloat);
+}
+
+function sleep(\$s)
+{
+    return \\$self::sleep(\$s);
+}
+
+function usleep(\$us)
+{
+    return \\$self::usleep(\$us);
+}
+
+EOPHP
+            );
+        }
+    }
+
+}

src/Symfony/Bridge/PhpUnit/SymfonyTestsListener.php

@@ -24,6 +24,7 @@ class SymfonyTestsListener extends \PHPUnit_Framework_BaseTestListener
     private $skippedFile = false;
     private $wasSkipped = array();
     private $isSkipped = array();
+    private $testsStack = array();
 
     public function __destruct()
     {
@@ -82,4 +83,27 @@ public function addSkippedTest(\PHPUnit_Framework_Test $test, \Exception $e, $ti
             $this->isSkipped[$class][$method] = 1;
         }
     }
+
+    public function startTest(\PHPUnit_Framework_Test $test)
+    {
+        if ($test instanceof \PHPUnit_Framework_TestCase) {
+            $groups = \PHPUnit_Util_Test::getGroups(get_class($test), $test->getName());
+
+            if (in_array('clockMock', $groups, true)) {
+                ClockMock::register(get_class($test));
+                ClockMock::withClockMock(true);
+            }
+        }
+    }
+
+    public function endTest(\PHPUnit_Framework_Test $test, $time)
+    {
+        if ($test instanceof \PHPUnit_Framework_TestCase) {
+            $groups = \PHPUnit_Util_Test::getGroups(get_class($test), $test->getName());
+
+            if (in_array('clockMock', $groups, true)) {
+                ClockMock::withClockMock(false);
+            }
+        }
+    }
 }

src/Symfony/Component/Console/Tests/Helper/LegacyProgressHelperTest.php

@@ -13,25 +13,13 @@
 
 use Symfony\Component\Console\Helper\ProgressHelper;
 use Symfony\Component\Console\Output\StreamOutput;
-use Symfony\Component\Console\Tests;
-
-require_once __DIR__.'/../ClockMock.php';
 
 /**
  * @group legacy
+ * @group clockMock
  */
 class LegacyProgressHelperTest extends \PHPUnit_Framework_TestCase
 {
-    protected function setUp()
-    {
-        Tests\with_clock_mock(true);
-    }
-
-    protected function tearDown()
-    {
-        Tests\with_clock_mock(false);
-    }
-
     public function testAdvance()
     {
         $progress = new ProgressHelper();

src/Symfony/Component/Console/Tests/Helper/ProgressBarTest.php

@@ -14,22 +14,12 @@
 use Symfony\Component\Console\Helper\ProgressBar;
 use Symfony\Component\Console\Helper\Helper;
 use Symfony\Component\Console\Output\StreamOutput;
-use Symfony\Component\Console\Tests;
-
-require_once __DIR__.'/../ClockMock.php';
 
+/**
+ * @group clockMock
+ */
 class ProgressBarTest extends \PHPUnit_Framework_TestCase
 {
-    protected function setUp()
-    {
-        Tests\with_clock_mock(true);
-    }
-
-    protected function tearDown()
-    {
-        Tests\with_clock_mock(false);
-    }
-
     public function testMultipleStart()
     {
         $bar = new ProgressBar($output = $this->getOutputStream());

src/Symfony/Component/HttpFoundation/Tests/CookieTest.php

@@ -18,8 +18,10 @@
  *
  * @author John Kary <john@johnkary.net>
  * @author Hugo Hamon <hugo.hamon@sensio.com>
+ *
+ * @group clockMock
  */
-class CookieTest extends ClockMockTestCase
+class CookieTest extends \PHPUnit_Framework_TestCase
 {
     public function invalidNames()
     {

src/Symfony/Component/HttpFoundation/Tests/ResponseHeaderBagTest.php

@@ -14,7 +14,10 @@
 use Symfony\Component\HttpFoundation\ResponseHeaderBag;
 use Symfony\Component\HttpFoundation\Cookie;
 
-class ResponseHeaderBagTest extends ClockMockTestCase
+/**
+ * @group clockMock
+ */
+class ResponseHeaderBagTest extends \PHPUnit_Framework_TestCase
 {
     /**
      * @covers Symfony\Component\HttpFoundation\ResponseHeaderBag::allPreserveCase

src/Symfony/Component/HttpFoundation/Tests/Session/Storage/Handler/LegacyPdoSessionHandlerTest.php

@@ -12,13 +12,13 @@
 namespace Symfony\Component\HttpFoundation\Tests\Session\Storage\Handler;
 
 use Symfony\Component\HttpFoundation\Session\Storage\Handler\LegacyPdoSessionHandler;
-use Symfony\Component\HttpFoundation\Tests\ClockMockTestCase;
 
 /**
  * @group legacy
  * @requires extension pdo_sqlite
+ * @group clockMock
  */
-class LegacyPdoSessionHandlerTest extends ClockMockTestCase
+class LegacyPdoSessionHandlerTest extends \PHPUnit_Framework_TestCase
 {
     private $pdo;
 

src/Symfony/Component/HttpFoundation/Tests/Session/Storage/Handler/MemcacheSessionHandlerTest.php

@@ -12,12 +12,12 @@
 namespace Symfony\Component\HttpFoundation\Tests\Session\Storage\Handler;
 
 use Symfony\Component\HttpFoundation\Session\Storage\Handler\MemcacheSessionHandler;
-use Symfony\Component\HttpFoundation\Tests\ClockMockTestCase;
 
 /**
  * @requires extension memcache
+ * @group clockMock
  */
-class MemcacheSessionHandlerTest extends ClockMockTestCase
+class MemcacheSessionHandlerTest extends \PHPUnit_Framework_TestCase
 {
     const PREFIX = 'prefix_';
     const TTL = 1000;