feat: add container builder reference to container

phetit · Jul 25, 2023 · e8d068b · e8d068b
1 parent 1ee3b4b
commit e8d068b
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # Changelog
 
+## [Unreleased]
+
+### Added
+
+- Auto inject container reference into container
+
 ## [0.6.0] - 2023-07-25
 
 ### Added

diff --git a/src/ContainerBuilder.php b/src/ContainerBuilder.php
@@ -7,9 +7,15 @@
 use Closure;
 use Phetit\DependencyInjection\Resolver\FactoryServiceResolver;
 use Phetit\DependencyInjection\Resolver\ServiceResolver;
+use Psr\Container\ContainerInterface;
 
 class ContainerBuilder extends Container
 {
+    public function __construct()
+    {
+        $this->parameter(ContainerInterface::class, $this);
+    }
+
     public function register(string $id, Closure $resolver): static
     {
         $this->set($id, new ServiceResolver($resolver));

diff --git a/tests/ContainerBuilderTest.php b/tests/ContainerBuilderTest.php
@@ -7,10 +7,11 @@
 use Phetit\DependencyInjection\ContainerBuilder;
 use Phetit\DependencyInjection\Tests\Fixtures\Service;
 use PHPUnit\Framework\TestCase;
+use Psr\Container\ContainerInterface;
 
 class ContainerBuilderTest extends TestCase
 {
-    public function testRegisterMethod(): void
+    public function testRegisterMethodShouldAddServiceResolver(): void
     {
         $builder = new ContainerBuilder();
 
@@ -25,7 +26,7 @@ public function testRegisterMethod(): void
         self::assertSame($serviceOne, $serviceTwo);
     }
 
-    public function testFactoryMethod(): void
+    public function testFactoryMethodShouldAddFactoryServiceResolver(): void
     {
         $builder = new ContainerBuilder();
 
@@ -39,4 +40,14 @@ public function testFactoryMethod(): void
 
         self::assertNotSame($serviceOne, $serviceTwo);
     }
+
+    public function testShouldInjectAutoReference(): void
+    {
+        $builder = new ContainerBuilder();
+
+        $container = $builder->get(ContainerInterface::class);
+
+        self::assertInstanceOf(ContainerBuilder::class, $container);
+        self::assertSame($builder, $container);
+    }
 }