From d83d40946f2b1e9911637984af0be9f7245a947c Mon Sep 17 00:00:00 2001
From: Mark Paluch <mpaluch@paluch.biz>
Date: Tue, 12 Jul 2022 12:23:41 +0200
Subject: [PATCH] Polishing #2146

---
 .../lettuce/core/cluster/RedisClusterClientIntegrationTests.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/test/java/io/lettuce/core/cluster/RedisClusterClientIntegrationTests.java b/src/test/java/io/lettuce/core/cluster/RedisClusterClientIntegrationTests.java
index e2001b83a9..c1cfc27b63 100644
--- a/src/test/java/io/lettuce/core/cluster/RedisClusterClientIntegrationTests.java
+++ b/src/test/java/io/lettuce/core/cluster/RedisClusterClientIntegrationTests.java
@@ -297,6 +297,7 @@ void testClusterRedirection() {
             for (RedisClusterNode partition : partitions) {
                 partition.setSlots(Collections.emptyList());
                 if (partition.getFlags().contains(RedisClusterNode.NodeFlag.MYSELF)) {
+                    partition.getFlags().add(RedisClusterNode.NodeFlag.UPSTREAM);
                     partition.setSlots(IntStream.range(0, SlotHash.SLOT_COUNT).boxed().collect(Collectors.toList()));
                 }
             }