diff --git a/tests/src/Functional/MasqueradeAccessTest.php b/tests/src/Functional/MasqueradeAccessTest.php index 95825cc..102f570 100644 --- a/tests/src/Functional/MasqueradeAccessTest.php +++ b/tests/src/Functional/MasqueradeAccessTest.php @@ -2,8 +2,6 @@ namespace Drupal\Tests\masquerade\Functional; -use Drupal\user\Entity\User; - /** * Tests masquerade access mechanism. * @@ -37,11 +35,12 @@ class MasqueradeAccessTest extends MasqueradeWebTestBase { * - auth ! * */ public function testAccess() { - $this->root_user = User::load(1); + $this->drupalLogin($this->rootUser); + $this->assertCanMasqueradeAs($this->admin_user); $this->drupalLogin($this->admin_user); // Permission 'masquerade as super user' granted by default. - $this->assertCanMasqueradeAs($this->root_user); + $this->assertCanMasqueradeAs($this->rootUser); // Permission 'masquerade as any user' granted by default. $this->assertCanMasqueradeAs($this->moderator_user); $this->assertCanMasqueradeAs($this->editor_user); @@ -50,7 +49,7 @@ class MasqueradeAccessTest extends MasqueradeWebTestBase { // Test 'masquerade as any user' permission except UID 1. $this->drupalLogin($this->moderator_user); - $this->assertCanNotMasqueradeAs($this->root_user); + $this->assertCanNotMasqueradeAs($this->rootUser); $this->assertCanMasqueradeAs($this->admin_user); $this->assertCanMasqueradeAs($this->editor_user); $this->assertCanMasqueradeAs($this->masquerade_user); @@ -58,7 +57,7 @@ class MasqueradeAccessTest extends MasqueradeWebTestBase { // Test 'masquerade as @role' permission. $this->drupalLogin($this->editor_user); - $this->assertCanNotMasqueradeAs($this->root_user); + $this->assertCanNotMasqueradeAs($this->rootUser); $this->assertCanNotMasqueradeAs($this->admin_user); $this->assertCanNotMasqueradeAs($this->moderator_user); $this->assertCanMasqueradeAs($this->masquerade_user); @@ -66,7 +65,7 @@ class MasqueradeAccessTest extends MasqueradeWebTestBase { // Test 'masquerade as authenticated' permission. $this->drupalLogin($this->masquerade_user); - $this->assertCanNotMasqueradeAs($this->root_user); + $this->assertCanNotMasqueradeAs($this->rootUser); $this->assertCanNotMasqueradeAs($this->admin_user); $this->assertCanNotMasqueradeAs($this->moderator_user); $this->assertCanNotMasqueradeAs($this->editor_user);