Removing final RegistryTest.java dependency on deprecated class AeadK…

…eyTemplates. PiperOrigin-RevId: 340475813
DanielSel · Nov 3, 2020 · e474a99 · e474a99
1 parent bbff574
commit e474a99
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/java_src/src/test/java/com/google/crypto/tink/BUILD.bazel b/java_src/src/test/java/com/google/crypto/tink/BUILD.bazel
@@ -218,7 +218,7 @@ java_test(
         "//src/main/java/com/google/crypto/tink:public_key_verify",
         "//src/main/java/com/google/crypto/tink:registry_cluster",
         "//src/main/java/com/google/crypto/tink/aead:aead_config",
-        "//src/main/java/com/google/crypto/tink/aead:aead_key_templates",
+        "//src/main/java/com/google/crypto/tink/aead:aes_ctr_hmac_aead_key_manager",
         "//src/main/java/com/google/crypto/tink/aead:aes_eax_key_manager",
         "//src/main/java/com/google/crypto/tink/config:tink_config",
         "//src/main/java/com/google/crypto/tink/mac:mac_config",

diff --git a/java_src/src/test/java/com/google/crypto/tink/RegistryTest.java b/java_src/src/test/java/com/google/crypto/tink/RegistryTest.java
@@ -23,7 +23,7 @@
 import static org.junit.Assert.fail;
 
 import com.google.crypto.tink.aead.AeadConfig;
-import com.google.crypto.tink.aead.AeadKeyTemplates;
+import com.google.crypto.tink.aead.AesCtrHmacAeadKeyManager;
 import com.google.crypto.tink.aead.AesEaxKeyManager;
 import com.google.crypto.tink.config.TinkConfig;
 import com.google.crypto.tink.mac.MacConfig;
@@ -447,9 +447,8 @@ public void testGetPrimitive_Hmac_shouldWork() throws Exception {
   @Test
   public void testGetPrimitives_CustomManager_shouldWork() throws Exception {
     // Create a keyset.
-    KeyTemplate template2 = AeadKeyTemplates.AES128_CTR_HMAC_SHA256;
     KeyData key1 = Registry.newKeyData(AesEaxKeyManager.aes128EaxTemplate());
-    KeyData key2 = Registry.newKeyData(template2);
+    KeyData key2 = Registry.newKeyData(AesCtrHmacAeadKeyManager.aes128CtrHmacSha256Template());
     KeysetHandle keysetHandle =
         KeysetHandle.fromKeyset(
             Keyset.newBuilder()