Allow dynamic entity reference 1.x or 2.x

modules/salesforce_jwt/tests/src/FunctionalJavascript/SalesforceJwtTest.php

@@ -3,8 +3,6 @@
 namespace Drupal\Tests\salesforce_jwt\FunctionalJavascript;
 
 use Drupal\FunctionalJavascriptTests\WebDriverTestBase;
-use Drupal\key\Entity\Key;
-use Drupal\Tests\key\Functional\KeyTestTrait;
 
 /**
  * Test JWT Auth.
@@ -20,7 +18,12 @@ class SalesforceJwtTest extends WebDriverTestBase {
    */
   protected $defaultTheme  = 'stark';
 
-  use KeyTestTrait;
+  /**
+   * A key entity to use for testing.
+   *
+   * @var \Drupal\key\KeyInterface
+   */
+  protected $testKey;
 
   /**
    * Modules.
@@ -56,7 +59,7 @@ class SalesforceJwtTest extends WebDriverTestBase {
     $this->adminUser = $this->drupalCreateUser(['authorize salesforce']);
     $this->drupalLogin($this->adminUser);
     $this->createTestKey(self::KEY_ID, 'authentication', 'file');
-    Key::load(self::KEY_ID)
+    \Drupal\key\Entity\Key::load(self::KEY_ID)
       ->set('key_provider_settings', [
         'file_location' => __DIR__ . '/testKey.pem',
         'strip_line_breaks' => FALSE,
@@ -104,4 +107,23 @@ class SalesforceJwtTest extends WebDriverTestBase {
     $assert_session->pageTextContainsOnce('Salesforce JWT OAuth');
   }
 
+  /**
+   * Make a key for testing operations that require a key.
+   */
+  protected function createTestKey($id, $type = NULL, $provider = NULL) {
+    $keyArgs = [
+      'id' => $id,
+      'label' => 'Test key',
+    ];
+    if ($type != NULL) {
+      $keyArgs['key_type'] = $type;
+    }
+    if ($provider != NULL) {
+      $keyArgs['key_provider'] = $provider;
+    }
+    $this->testKey = \Drupal\key\Entity\Key::create($keyArgs);
+    $this->testKey->save();
+    return $this->testKey;
+  }
+
 }