cache: TestCacheMap: wait grow-shrink goroutine to stop

nutsdb · Jun 13, 2022 · 686510c · 686510c
1 parent 467c5cb
commit 686510c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/leveldb/cache/cache_test.go b/leveldb/cache/cache_test.go
@@ -200,6 +200,7 @@ func TestCacheMap(t *testing.T) {
 	}
 
 	// Emulate constant grow-shrink.
+	growShrinkStop := make(chan bool, 1)
 	go func() {
 		handles := make([]*Handle, 100000)
 		for atomic.LoadInt32(&done) == 0 {
@@ -212,6 +213,7 @@ func TestCacheMap(t *testing.T) {
 				h.Release()
 			}
 		}
+		growShrinkStop <- true
 	}()
 
 	wg.Wait()
@@ -237,6 +239,8 @@ func TestCacheMap(t *testing.T) {
 		}
 	}
 
+	<-growShrinkStop
+
 	require.Zero(t, c.Nodes())
 	require.Zero(t, c.Size())
 	t.Logf("STATS: %#v", c.GetStats())