From ee04a409d9284a65bed92154caa239a9eecdd26c Mon Sep 17 00:00:00 2001
From: Matteo Merli <mmerli@apache.org>
Date: Wed, 9 Oct 2024 16:19:10 -0700
Subject: [PATCH] Added deletion test

---
 .../oxia/client/it/OxiaClientIT.java          | 64 ++++++++++---------
 1 file changed, 33 insertions(+), 31 deletions(-)

diff --git a/client-it/src/test/java/io/streamnative/oxia/client/it/OxiaClientIT.java b/client-it/src/test/java/io/streamnative/oxia/client/it/OxiaClientIT.java
index 07df7f8d..20c633d6 100644
--- a/client-it/src/test/java/io/streamnative/oxia/client/it/OxiaClientIT.java
+++ b/client-it/src/test/java/io/streamnative/oxia/client/it/OxiaClientIT.java
@@ -625,47 +625,49 @@ void testSecondaryIndex() throws Exception {
         @Cleanup
         SyncOxiaClient client = OxiaClientBuilder.create(oxia.getServiceAddress()).syncClient();
 
-        client.put(
-                "si-a",
-                "0".getBytes(),
-                Set.of(PutOption.PartitionKey("x"), PutOption.SecondaryIndex("val", "0")));
-        client.put(
-                "si-b",
-                "1".getBytes(),
-                Set.of(PutOption.PartitionKey("x"), PutOption.SecondaryIndex("val", "1")));
-        client.put(
-                "si-c",
-                "2".getBytes(),
-                Set.of(PutOption.PartitionKey("x"), PutOption.SecondaryIndex("val", "2")));
-        client.put(
-                "si-d",
-                "3".getBytes(),
-                Set.of(PutOption.PartitionKey("x"), PutOption.SecondaryIndex("val", "3")));
-        client.put(
-                "si-e",
-                "4".getBytes(),
-                Set.of(PutOption.PartitionKey("x"), PutOption.SecondaryIndex("val", "4")));
-        client.put(
-                "si-f",
-                "5".getBytes(),
-                Set.of(PutOption.PartitionKey("x"), PutOption.SecondaryIndex("val", "5")));
-        client.put(
-                "si-g",
-                "6".getBytes(),
-                Set.of(PutOption.PartitionKey("x"), PutOption.SecondaryIndex("val", "6")));
+        client.put("si-a", "0".getBytes(),
+                Set.of(PutOption.SecondaryIndex("val", "0")));
+        client.put("si-b", "1".getBytes(),
+                Set.of(PutOption.SecondaryIndex("val", "1")));
+        client.put("si-c", "2".getBytes(),
+                Set.of(PutOption.SecondaryIndex("val", "2")));
+        client.put("si-d", "3".getBytes(),
+                Set.of(PutOption.SecondaryIndex("val", "3")));
+        client.put("si-e", "4".getBytes(),
+                Set.of(PutOption.SecondaryIndex("val", "4")));
+        client.put("si-f", "5".getBytes(),
+                Set.of(PutOption.SecondaryIndex("val", "5")));
+        client.put("si-g", "6".getBytes(),
+                Set.of(PutOption.SecondaryIndex("val", "6")));
 
         List<String> list =
-                client.list("1", "4", Set.of(ListOption.PartitionKey("x"), ListOption.UseIndex("val")));
-        assertThat(list).isEqualTo(List.of("si-b", "si-c", "si-d"));
+                client.list("1", "4", Set.of(ListOption.UseIndex("val")));
+        assertThat(list).containsExactly("si-b", "si-c", "si-d");
 
         Iterable<GetResult> iterable =
                 client.rangeScan(
-                        "2", "5", Set.of(RangeScanOption.PartitionKey("x"), RangeScanOption.UseIndex("val")));
+                        "2", "5", Set.of(RangeScanOption.UseIndex("val")));
         list =
                 StreamSupport.stream(iterable.spliterator(), false)
                         .map(GetResult::getKey)
                         .sorted()
                         .toList();
         assertThat(list).containsExactly("si-c", "si-d", "si-e");
+
+        // Deletion
+        client.delete("si-b");
+
+        list = client.list("0", "3", Set.of(ListOption.UseIndex("val")));
+        assertThat(list).containsExactly("si-a", "si-c");
+
+        iterable =
+                client.rangeScan(
+                        "0", "3", Set.of(RangeScanOption.UseIndex("val")));
+        list =
+                StreamSupport.stream(iterable.spliterator(), false)
+                        .map(GetResult::getKey)
+                        .sorted()
+                        .toList();
+        assertThat(list).containsExactly("si-a", "si-c");
     }
 }