Add support for full PRF primitive creation to AesCmacPrfKeyManager.

PiperOrigin-RevId: 554415146

2 files changed, 17 insertions, 0 deletions

diff --git a/java_src/src/test/java/com/google/crypto/tink/prf/AesCmacPrfKeyManagerTest.java b/java_src/src/test/java/com/google/crypto/tink/prf/AesCmacPrfKeyManagerTest.java
index f7331e235..690ca746b 100644
--- a/java_src/src/test/java/com/google/crypto/tink/prf/AesCmacPrfKeyManagerTest.java
+++ b/java_src/src/test/java/com/google/crypto/tink/prf/AesCmacPrfKeyManagerTest.java
@@ -23,10 +23,12 @@ import static org.junit.Assert.assertThrows;
 import com.google.crypto.tink.KeyTemplate;
 import com.google.crypto.tink.KeyTemplates;
 import com.google.crypto.tink.KeysetHandle;
+import com.google.crypto.tink.internal.MutablePrimitiveRegistry;
 import com.google.crypto.tink.proto.AesCmacPrfKey;
 import com.google.crypto.tink.proto.AesCmacPrfKeyFormat;
 import com.google.crypto.tink.subtle.PrfAesCmac;
 import com.google.crypto.tink.subtle.Random;
+import com.google.crypto.tink.util.SecretBytes;
 import com.google.protobuf.ByteString;
 import java.security.GeneralSecurityException;
 import org.junit.Before;
@@ -169,4 +171,16 @@ public class AesCmacPrfKeyManagerTest {
     assertThat(h.getAt(0).getKey().getParameters())
         .isEqualTo(KeyTemplates.get(templateName).toParameters());
   }
+
+  @Test
+  public void registersPrfPrimitiveConstructor() throws Exception {
+    Prf prf =
+        MutablePrimitiveRegistry.globalInstance()
+            .getPrimitive(
+                com.google.crypto.tink.prf.AesCmacPrfKey.create(
+                    AesCmacPrfParameters.create(32), SecretBytes.randomBytes(32)),
+                Prf.class);
+
+    assertThat(prf).isInstanceOf(PrfAesCmac.class);
+  }
 }
diff --git a/java_src/src/test/java/com/google/crypto/tink/prf/BUILD.bazel b/java_src/src/test/java/com/google/crypto/tink/prf/BUILD.bazel
index 368164add..f3e09ef1e 100644
--- a/java_src/src/test/java/com/google/crypto/tink/prf/BUILD.bazel
+++ b/java_src/src/test/java/com/google/crypto/tink/prf/BUILD.bazel
@@ -56,6 +56,8 @@ java_test(
         "//src/main/java/com/google/crypto/tink:key_template",
         "//src/main/java/com/google/crypto/tink:key_templates",
         "//src/main/java/com/google/crypto/tink:registry_cluster",
+        "//src/main/java/com/google/crypto/tink/internal:mutable_primitive_registry",
+        "//src/main/java/com/google/crypto/tink/prf:aes_cmac_prf_key",
         "//src/main/java/com/google/crypto/tink/prf:aes_cmac_prf_key_manager",
         "//src/main/java/com/google/crypto/tink/prf:aes_cmac_prf_parameters",
         "//src/main/java/com/google/crypto/tink/prf:prf_config",
@@ -63,6 +65,7 @@ java_test(
         "//src/main/java/com/google/crypto/tink/subtle:prf_aes_cmac",
         "//src/main/java/com/google/crypto/tink/subtle:random",
         "//src/main/java/com/google/crypto/tink/testing:key_type_manager_test_util",
+        "//src/main/java/com/google/crypto/tink/util:secret_bytes",
         "@maven//:com_google_protobuf_protobuf_java",
         "@maven//:com_google_truth_truth",
         "@maven//:junit_junit",