Fix ContainerAware deprecation notice

KnpLabs · Nov 13, 2015 · cca603c · cca603c
1 parent 135044b
commit cca603c
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/Tests/Stubs/Menu/ContainerAwareBuilder.php b/Tests/Stubs/Menu/ContainerAwareBuilder.php
@@ -3,10 +3,18 @@
 namespace Knp\Bundle\MenuBundle\Tests\Stubs\Menu;
 
 use Knp\Menu\FactoryInterface;
-use Symfony\Component\DependencyInjection\ContainerAware;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+use Symfony\Component\DependencyInjection\ContainerAwareInterface;
 
-class ContainerAwareBuilder extends ContainerAware
+class ContainerAwareBuilder implements ContainerAwareInterface
 {
+    private $container;
+
+    public function setContainer(ContainerInterface $container = null)
+    {
+        $this->container = $container;
+    }
+
     public function mainMenu(FactoryInterface $factory)
     {
         // Check that the container is really set.