fix: issue on ContainerAware when theres no Container instance assigned

Signed-off-by: Fery Wardiyanto <ferywardiyanto@gmail.com>

Author: feryardiant

src/Container/ContainerAware.php

@@ -7,7 +7,7 @@
 trait ContainerAware
 {
     /**
-     * @var ContainerInterface
+     * @var null|ContainerInterface
      */
     protected $container = null;
 
@@ -22,9 +22,9 @@ public function setContainer(ContainerInterface $container): void
     /**
      * @see ContainerAwareInterface::getContainer()
      */
-    public function getContainer(?string $name = null)
+    public function getContainer(?string $name = null): ?object
     {
-        if ($name) {
+        if ($this->container && $name) {
             return $this->container->get($name);
         }
 

src/Container/ContainerAwareInterface.php

@@ -44,5 +44,5 @@ public function setContainer(ContainerInterface $container): void;
      * @param null|string $name Optionally pass a container name, if needed.
      * @return null|mixed|ContainerInterface
      */
-    public function getContainer(?string $name = null);
+    public function getContainer(?string $name = null): ?object;
 }

test/spec/ContainerAware.spec.php

@@ -0,0 +1,28 @@
+<?php
+
+use Projek\Container;
+use Projek\Container\{ContainerAware, ContainerAwareInterface, ContainerInterface};
+
+use function Kahlan\{describe, expect, given, it};
+
+describe(ContainerAware::class, function () {
+    given('container', function () {
+        return new Container;
+    });
+
+    given('stub', function () {
+        return new class implements ContainerAwareInterface {
+            use ContainerAware;
+        };
+    });
+
+    it('should returns null if no container assigned', function () {
+        expect($this->stub->getContainer())->toBeNull();
+    });
+
+    it('should assign container', function () {
+        $this->stub->setContainer($this->container);
+
+        expect($this->stub->getContainer())->toBeAnInstanceOf(ContainerInterface::class);
+    });
+});