Added Twig template functions

2 files changed, 9 insertions, 2 deletions

diff --git a/tests/Unit/Http/SessionServiceProviderTest.php b/tests/Unit/Http/SessionServiceProviderTest.php
index a78b4f72..d0125bc2 100644
--- a/tests/Unit/Http/SessionServiceProviderTest.php
+++ b/tests/Unit/Http/SessionServiceProviderTest.php
@@ -54,6 +54,7 @@ class SessionServiceProviderTest extends ServiceProviderTest
             ->method('instance')
             ->withConsecutive(
                 ['session.storage', $sessionStorage],
+                [Session::class, $session],
                 ['session', $session]
             );
 
@@ -88,10 +89,11 @@ class SessionServiceProviderTest extends ServiceProviderTest
                 $sessionStorage,
                 $session
             );
-        $app->expects($this->exactly(2))
+        $app->expects($this->exactly(3))
             ->method('instance')
             ->withConsecutive(
                 ['session.storage', $sessionStorage],
+                [Session::class, $session],
                 ['session', $session]
             );
 
diff --git a/tests/Unit/Http/UrlGeneratorServiceProviderTest.php b/tests/Unit/Http/UrlGeneratorServiceProviderTest.php
index f3d5e018..787789fe 100644
--- a/tests/Unit/Http/UrlGeneratorServiceProviderTest.php
+++ b/tests/Unit/Http/UrlGeneratorServiceProviderTest.php
@@ -21,7 +21,12 @@ class UrlGeneratorServiceProviderTest extends ServiceProviderTest
         $app = $this->getApp();
 
         $this->setExpects($app, 'make', [UrlGenerator::class], $urlGenerator);
-        $this->setExpects($app, 'instance', ['http.urlGenerator', $urlGenerator]);
+        $app->expects($this->exactly(2))
+            ->method('instance')
+            ->withConsecutive(
+                [UrlGenerator::class, $urlGenerator],
+                ['http.urlGenerator', $urlGenerator]
+            );
 
         $serviceProvider = new UrlGeneratorServiceProvider($app);
         $serviceProvider->register();