feat: [tpu] publish boot_disk_config to support CHD with KMS (#9795)

* feat: publish boot_disk_config to support CHD with KMS PiperOrigin-RevId: 561495678 Source-Link: googleapis/googleapis@ae2f515 Source-Link: googleapis/googleapis-gen@c786265 Copy-Tag: eyJwIjoiamF2YS10cHUvLk93bEJvdC55YW1sIiwiaCI6ImM3ODYyNjVhODJjODMwNWQ3ZDk5YTJjZjQ3MTgyYTdiN2Q3NjNhNTIifQ== * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
googleapis · Aug 31, 2023 · 87fc3d3 · 87fc3d3
1 parent d21a74c
commit 87fc3d3
Show file tree

Hide file tree

Showing 10 changed files with 2,624 additions and 289 deletions.
diff --git a/...rc/main/resources/META-INF/native-image/com.google.cloud.tpu.v2alpha1/reflect-config.json b/...rc/main/resources/META-INF/native-image/com.google.cloud.tpu.v2alpha1/reflect-config.json
@@ -539,6 +539,24 @@
     "allDeclaredClasses": true,
     "allPublicClasses": true
   },
+  {
+    "name": "com.google.cloud.tpu.v2alpha1.BootDiskConfig",
+    "queryAllDeclaredConstructors": true,
+    "queryAllPublicConstructors": true,
+    "queryAllDeclaredMethods": true,
+    "allPublicMethods": true,
+    "allDeclaredClasses": true,
+    "allPublicClasses": true
+  },
+  {
+    "name": "com.google.cloud.tpu.v2alpha1.BootDiskConfig$Builder",
+    "queryAllDeclaredConstructors": true,
+    "queryAllPublicConstructors": true,
+    "queryAllDeclaredMethods": true,
+    "allPublicMethods": true,
+    "allDeclaredClasses": true,
+    "allPublicClasses": true
+  },
   {
     "name": "com.google.cloud.tpu.v2alpha1.CreateNodeRequest",
     "queryAllDeclaredConstructors": true,
@@ -575,6 +593,24 @@
     "allDeclaredClasses": true,
     "allPublicClasses": true
   },
+  {
+    "name": "com.google.cloud.tpu.v2alpha1.CustomerEncryptionKey",
+    "queryAllDeclaredConstructors": true,
+    "queryAllPublicConstructors": true,
+    "queryAllDeclaredMethods": true,
+    "allPublicMethods": true,
+    "allDeclaredClasses": true,
+    "allPublicClasses": true
+  },
+  {
+    "name": "com.google.cloud.tpu.v2alpha1.CustomerEncryptionKey$Builder",
+    "queryAllDeclaredConstructors": true,
+    "queryAllPublicConstructors": true,
+    "queryAllDeclaredMethods": true,
+    "allPublicMethods": true,
+    "allDeclaredClasses": true,
+    "allPublicClasses": true
+  },
   {
     "name": "com.google.cloud.tpu.v2alpha1.DeleteNodeRequest",
     "queryAllDeclaredConstructors": true,

diff --git a/java-tpu/google-cloud-tpu/src/test/java/com/google/cloud/tpu/v2alpha1/TpuClientTest.java b/java-tpu/google-cloud-tpu/src/test/java/com/google/cloud/tpu/v2alpha1/TpuClientTest.java
@@ -210,6 +210,7 @@ public void getNodeTest() throws Exception {
             .setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
             .setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
             .setMultisliceNode(true)
+            .setBootDiskConfig(BootDiskConfig.newBuilder().build())
             .build();
     mockTpu.addResponse(expectedResponse);
 
@@ -268,6 +269,7 @@ public void getNodeTest2() throws Exception {
             .setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
             .setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
             .setMultisliceNode(true)
+            .setBootDiskConfig(BootDiskConfig.newBuilder().build())
             .build();
     mockTpu.addResponse(expectedResponse);
 
@@ -326,6 +328,7 @@ public void createNodeTest() throws Exception {
             .setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
             .setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
             .setMultisliceNode(true)
+            .setBootDiskConfig(BootDiskConfig.newBuilder().build())
             .build();
     Operation resultOperation =
         Operation.newBuilder()
@@ -398,6 +401,7 @@ public void createNodeTest2() throws Exception {
             .setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
             .setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
             .setMultisliceNode(true)
+            .setBootDiskConfig(BootDiskConfig.newBuilder().build())
             .build();
     Operation resultOperation =
         Operation.newBuilder()
@@ -554,6 +558,7 @@ public void stopNodeTest() throws Exception {
             .setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
             .setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
             .setMultisliceNode(true)
+            .setBootDiskConfig(BootDiskConfig.newBuilder().build())
             .build();
     Operation resultOperation =
         Operation.newBuilder()
@@ -626,6 +631,7 @@ public void startNodeTest() throws Exception {
             .setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
             .setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
             .setMultisliceNode(true)
+            .setBootDiskConfig(BootDiskConfig.newBuilder().build())
             .build();
     Operation resultOperation =
         Operation.newBuilder()
@@ -698,6 +704,7 @@ public void updateNodeTest() throws Exception {
             .setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
             .setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
             .setMultisliceNode(true)
+            .setBootDiskConfig(BootDiskConfig.newBuilder().build())
             .build();
     Operation resultOperation =
         Operation.newBuilder()
@@ -1698,6 +1705,7 @@ public void simulateMaintenanceEventTest() throws Exception {
             .setAcceleratorConfig(AcceleratorConfig.newBuilder().build())
             .setShieldedInstanceConfig(ShieldedInstanceConfig.newBuilder().build())
             .setMultisliceNode(true)
+            .setBootDiskConfig(BootDiskConfig.newBuilder().build())
             .build();
     Operation resultOperation =
         Operation.newBuilder()