Fix test

PHP-DI · May 24, 2017 · 9a6eb98 · 9a6eb98
1 parent f78897f
commit 9a6eb98
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/UnitTest/Definition/Source/CachedDefinitionSourceTest.php b/tests/UnitTest/Definition/Source/CachedDefinitionSourceTest.php
@@ -2,6 +2,7 @@
 
 namespace DI\Test\UnitTest\Definition\Source;
 
+use DI\Definition\AliasDefinition;
 use DI\Definition\ObjectDefinition;
 use DI\Definition\Source\CachedDefinitionSource;
 use DI\Definition\Source\DefinitionArray;
@@ -28,16 +29,18 @@ public function setUp()
      */
     public function should_get_from_cache()
     {
+        $definition = new AliasDefinition('foo', 'bar');
+
         $source = $this->createMock(DefinitionSource::class);
         $source
             ->expects($this->once()) // The sub-source should be called ONLY ONCE
             ->method('getDefinition')
-            ->willReturn('bar');
+            ->willReturn($definition);
 
         $source = new CachedDefinitionSource($source);
 
-        self::assertEquals('bar', $source->getDefinition('foo'));
-        self::assertEquals('bar', $source->getDefinition('foo'));
+        self::assertEquals($definition, $source->getDefinition('foo'));
+        self::assertEquals($definition, $source->getDefinition('foo'));
     }
 
     /**