cleanup

aerospike · Oct 5, 2023 · bb78571 · bb78571
1 parent 06e3aa3
commit bb78571
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/test/java/org/springframework/data/aerospike/core/AerospikeTemplateDeleteTests.java b/src/test/java/org/springframework/data/aerospike/core/AerospikeTemplateDeleteTests.java
@@ -15,9 +15,9 @@
  */
 package org.springframework.data.aerospike.core;
 
-import com.aerospike.client.AerospikeException;
 import com.aerospike.client.policy.GenerationPolicy;
 import org.junit.jupiter.api.Test;
+import org.springframework.dao.RecoverableDataAccessException;
 import org.springframework.data.aerospike.BaseBlockingIntegrationTests;
 import org.springframework.data.aerospike.SampleClasses.CustomCollectionClassToDelete;
 import org.springframework.data.aerospike.SampleClasses.DocumentWithExpiration;
@@ -192,7 +192,7 @@ public void deleteAll_rejectsDuplicateIds() {
 
             List<String> ids = List.of(id1, id1);
             assertThatThrownBy(() -> template.deleteByIds(ids, DocumentWithExpiration.class))
-                .isInstanceOf(AerospikeException.BatchRecordArray.class)
+                .isInstanceOf(RecoverableDataAccessException.class)
                 .hasMessageContaining("Errors during batch delete");
         }
     }

diff --git a/...ingframework/data/aerospike/query/reactive/ReactiveIndexedPersonRepositoryQueryTests.java b/...ingframework/data/aerospike/query/reactive/ReactiveIndexedPersonRepositoryQueryTests.java
@@ -1,13 +1,13 @@
 package org.springframework.data.aerospike.query.reactive;
 
-import com.aerospike.client.AerospikeException;
 import com.aerospike.client.query.IndexCollectionType;
 import com.aerospike.client.query.IndexType;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.TestInstance;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.dao.RecoverableDataAccessException;
 import org.springframework.data.aerospike.BaseReactiveIntegrationTests;
 import org.springframework.data.aerospike.repository.query.CriteriaDefinition;
 import org.springframework.data.aerospike.sample.Address;
@@ -56,7 +56,7 @@ public void beforeAll() {
         if (ServerVersionUtils.isBatchWriteSupported(reactorClient.getAerospikeClient())) {
             try {
                 reactiveRepository.deleteAll(allIndexedPersons).block();
-            } catch (AerospikeException.BatchRecordArray ignored) {
+            } catch (RecoverableDataAccessException ignored) {
                 // KEY_NOT_FOUND ResultCode causes exception if there are no entities
             }
         } else {
@@ -96,7 +96,7 @@ public void afterAll() {
         if (ServerVersionUtils.isBatchWriteSupported(reactorClient.getAerospikeClient())) {
             try {
                 reactiveRepository.deleteAll(allIndexedPersons).block();
-            } catch (AerospikeException.BatchRecordArray ignored) {
+            } catch (RecoverableDataAccessException ignored) {
                 // KEY_NOT_FOUND ResultCode causes exception if there are no entities
             }
         } else {