|
8 | 8 | import com.swirlds.state.test.fixtures.merkle.MerkleTestBase;
|
9 | 9 | import java.util.Spliterators;
|
10 | 10 | import java.util.stream.StreamSupport;
|
| 11 | + |
| 12 | +import org.junit.jupiter.api.AfterEach; |
11 | 13 | import org.junit.jupiter.api.BeforeEach;
|
12 | 14 | import org.junit.jupiter.api.DisplayName;
|
13 | 15 | import org.junit.jupiter.api.Nested;
|
@@ -45,6 +47,13 @@ void stateKey() {
|
45 | 47 | final var state = new OnDiskWritableKVState<>(FRUIT_STATE_KEY, STRING_CODEC, STRING_CODEC, fruitVirtualMap);
|
46 | 48 | assertThat(state.getStateKey()).isEqualTo(FRUIT_STATE_KEY);
|
47 | 49 | }
|
| 50 | + |
| 51 | + @AfterEach |
| 52 | + void tearDown() { |
| 53 | + if (fruitVirtualMap != null && fruitVirtualMap.getReservationCount() > -1) { |
| 54 | + fruitVirtualMap.release(); |
| 55 | + } |
| 56 | + } |
48 | 57 | }
|
49 | 58 |
|
50 | 59 | private void add(String key, String value) {
|
@@ -87,6 +96,14 @@ void iterateIncludesMutations() {
|
87 | 96 | .toList();
|
88 | 97 | assertThat(actual).containsExactlyInAnyOrder(A_KEY, B_KEY, C_KEY);
|
89 | 98 | }
|
| 99 | + |
| 100 | + @AfterEach |
| 101 | + void tearDown() { |
| 102 | + if (fruitVirtualMap != null && fruitVirtualMap.getReservationCount() > -1) { |
| 103 | + fruitVirtualMap.release(); |
| 104 | + } |
| 105 | + } |
| 106 | + |
90 | 107 | }
|
91 | 108 |
|
92 | 109 | @Nested
|
@@ -245,5 +262,12 @@ void smorgasbord() throws ParseException {
|
245 | 262 | assertThat(readValueFromMerkleMap(D_KEY)).isEqualTo(DATE);
|
246 | 263 | assertThat(readValueFromMerkleMap(E_KEY)).isEqualTo(ELDERBERRY);
|
247 | 264 | }
|
| 265 | + |
| 266 | + @AfterEach |
| 267 | + void tearDown() { |
| 268 | + if (fruitVirtualMap != null && fruitVirtualMap.getReservationCount() > -1) { |
| 269 | + fruitVirtualMap.release(); |
| 270 | + } |
| 271 | + } |
248 | 272 | }
|
249 | 273 | }
|
0 commit comments