Add support for deletion protection

src/integration/java/io/pinecone/helpers/TestResourcesManager.java

@@ -317,7 +317,7 @@ public String getOrCreateServerlessIndex() throws InterruptedException, Pinecone
         String indexName = RandomStringBuilder.build("serverless-index", 8);
 
         serverlessIndexModel = pineconeClient.createServerlessIndex(indexName, metric, dimension, cloud,
-                region);
+                region, DeletionProtection.DISABLED);
         waitUntilIndexIsReady(pineconeClient, indexName);
 
         // Explicitly wait after ready to avoid the "no healthy upstream" issue

src/integration/java/io/pinecone/integration/controlPlane/pod/ConfigureIndexTest.java

@@ -8,6 +8,7 @@
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
+import org.openapitools.control.client.model.DeletionProtection;
 import org.openapitools.control.client.model.IndexModel;
 import org.openapitools.control.client.model.IndexModelStatus;
 import org.openapitools.control.client.model.PodSpec;
@@ -55,7 +56,7 @@ public void afterEach() throws InterruptedException {
     @Test
     public void configureIndexWithInvalidIndexName() {
         try {
-            controlPlaneClient.configureIndex("non-existent-index", 3);
+            controlPlaneClient.configurePodsIndex("non-existent-index", 3, DeletionProtection.DISABLED);
 
             fail("Expected to throw PineconeNotFoundException");
         } catch (PineconeNotFoundException expected) {
@@ -66,7 +67,7 @@ public void configureIndexWithInvalidIndexName() {
     @Test
     public void configureIndexExceedingQuota() {
         try {
-            controlPlaneClient.configureIndex(indexName, 600);
+            controlPlaneClient.configurePodsIndex(indexName, 30, DeletionProtection.DISABLED);
             fail("Expected to throw PineconeForbiddenException");
         } catch (PineconeForbiddenException expected) {
             assertTrue(expected.getLocalizedMessage().contains("reached the max pods allowed"));
@@ -81,7 +82,7 @@ public void scaleUpAndDown() throws InterruptedException {
 
         // Verify the scaled up replicas
         assertWithRetry(() -> {
-            controlPlaneClient.configureIndex(indexName, 3);
+            controlPlaneClient.configurePodsIndex(indexName, 3, DeletionProtection.DISABLED);
             PodSpec podSpec = controlPlaneClient.describeIndex(indexName).getSpec().getPod();
             assertNotNull(podSpec);
             assertEquals(podSpec.getReplicas(), 3);
@@ -91,7 +92,7 @@ public void scaleUpAndDown() throws InterruptedException {
 
         // Verify replicas were scaled down
         assertWithRetry(() -> {
-            controlPlaneClient.configureIndex(indexName, 1);
+            controlPlaneClient.configurePodsIndex(indexName, 1, DeletionProtection.DISABLED);
             PodSpec podSpec = controlPlaneClient.describeIndex(indexName).getSpec().getPod();
             assertNotNull(podSpec);
             assertEquals(podSpec.getReplicas(), 1);
@@ -107,7 +108,7 @@ public void changingBasePodType() throws InterruptedException {
             assertEquals(1, indexModel.getSpec().getPod().getReplicas());
 
             // Try to change the base pod type
-            controlPlaneClient.configureIndex(indexName, "p2.x2");
+            controlPlaneClient.configurePodsIndex(indexName, "p2.x2");
 
             fail("Expected to throw PineconeBadRequestException");
         } catch (PineconeBadRequestException expected) {
@@ -125,7 +126,7 @@ public void sizeIncrease() throws InterruptedException {
         // Change the pod type to a larger one
         // Get the index description to verify the new pod type
         assertWithRetry(() -> {
-            controlPlaneClient.configureIndex(indexName, "p1.x2");
+            controlPlaneClient.configurePodsIndex(indexName, "p1.x2");
             PodSpec podSpec = controlPlaneClient.describeIndex(indexName).getSpec().getPod();
             assertNotNull(podSpec);
             assertEquals(podSpec.getPodType(), "p1.x2");

src/integration/java/io/pinecone/integration/controlPlane/pod/DeletionProtectionTest.java

@@ -0,0 +1,48 @@
+package io.pinecone.integration.controlPlane.pod;
+
+import io.pinecone.clients.Pinecone;
+import io.pinecone.helpers.RandomStringBuilder;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.openapitools.control.client.model.DeletionProtection;
+import org.openapitools.control.client.model.IndexModel;
+
+public class DeletionProtectionTest {
+    private static final Pinecone controlPlaneClient = new Pinecone
+            .Builder(System.getenv("PINECONE_API_KEY"))
+            .withSourceTag("pinecone_test")
+            .build();
+
+    @Test
+    public void createPodIndexWithDeletionProtectionEnabled() {
+        String indexName = RandomStringBuilder.build("create-pod", 8);
+        // Create pod index with deletion protection enabled
+        controlPlaneClient.createPodsIndex(indexName, 3, "us-east-1-aws", "p1.x1", DeletionProtection.ENABLED);
+        IndexModel indexModel = controlPlaneClient.describeIndex(indexName);
+        DeletionProtection deletionProtection = indexModel.getDeletionProtection();
+        Assertions.assertEquals(deletionProtection, DeletionProtection.ENABLED);
+        // Configure index to disable deletionProtection
+        controlPlaneClient.configurePodsIndex(indexName, DeletionProtection.DISABLED);
+        // Delete index
+        controlPlaneClient.deleteIndex(indexName);
+    }
+
+    @Test
+    public void createPodIndexWithDeletionProtectionDisabled() {
+        String indexName = RandomStringBuilder.build("create-pod", 8);
+        // Create pod index with deletion protection disabled
+        controlPlaneClient.createPodsIndex(indexName, 3, "us-east-1-aws", "p1.x1");
+        IndexModel indexModel = controlPlaneClient.describeIndex(indexName);
+        DeletionProtection deletionProtection = indexModel.getDeletionProtection();
+        Assertions.assertEquals(deletionProtection, DeletionProtection.DISABLED);
+        // Configure index to enable deletionProtection
+        controlPlaneClient.configurePodsIndex(indexName, DeletionProtection.ENABLED);
+        indexModel = controlPlaneClient.describeIndex(indexName);
+        deletionProtection = indexModel.getDeletionProtection();
+        Assertions.assertEquals(deletionProtection, DeletionProtection.ENABLED);
+        // Configure index to disable deletionProtection
+        controlPlaneClient.configurePodsIndex(indexName, DeletionProtection.DISABLED);
+        // Delete index
+        controlPlaneClient.deleteIndex(indexName);
+    }
+}

src/integration/java/io/pinecone/integration/controlPlane/serverless/CreateDescribeListAndDeleteIndexTest.java

@@ -48,7 +48,7 @@ public void describeAndListIndex() {
     @Test
     public void createServerlessIndexWithInvalidName() {
         try {
-            controlPlaneClient.createServerlessIndex("Invalid-name", "cosine", 3, "aws", "us-west-2");
+            controlPlaneClient.createServerlessIndex("Invalid-name", "cosine", 3, "aws", "us-west-2", DeletionProtection.DISABLED);
 
             fail("Expected to throw PineconeBadRequestException");
         } catch (PineconeBadRequestException expected) {
@@ -59,7 +59,7 @@ public void createServerlessIndexWithInvalidName() {
     @Test
     public void createServerlessIndexWithInvalidDimension() {
         try {
-            controlPlaneClient.createServerlessIndex("serverless-test-index", "cosine", -3, "aws", "us-west-2");
+            controlPlaneClient.createServerlessIndex("serverless-test-index", "cosine", -3, "aws", "us-west-2", DeletionProtection.DISABLED);
             fail("Expected to throw PineconeValidationException");
         } catch (PineconeValidationException expected) {
             assertTrue(expected.getLocalizedMessage().contains("Dimension must be greater than 0"));
@@ -69,7 +69,7 @@ public void createServerlessIndexWithInvalidDimension() {
     @Test
     public void createServerlessIndexWithInvalidCloud() {
         try {
-            controlPlaneClient.createServerlessIndex("serverless-test-index", "cosine", 3, "blah", "us-west-2");
+            controlPlaneClient.createServerlessIndex("serverless-test-index", "cosine", 3, "blah", "us-west-2", DeletionProtection.DISABLED);
             fail("Expected to throw PineconeValidationException");
         } catch (PineconeValidationException expected) {
             assertTrue(expected.getLocalizedMessage().contains("Cloud cannot be null or empty. Must be one of " + Arrays.toString(ServerlessSpec.CloudEnum.values())));
@@ -79,7 +79,7 @@ public void createServerlessIndexWithInvalidCloud() {
     @Test
     public void createServerlessIndexWithInvalidRegion() {
         try {
-            controlPlaneClient.createServerlessIndex("serverless-test-index", "cosine", 3, "aws", "invalid-region");
+            controlPlaneClient.createServerlessIndex("serverless-test-index", "cosine", 3, "aws", "invalid-region", DeletionProtection.DISABLED);
             fail("Expected to throw PineconeNotFoundException");
         } catch (PineconeNotFoundException expected) {
             assertTrue(expected.getLocalizedMessage().contains("Resource cloud: aws region: invalid-region not found"));

src/integration/java/io/pinecone/integration/controlPlane/serverless/DeletionProtectionTest.java

@@ -0,0 +1,45 @@
+package io.pinecone.integration.controlPlane.serverless;
+
+import io.pinecone.clients.Pinecone;
+import io.pinecone.helpers.RandomStringBuilder;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.openapitools.control.client.model.DeletionProtection;
+import org.openapitools.control.client.model.IndexModel;
+
+public class DeletionProtectionTest {
+    private static final Pinecone controlPlaneClient = new Pinecone
+            .Builder(System.getenv("PINECONE_API_KEY"))
+            .withSourceTag("pinecone_test")
+            .build();
+
+    @Test
+    public void createIndexWithDeletionProtectionEnabled() {
+        String indexName = RandomStringBuilder.build("create-serv", 8);
+        // Create serverless index with deletion protection enabled
+        controlPlaneClient.createServerlessIndex(indexName, "cosine", 3, "aws", "us-west-2", DeletionProtection.ENABLED);
+        // Describe index to verify deletion protection is enabled
+        IndexModel indexModel = controlPlaneClient.describeIndex(indexName);
+        DeletionProtection deletionProtection = indexModel.getDeletionProtection();
+        Assertions.assertEquals(deletionProtection, DeletionProtection.ENABLED);
+    }
+
+    @Test
+    public void createPodIndexWithDeletionProtectionDisabled() {
+        String indexName = RandomStringBuilder.build("create-pod", 8);
+        // Create serverless index with deletion protection disabled
+        controlPlaneClient.createServerlessIndex(indexName, "cosine", 3, "aws", "us-west-2", DeletionProtection.DISABLED);
+        IndexModel indexModel = controlPlaneClient.describeIndex(indexName);
+        DeletionProtection deletionProtection = indexModel.getDeletionProtection();
+        Assertions.assertEquals(deletionProtection, DeletionProtection.DISABLED);
+        // Configure index to enable deletionProtection
+        controlPlaneClient.configureServerlessIndex(indexName, DeletionProtection.ENABLED);
+        indexModel = controlPlaneClient.describeIndex(indexName);
+        deletionProtection = indexModel.getDeletionProtection();
+        Assertions.assertEquals(deletionProtection, DeletionProtection.ENABLED);
+        // Configure index to disable deletionProtection
+        controlPlaneClient.configureServerlessIndex(indexName, DeletionProtection.DISABLED);
+        // Delete index
+        controlPlaneClient.deleteIndex(indexName);
+    }
+}