Replace \Drupal with $this->container in tests

diff --git a/tests/src/Kernel/QueryAccess/QueryAccessTest.php b/tests/src/Kernel/QueryAccess/QueryAccessTest.php
index 4302786..a8e7999 100644
--- a/tests/src/Kernel/QueryAccess/QueryAccessTest.php
+++ b/tests/src/Kernel/QueryAccess/QueryAccessTest.php
@@ -50,6 +50,7 @@ class QueryAccessTest extends EntityKernelTestBase {
    */
   protected function setUp() {
     parent::setUp();
+    $this->setUpCurrentUser(['uid' => 1]);
 
     $this->installEntitySchema('entity_test_enhanced');
     $this->installConfig(['entity_module_test']);
@@ -93,17 +94,13 @@ class QueryAccessTest extends EntityKernelTestBase {
     $third_entity->save();
 
     $this->entities = [$first_entity, $second_entity, $third_entity];
-    $this->storage = \Drupal::entityTypeManager()->getStorage('entity_test_enhanced');
+    $this->storage = $this->entityTypeManager->getStorage('entity_test_enhanced');
   }
 
   /**
    * Tests EntityQuery filtering.
    */
   public function testEntityQuery() {
-    // Admin permission, full access.
-    $admin_user = $this->createUser([], ['administer entity_test_enhanced']);
-    \Drupal::currentUser()->setAccount($admin_user);

This code is not equivalent. We were testing a user with the "administer entity_test_enhanced" permission, but now we are using uid 1 which satisfies any permission. That means that if there's a bug with fetching the admin permission from the entity type, it will no longer get caught.

Thanks, everyone!