From 71105702e89300bb85ea42cd88e0310be51a654d Mon Sep 17 00:00:00 2001
From: Mark Paluch <mpaluch@vmware.com>
Date: Thu, 19 May 2022 11:20:55 +0200
Subject: [PATCH] Provide deleted object to `KeyValueTemplate` instead of
 deleting entities by Id.

Closes #438
---
 .../keyvalue/repository/support/SimpleKeyValueRepository.java  | 2 +-
 .../keyvalue/repository/SimpleKeyValueRepositoryUnitTests.java | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/springframework/data/keyvalue/repository/support/SimpleKeyValueRepository.java b/src/main/java/org/springframework/data/keyvalue/repository/support/SimpleKeyValueRepository.java
index 52d90c8..ddf6610 100644
--- a/src/main/java/org/springframework/data/keyvalue/repository/support/SimpleKeyValueRepository.java
+++ b/src/main/java/org/springframework/data/keyvalue/repository/support/SimpleKeyValueRepository.java
@@ -138,7 +138,7 @@ public class SimpleKeyValueRepository<T, ID> implements KeyValueRepository<T, ID
 
 		Assert.notNull(entity, "The given entity must not be null!");
 
-		deleteById(entityInformation.getRequiredId(entity));
+		operations.delete(entity);
 	}
 
 	@Override
diff --git a/src/test/java/org/springframework/data/keyvalue/repository/SimpleKeyValueRepositoryUnitTests.java b/src/test/java/org/springframework/data/keyvalue/repository/SimpleKeyValueRepositoryUnitTests.java
index e732431..61c5b7d 100644
--- a/src/test/java/org/springframework/data/keyvalue/repository/SimpleKeyValueRepositoryUnitTests.java
+++ b/src/test/java/org/springframework/data/keyvalue/repository/SimpleKeyValueRepositoryUnitTests.java
@@ -47,6 +47,7 @@ import org.springframework.data.repository.core.support.PersistentEntityInformat
  * @author Christoph Strobl
  * @author Eugene Nikiforov
  * @author Jens Schauder
+ * @author Mark Paluch
  */
 @ExtendWith(MockitoExtension.class)
 class SimpleKeyValueRepositoryUnitTests {
@@ -108,7 +109,7 @@ class SimpleKeyValueRepositoryUnitTests {
 		repo.save(one);
 		repo.delete(one);
 
-		verify(opsMock, times(1)).delete(eq(one.getId()), eq(Foo.class));
+		verify(opsMock, times(1)).delete(eq(one));
 	}
 
 	@Test // DATACMNS-525