From 0a4d6222e6bf5a6c04de1ed9435cd431fc481360 Mon Sep 17 00:00:00 2001
From: Christoph Strobl <cstrobl@pivotal.io>
Date: Thu, 28 Feb 2019 11:30:49 +0100
Subject: [PATCH] DATAREDIS-888 - Polishing

Original Pull Request: #384
---
 .../lettuce/LettuceClusterConnectionUnitTests.java         | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/test/java/org/springframework/data/redis/connection/lettuce/LettuceClusterConnectionUnitTests.java b/src/test/java/org/springframework/data/redis/connection/lettuce/LettuceClusterConnectionUnitTests.java
index 933909892..d893ac207 100644
--- a/src/test/java/org/springframework/data/redis/connection/lettuce/LettuceClusterConnectionUnitTests.java
+++ b/src/test/java/org/springframework/data/redis/connection/lettuce/LettuceClusterConnectionUnitTests.java
@@ -102,9 +102,10 @@ public class LettuceClusterConnectionUnitTests {
 		partition3.setFlags(Collections.singleton(NodeFlag.MASTER));
 		partition3.setUri(RedisURI.create("redis://" + CLUSTER_HOST + ":" + MASTER_NODE_3_PORT));
 
-		partitions.add(partition1);
-		partitions.add(partition2);
-		partitions.add(partition3);
+		partitions.addPartition(partition1);
+		partitions.addPartition(partition2);
+		partitions.addPartition(partition3);
+		partitions.updateCache();
 
 		when(resourceProvider.getResourceForSpecificNode(CLUSTER_NODE_1)).thenReturn(clusterConnection1Mock);
 		when(resourceProvider.getResourceForSpecificNode(CLUSTER_NODE_2)).thenReturn(clusterConnection2Mock);