Merge pull request #4110 from confluentinc/test-kek

DGS-23351 Add API to test shared KEKs to DEK Registry client

Author: rayokota

dek-registry-client/src/main/java/io/confluent/dekregistry/client/CachedDekRegistryClient.java

@@ -469,6 +469,12 @@ public void undeleteDekVersion(
     dekCache.invalidateAll();
   }
 
+  @Override
+  public void testKek(String name)
+      throws IOException, RestClientException {
+    restService.testKek(DEFAULT_REQUEST_PROPERTIES, name);
+  }
+
   @Override
   public void reset() {
     kekCache.invalidateAll();

dek-registry-client/src/main/java/io/confluent/dekregistry/client/DekRegistryClient.java

@@ -140,6 +140,11 @@ void undeleteDek(String kekName, String subject, DekFormat algorithm)
   void undeleteDekVersion(String kekName, String subject, int version, DekFormat algorithm)
       throws IOException, RestClientException;
 
+  default void testKek(String name)
+      throws IOException, RestClientException {
+    throw new UnsupportedOperationException();
+  }
+
   void reset();
 
   @Override

dek-registry-client/src/main/java/io/confluent/dekregistry/client/rest/DekRegistryRestService.java

@@ -431,4 +431,12 @@ public void undeleteDekVersion(Map<String, String> requestProperties, String nam
 
     httpRequest(path, "POST", null, requestProperties, VOID_TYPE);
   }
+
+  public void testKek(Map<String, String> requestProperties, String name)
+      throws IOException, RestClientException {
+    UriBuilder builder = UriBuilder.fromPath("/dek-registry/v1/keks/{name}/test");
+    String path = builder.build(name).toString();
+
+    httpRequest(path, "POST", null, requestProperties, VOID_TYPE);
+  }
 }

dek-registry/src/test/java/io/confluent/dekregistry/web/rest/RestApiTest.java

@@ -267,6 +267,9 @@ private void testBasic(Map<String, String> headers, boolean isImport) throws Exc
     newKek = client.updateKek(headers, kekName, kmsProps, doc, true);
     assertEquals(kek2, newKek);
 
+    // Test shared kek
+    client.testKek(kekName);
+
     // Advance ticker
     fakeTicker.advance(61, TimeUnit.SECONDS);