core/tests/Drupal/Tests/Core/Template/TwigExtensionTest.php | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/core/tests/Drupal/Tests/Core/Template/TwigExtensionTest.php b/core/tests/Drupal/Tests/Core/Template/TwigExtensionTest.php
index ea93703..4a1c5e8 100644
--- a/core/tests/Drupal/Tests/Core/Template/TwigExtensionTest.php
+++ b/core/tests/Drupal/Tests/Core/Template/TwigExtensionTest.php
@@ -15,6 +15,7 @@
 use Drupal\Core\Template\Loader\StringLoader;
 use Drupal\Core\Template\TwigEnvironment;
 use Drupal\Core\Template\TwigExtension;
+use Drupal\Core\Theme\Registry;
 use Drupal\Tests\UnitTestCase;
 
 /**
@@ -39,7 +40,8 @@ public function testEscaping($template, $expected) {
       'autoescape' => 'html',
       'optimizations' => 0
     ));
-    $twig->addExtension((new TwigExtension($renderer))->setUrlGenerator($this->getMock('Drupal\Core\Routing\UrlGeneratorInterface')));
+    $registry = $this->prophesize(Registry::class);
+    $twig->addExtension((new TwigExtension($renderer))->setUrlGenerator($this->getMock('Drupal\Core\Routing\UrlGeneratorInterface'))->setThemeRegistry($registry->reveal()));
 
     $nodes = $twig->parse($twig->tokenize($template));
 
@@ -84,7 +86,9 @@ public function providerTestEscaping() {
    */
   public function testActiveTheme() {
     $renderer = $this->getMock('\Drupal\Core\Render\RendererInterface');
+    $registry = $this->prophesize(Registry::class);
     $extension = new TwigExtension($renderer);
+    $extension->setThemeRegistry($registry->reveal());
     $theme_manager = $this->getMock('\Drupal\Core\Theme\ThemeManagerInterface');
     $active_theme = $this->getMockBuilder('\Drupal\Core\Theme\ActiveTheme')
       ->disableOriginalConstructor()
@@ -117,7 +121,9 @@ public function testFormatDate() {
       ->method('format')
       ->willReturn('1978-11-19');
     $renderer = $this->getMock('\Drupal\Core\Render\RendererInterface');
+    $registry = $this->prophesize(Registry::class);
     $extension = new TwigExtension($renderer);
+    $extension->setThemeRegistry($registry->reveal());
     $extension->setDateFormatter($date_formatter);
 
     $loader = new StringLoader();
@@ -132,7 +138,9 @@ public function testFormatDate() {
    */
   public function testActiveThemePath() {
     $renderer = $this->getMock('\Drupal\Core\Render\RendererInterface');
+    $registry = $this->prophesize(Registry::class);
     $extension = new TwigExtension($renderer);
+    $extension->setThemeRegistry($registry->reveal());
     $theme_manager = $this->getMock('\Drupal\Core\Theme\ThemeManagerInterface');
     $active_theme = $this->getMockBuilder('\Drupal\Core\Theme\ActiveTheme')
       ->disableOriginalConstructor()