Fixed token provider tests

Author: dantleech

Tests/Unit/AutoRoute/TokenProvider/ContentDateTimeProviderTest.php

@@ -9,13 +9,15 @@ class ContentDateTimeProviderTest extends BaseTestCase
 {
     protected $slugifier;
     protected $article;
+    protected $urlContext;
 
     public function setUp()
     {
         parent::setUp();
 
-        $this->slugifier = $this->prophet->prophesize('Symfony\Cmf\Bundle\CoreBundle\Slugifier\SlugifierInterface');
-        $this->article = $this->prophet->prophesize('Symfony\Cmf\Bundle\RoutingAutoBundle\Tests\Resources\Document\Article');
+        $this->slugifier = $this->prophesize('Symfony\Cmf\Bundle\CoreBundle\Slugifier\SlugifierInterface');
+        $this->article = $this->prophesize('Symfony\Cmf\Bundle\RoutingAutoBundle\Tests\Resources\Document\Article');
+        $this->urlContext = $this->prophesize('Symfony\Cmf\Bundle\RoutingAutoBundle\AutoRoute\UrlContext');
         $this->provider = new ContentDateTimeProvider($this->slugifier->reveal());
     }
 
@@ -47,9 +49,10 @@ public function testGetValue($options, $expectedResult)
             'slugify' => true,
         ), $options);
 
+        $this->urlContext->getSubjectObject()->willReturn($this->article);
         $this->article->getDate()->willReturn(new \DateTime('2014-10-09'));
 
-        $res = $this->provider->provideValue($this->article->reveal(), $options);
+        $res = $this->provider->provideValue($this->urlContext->reveal(), $options);
 
         $this->assertEquals($expectedResult, $res);
     }

Tests/Unit/AutoRoute/TokenProvider/ContentMethodProviderTest.php

@@ -9,13 +9,15 @@ class ContentMethodProviderTest extends BaseTestCase
 {
     protected $slugifier;
     protected $article;
+    protected $urlContext;
 
     public function setUp()
     {
         parent::setUp();
 
-        $this->slugifier = $this->prophet->prophesize('Symfony\Cmf\Bundle\CoreBundle\Slugifier\SlugifierInterface');
-        $this->article = $this->prophet->prophesize('Symfony\Cmf\Bundle\RoutingAutoBundle\Tests\Resources\Document\Article');
+        $this->slugifier = $this->prophesize('Symfony\Cmf\Bundle\CoreBundle\Slugifier\SlugifierInterface');
+        $this->article = $this->prophesize('Symfony\Cmf\Bundle\RoutingAutoBundle\Tests\Resources\Document\Article');
+        $this->urlContext = $this->prophesize('Symfony\Cmf\Bundle\RoutingAutoBundle\AutoRoute\UrlContext');
         $this->provider = new ContentMethodProvider($this->slugifier->reveal());
     }
 
@@ -52,6 +54,7 @@ public function provideGetValue()
     public function testGetValue($options, $methodExists = false)
     {
         $method = $options['method'];
+        $this->urlContext->getSubjectObject()->willReturn($this->article);
 
         if (!$methodExists) {
             $this->setExpectedException(
@@ -67,7 +70,7 @@ public function testGetValue($options, $methodExists = false)
             $this->slugifier->slugify('This is value')->willReturn($expectedResult);
         }
 
-        $res = $this->provider->provideValue($this->article->reveal(), $options);
+        $res = $this->provider->provideValue($this->urlContext->reveal(), $options);
 
         $this->assertEquals($expectedResult, $res);
     }

Tests/Unit/BaseTestCase.php

@@ -17,4 +17,9 @@ public function tearDown()
     {
         $this->prophet->checkPredictions();
     }
+
+    public function prophesize($classOrInterface = null)
+    {
+        return $this->prophet->prophesize($classOrInterface);
+    }
 }