Skip to content

Commit eec9b09

Browse files
committed
Fix Redis prefix test after main merge
1 parent 780d5b3 commit eec9b09

1 file changed

Lines changed: 17 additions & 17 deletions

File tree

kv_cache_manager/config/test/coordination_redis_backend_prefix_test.cc

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,9 @@ StandardUri RedisUriWithParams(const std::string &params) {
1111
return StandardUri::FromUri("redis://localhost:6379/?" + params);
1212
}
1313

14-
CoordinationRedisBackend MakeBackendWithPrefixes(const StandardUri &uri) {
15-
CoordinationRedisBackend backend;
14+
void SetBackendPrefixes(CoordinationRedisBackend &backend, const StandardUri &uri) {
1615
backend.lock_key_prefix_ = CoordinationRedisBackend::BuildRedisKeyPrefix("lock", uri);
1716
backend.kv_key_prefix_ = CoordinationRedisBackend::BuildRedisKeyPrefix("kv", uri);
18-
return backend;
1917
}
2018

2119
} // namespace
@@ -50,37 +48,37 @@ TEST(CoordinationRedisBackendPrefixTest, OtherRedisParamsDoNotAffectClusterScope
5048
}
5149

5250
TEST(CoordinationRedisBackendPrefixTest, GeneratedLockAndKVKeysAreScopedByClusterName) {
53-
CoordinationRedisBackend backend =
54-
MakeBackendWithPrefixes(RedisUriWithParams("cluster_name=test_cluster"));
51+
CoordinationRedisBackend backend;
52+
SetBackendPrefixes(backend, RedisUriWithParams("cluster_name=test_cluster"));
5553

5654
EXPECT_EQ("kvcm_lock:test_cluster:leader", backend.GetRedisLockKey("leader"));
5755
EXPECT_EQ("kvcm_kv:test_cluster:node_endpoint/127.0.0.1:6381",
5856
backend.GetRedisKVKey("node_endpoint/127.0.0.1:6381"));
5957
}
6058

6159
TEST(CoordinationRedisBackendPrefixTest, LockAndKVKeysAreSeparatedWithinSameCluster) {
62-
CoordinationRedisBackend backend =
63-
MakeBackendWithPrefixes(RedisUriWithParams("cluster_name=test_cluster"));
60+
CoordinationRedisBackend backend;
61+
SetBackendPrefixes(backend, RedisUriWithParams("cluster_name=test_cluster"));
6462

6563
EXPECT_NE(backend.GetRedisLockKey("leader"), backend.GetRedisKVKey("leader"));
6664
EXPECT_EQ("kvcm_lock:test_cluster:leader", backend.GetRedisLockKey("leader"));
6765
EXPECT_EQ("kvcm_kv:test_cluster:leader", backend.GetRedisKVKey("leader"));
6866
}
6967

7068
TEST(CoordinationRedisBackendPrefixTest, LogicalKeysAreAppendedWithoutMutation) {
71-
CoordinationRedisBackend backend =
72-
MakeBackendWithPrefixes(RedisUriWithParams("cluster_name=test_cluster"));
69+
CoordinationRedisBackend backend;
70+
SetBackendPrefixes(backend, RedisUriWithParams("cluster_name=test_cluster"));
7371

7472
EXPECT_EQ("kvcm_lock:test_cluster:model/a/b:0", backend.GetRedisLockKey("model/a/b:0"));
7573
EXPECT_EQ("kvcm_kv:test_cluster:node_endpoint/10.0.0.1:6381",
7674
backend.GetRedisKVKey("node_endpoint/10.0.0.1:6381"));
7775
}
7876

7977
TEST(CoordinationRedisBackendPrefixTest, DifferentClustersGenerateDifferentRedisKeysForSameLogicalKey) {
80-
CoordinationRedisBackend first =
81-
MakeBackendWithPrefixes(RedisUriWithParams("cluster_name=cluster_a"));
82-
CoordinationRedisBackend second =
83-
MakeBackendWithPrefixes(RedisUriWithParams("cluster_name=cluster_b"));
78+
CoordinationRedisBackend first;
79+
CoordinationRedisBackend second;
80+
SetBackendPrefixes(first, RedisUriWithParams("cluster_name=cluster_a"));
81+
SetBackendPrefixes(second, RedisUriWithParams("cluster_name=cluster_b"));
8482

8583
EXPECT_NE(first.GetRedisLockKey("leader"), second.GetRedisLockKey("leader"));
8684
EXPECT_NE(first.GetRedisKVKey("node_endpoint"), second.GetRedisKVKey("node_endpoint"));
@@ -93,16 +91,18 @@ TEST(CoordinationRedisBackendPrefixTest, SameClusterNameGeneratesSameScopeAcross
9391
"redis://user:pwd@redis-a:6379/?timeout_ms=1000&cluster_name=test_cluster");
9492
StandardUri second_uri = StandardUri::FromUri(
9593
"redis://redis-b:6380/?db=3&retry_count=5&cluster_name=test_cluster");
96-
CoordinationRedisBackend first = MakeBackendWithPrefixes(first_uri);
97-
CoordinationRedisBackend second = MakeBackendWithPrefixes(second_uri);
94+
CoordinationRedisBackend first;
95+
CoordinationRedisBackend second;
96+
SetBackendPrefixes(first, first_uri);
97+
SetBackendPrefixes(second, second_uri);
9898

9999
EXPECT_EQ(first.GetRedisLockKey("leader"), second.GetRedisLockKey("leader"));
100100
EXPECT_EQ(first.GetRedisKVKey("node_endpoint"), second.GetRedisKVKey("node_endpoint"));
101101
}
102102

103103
TEST(CoordinationRedisBackendPrefixTest, LegacyGeneratedKeysStayBackwardCompatible) {
104-
CoordinationRedisBackend backend =
105-
MakeBackendWithPrefixes(StandardUri::FromUri("redis://localhost:6379/"));
104+
CoordinationRedisBackend backend;
105+
SetBackendPrefixes(backend, StandardUri::FromUri("redis://localhost:6379/"));
106106

107107
EXPECT_EQ("kvcm_lock:leader", backend.GetRedisLockKey("leader"));
108108
EXPECT_EQ("kvcm_kv:node_endpoint", backend.GetRedisKVKey("node_endpoint"));

0 commit comments

Comments
 (0)