Skip to content

Commit fbb381c

Browse files
committed
fix shared cache tests
1 parent bebab98 commit fbb381c

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

ydb/core/tablet_flat/ut/ut_shared_sausagecache.cpp

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ void RestartAndClearCache(TMyEnvBase& env) {
128128
env->GetMemObserver()->NotifyStat({200*MB, 100*MB, 100*MB});
129129
WaitEvent(env, NSharedCache::EvMem);
130130

131-
env->GetMemObserver()->NotifyStat({100*MB, 200*MB, 200*MB});
131+
env->GetMemObserver()->NotifyStat({100*MB, 108*MB, 108*MB});
132132
WaitEvent(env, NSharedCache::EvMem);
133133

134134
env.FireDummyTablet(ui32(NFake::TDummy::EFlg::Comp));
@@ -137,7 +137,7 @@ void RestartAndClearCache(TMyEnvBase& env) {
137137
void SwitchPolicy(TMyEnvBase& env, NKikimrSharedCache::TReplacementPolicy policy) {
138138
auto configure = MakeHolder<TEvSharedPageCache::TEvConfigure>();
139139
configure->Record.SetReplacementPolicy(policy);
140-
configure->Record.SetMemoryLimit(8_MB);
140+
configure->Record.SetMemoryLimit(0); // no limit
141141
env->Send(MakeSharedPageCacheId(), TActorId{}, configure.Release());
142142
WaitEvent(env, TEvSharedPageCache::EvConfigure);
143143
}
@@ -152,6 +152,11 @@ Y_UNIT_TEST(Limits) {
152152
env.FireDummyTablet(ui32(NFake::TDummy::EFlg::Comp));
153153
env.SendSync(new NFake::TEvExecute{ new TTxInitSchema() });
154154

155+
auto configure = MakeHolder<TEvSharedPageCache::TEvConfigure>();
156+
configure->Record.SetMemoryLimit(8_MB);
157+
env->Send(MakeSharedPageCacheId(), TActorId{}, configure.Release());
158+
WaitEvent(env, TEvSharedPageCache::EvConfigure);
159+
155160
SwitchPolicy(env, NKikimrSharedCache::ThreeLeveledLRU);
156161

157162
// write 300 rows, each ~100KB (~30MB)
@@ -320,6 +325,9 @@ Y_UNIT_TEST(ThreeLeveledLRU) {
320325
SwitchPolicy(env, NKikimrSharedCache::ThreeLeveledLRU);
321326
env->GetAppData().FeatureFlags.SetEnableLocalDBBtreeIndex(true);
322327

328+
env->GetMemObserver()->NotifyStat({100*MB, 108*MB, 108*MB});
329+
WaitEvent(env, NSharedCache::EvMem);
330+
323331
// write 100 rows, each ~100KB (~10MB)
324332
for (i64 key = 0; key < 100; ++key) {
325333
TString value(size_t(100 * 1024), char('a' + key % 26));
@@ -419,6 +427,9 @@ Y_UNIT_TEST(S3FIFO) {
419427
SwitchPolicy(env, NKikimrSharedCache::S3FIFO);
420428
env->GetAppData().FeatureFlags.SetEnableLocalDBBtreeIndex(true);
421429

430+
env->GetMemObserver()->NotifyStat({100*MB, 108*MB, 108*MB});
431+
WaitEvent(env, NSharedCache::EvMem);
432+
422433
// write 100 rows, each ~100KB (~10MB)
423434
for (i64 key = 0; key < 100; ++key) {
424435
TString value(size_t(100 * 1024), char('a' + key % 26));
@@ -518,6 +529,9 @@ Y_UNIT_TEST(ClockPro) {
518529
SwitchPolicy(env, NKikimrSharedCache::ClockPro);
519530
env->GetAppData().FeatureFlags.SetEnableLocalDBBtreeIndex(true);
520531

532+
env->GetMemObserver()->NotifyStat({100*MB, 108*MB, 108*MB});
533+
WaitEvent(env, NSharedCache::EvMem);
534+
521535
// write 100 rows, each ~100KB (~10MB)
522536
for (i64 key = 0; key < 100; ++key) {
523537
TString value(size_t(100 * 1024), char('a' + key % 26));

0 commit comments

Comments
 (0)