Check if method getLocale exists before to use it (#298)

src/EventSubscriber/UserLocaleSubscriber.php

@@ -40,7 +40,7 @@ public function onInteractiveLogin(InteractiveLoginEvent $event)
     {
         $user = $event->getAuthenticationToken()->getUser();
 
-        if (null !== $user->getLocale()) {
+        if (\is_callable([$user, 'getLocale']) && null !== $user->getLocale()) {
             $this->session->set('_locale', $user->getLocale());
         }
     }

tests/EventSubscriber/LocalizedUser.php

@@ -0,0 +1,39 @@
+<?php
+
+declare(strict_types=1);
+
+/*
+ * This file is part of the Sonata Project package.
+ *
+ * (c) Thomas Rabaix <thomas.rabaix@sonata-project.org>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Sonata\TranslationBundle\Tests\EventSubscriber;
+
+use Symfony\Component\Security\Core\User\UserInterface;
+
+/**
+ * @author Jonathan Vautrin <jvautrin@pro-info.be>
+ */
+class LocalizedUser extends User implements UserInterface
+{
+    private $locale;
+
+    public function __construct(string $locale)
+    {
+        $this->locale = $locale;
+    }
+
+    public function getLocale()
+    {
+        return $this->locale;
+    }
+
+    public function setLocale($locale)
+    {
+        $this->locale = $locale;
+    }
+}

tests/EventSubscriber/User.php

@@ -20,13 +20,6 @@
  */
 class User implements UserInterface
 {
-    private $locale;
-
-    public function __construct($locale)
-    {
-        $this->locale = $locale;
-    }
-
     public function getRoles()
     {
         return [];
@@ -50,14 +43,4 @@ public function getUsername()
     public function eraseCredentials()
     {
     }
-
-    public function getLocale()
-    {
-        return $this->locale;
-    }
-
-    public function setLocale($locale)
-    {
-        $this->locale = $locale;
-    }
 }

tests/EventSubscriber/UserLocaleSubscriberTest.php

@@ -19,6 +19,7 @@
 use Symfony\Component\HttpFoundation\Session\Session;
 use Symfony\Component\HttpFoundation\Session\Storage\MockArraySessionStorage;
 use Symfony\Component\Security\Core\Authentication\Token\UsernamePasswordToken;
+use Symfony\Component\Security\Core\User\UserInterface;
 use Symfony\Component\Security\Http\Event\InteractiveLoginEvent;
 
 /**
@@ -28,21 +29,30 @@ class UserLocaleSubscriberTest extends TestCase
 {
     /**
      * Check if session locale is set to user locale at login.
+     *
+     * @dataProvider userLocaleSubscriberDataProvider
      */
-    public function testUserLocaleSubscriber()
+    public function testUserLocaleSubscriber(UserInterface $user, string $expectedLocale): void
     {
         $session = new Session(new MockArraySessionStorage());
         $session->set('_locale', 'en');
         $request = new Request();
         $request->setSession($session);
-        $user = new User('fr');
         $event = $this->getEvent($request, $user);
         $userLocaleSubscriber = new UserLocaleSubscriber($session);
         $userLocaleSubscriber->onInteractiveLogin($event);
-        $this->assertSame('fr', $session->get('_locale'));
+        $this->assertSame($expectedLocale, $session->get('_locale'));
     }
 
-    private function getEvent(Request $request, User $user)
+    public function userLocaleSubscriberDataProvider(): array
+    {
+        return [
+            [new LocalizedUser('fr'), 'fr'],
+            [new User(), 'en'],
+        ];
+    }
+
+    private function getEvent(Request $request, UserInterface $user): InteractiveLoginEvent
     {
         $token = new UsernamePasswordToken($user, null, 'dev', []);