From d1a415519032e3754621ea32cbc382eb80b48f02 Mon Sep 17 00:00:00 2001
From: Marcus Da Coregio <marcusdacoregio@gmail.com>
Date: Wed, 18 Oct 2023 11:32:12 -0300
Subject: [PATCH] Improve mocking on tests

---
 .../data/redis/RedisIndexedSessionRepositoryITests.java     | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/spring-session-data-redis/src/integration-test/java/org/springframework/session/data/redis/RedisIndexedSessionRepositoryITests.java b/spring-session-data-redis/src/integration-test/java/org/springframework/session/data/redis/RedisIndexedSessionRepositoryITests.java
index 089d3204..d7851add 100644
--- a/spring-session-data-redis/src/integration-test/java/org/springframework/session/data/redis/RedisIndexedSessionRepositoryITests.java
+++ b/spring-session-data-redis/src/integration-test/java/org/springframework/session/data/redis/RedisIndexedSessionRepositoryITests.java
@@ -693,9 +693,8 @@ class RedisIndexedSessionRepositoryITests extends AbstractRedisITests {
 	@Test
 	@SuppressWarnings("unchecked")
 	void saveChangeSessionIdWhenFailedRenameOperationExceptionContainsMoreDetailsThenIgnoreError() {
-		RedisOperations<Object, Object> sessionRedisOperations = (RedisOperations<Object, Object>) ReflectionTestUtils
-				.getField(this.repository, "sessionRedisOperations");
-		RedisOperations<Object, Object> spyOperations = spy(sessionRedisOperations);
+		RedisOperations<String, Object> sessionRedisOperations = this.repository.getSessionRedisOperations();
+		RedisOperations<String, Object> spyOperations = spy(sessionRedisOperations);
 		ReflectionTestUtils.setField(this.repository, "sessionRedisOperations", spyOperations);
 
 		RedisSession toSave = this.repository.createSession();
@@ -714,6 +713,7 @@ class RedisIndexedSessionRepositoryITests extends AbstractRedisITests {
 		assertThat(this.repository.findById(sessionId)).isNull();
 		assertThat(this.repository.findById(newSessionId)).isNull();
 		reset(spyOperations);
+		ReflectionTestUtils.setField(this.repository, "sessionRedisOperations", sessionRedisOperations);
 	}
 
 	private String getSecurityName() {