forward compatibility with Symfony 4

xabbuh · xabbuh · commit af7d48dfbfbf · 2017-05-21T15:20:06.000+02:00
diff --git a/src/Symfony/Bundle/SecurityBundle/DataCollector/SecurityDataCollector.php b/src/Symfony/Bundle/SecurityBundle/DataCollector/SecurityDataCollector.php
@@ -12,6 +12,7 @@
 namespace Symfony\Bundle\SecurityBundle\DataCollector;
 
 use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface;
+use Symfony\Component\Security\Core\Role\Role;
 use Symfony\Component\Security\Core\Role\RoleHierarchyInterface;
 use Symfony\Component\HttpFoundation\Request;
 use Symfony\Component\HttpFoundation\Response;
@@ -117,8 +118,26 @@ public function collect(Request $request, Response $response, \Exception $except
                 'token_class' => $this->hasVarDumper ? new ClassStub(get_class($token)) : get_class($token),
                 'logout_url' => $logoutUrl,
                 'user' => $token->getUsername(),
-                'roles' => array_map(function (RoleInterface $role) { return $role->getRole(); }, $assignedRoles),
-                'inherited_roles' => array_map(function (RoleInterface $role) { return $role->getRole(); }, $inheritedRoles),
+                'roles' => array_map(
+                    function ($role) {
+                        if (!$role instanceof RoleInterface && !$role instanceof Role) {
+                            throw new \InvalidArgumentException(sprintf('Roles must be instances of %s or %s (%s given).', RoleInterface::class, Role::class, is_object($role) ? get_class($role) : gettype($role)));
+                        }
+
+                        return $role->getRole();
+                    },
+                    $assignedRoles
+                ),
+                'inherited_roles' => array_map(
+                    function ($role) {
+                        if (!$role instanceof RoleInterface && !$role instanceof Role) {
+                            throw new \InvalidArgumentException(sprintf('Roles must be instances of %s or %s (%s given).', RoleInterface::class, Role::class, is_object($role) ? get_class($role) : gettype($role)));
+                        }
+
+                        return $role->getRole();
+                    },
+                    $inheritedRoles
+                ),
                 'supports_role_hierarchy' => null !== $this->roleHierarchy,
             );
         }
