Skip to content

Commit 8d673f3

Browse files
committed
Added missing tearDown methods.
Signed-off-by: Ivan Malygin <[email protected]>
1 parent 86ca1bd commit 8d673f3

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

platform-sdk/swirlds-state-impl/src/test/java/com/swirlds/state/merkle/disk/OnDiskWritableStateTest.java

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@
88
import com.swirlds.state.test.fixtures.merkle.MerkleTestBase;
99
import java.util.Spliterators;
1010
import java.util.stream.StreamSupport;
11+
12+
import org.junit.jupiter.api.AfterEach;
1113
import org.junit.jupiter.api.BeforeEach;
1214
import org.junit.jupiter.api.DisplayName;
1315
import org.junit.jupiter.api.Nested;
@@ -45,6 +47,13 @@ void stateKey() {
4547
final var state = new OnDiskWritableKVState<>(FRUIT_STATE_KEY, STRING_CODEC, STRING_CODEC, fruitVirtualMap);
4648
assertThat(state.getStateKey()).isEqualTo(FRUIT_STATE_KEY);
4749
}
50+
51+
@AfterEach
52+
void tearDown() {
53+
if (fruitVirtualMap != null && fruitVirtualMap.getReservationCount() > -1) {
54+
fruitVirtualMap.release();
55+
}
56+
}
4857
}
4958

5059
private void add(String key, String value) {
@@ -87,6 +96,14 @@ void iterateIncludesMutations() {
8796
.toList();
8897
assertThat(actual).containsExactlyInAnyOrder(A_KEY, B_KEY, C_KEY);
8998
}
99+
100+
@AfterEach
101+
void tearDown() {
102+
if (fruitVirtualMap != null && fruitVirtualMap.getReservationCount() > -1) {
103+
fruitVirtualMap.release();
104+
}
105+
}
106+
90107
}
91108

92109
@Nested
@@ -245,5 +262,12 @@ void smorgasbord() throws ParseException {
245262
assertThat(readValueFromMerkleMap(D_KEY)).isEqualTo(DATE);
246263
assertThat(readValueFromMerkleMap(E_KEY)).isEqualTo(ELDERBERRY);
247264
}
265+
266+
@AfterEach
267+
void tearDown() {
268+
if (fruitVirtualMap != null && fruitVirtualMap.getReservationCount() > -1) {
269+
fruitVirtualMap.release();
270+
}
271+
}
248272
}
249273
}

0 commit comments

Comments
 (0)