Allow for initial null state for container property in ContainerAwareTrait

Author: Ben Plunkett

src/Neto/Container/ContainerAwareTrait.php

@@ -6,7 +6,7 @@
 
 trait ContainerAwareTrait
 {
-    /** @var ContainerInterface */
+    /** @var ContainerInterface|null */
     private $container;
 
     /**
@@ -22,9 +22,9 @@ public function setContainer(ContainerInterface $container)
     }
 
     /**
-     * @return ContainerInterface
+     * @return ContainerInterface|null
      */
-    public function getContainer(): ContainerInterface
+    public function getContainer(): ?ContainerInterface
     {
         return $this->container;
     }

tests/ContainerAwareTest.php

@@ -17,4 +17,10 @@ public function testTraitGetsAndSetsContainer()
         $trait->setContainer($container);
         $this->assertSame($container, $trait->getContainer());
     }
-}
+
+    public function testTraitCanGetNullContainer()
+    {
+        $trait = $this->getMockForTrait(ContainerAwareTrait::class);
+        $this->assertNull($trait->getContainer());
+    }
+}