Skip to content

Commit ecbea97

Browse files
committed
restore the original value
1 parent d11814c commit ecbea97

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

framework/src/test/java/org/tron/core/exception/TronErrorTest.java

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -60,13 +60,10 @@ public void testTronError() {
6060

6161
@Test
6262
public void ZksnarkInitTest() throws IllegalAccessException, NoSuchFieldException {
63-
Field initializedField = ZksnarkInitService.class.getDeclaredField("initialized");
64-
initializedField.setAccessible(true);
65-
Field modifiersField = Field.class.getDeclaredField("modifiers");
66-
modifiersField.setAccessible(true);
67-
modifiersField.setInt(initializedField,
68-
initializedField.getModifiers() & ~java.lang.reflect.Modifier.FINAL);
69-
AtomicBoolean atomicBoolean = (AtomicBoolean) initializedField.get(null);
63+
Field field = ZksnarkInitService.class.getDeclaredField("initialized");
64+
field.setAccessible(true);
65+
AtomicBoolean atomicBoolean = (AtomicBoolean) field.get(null);
66+
boolean originalValue = atomicBoolean.get();
7067
atomicBoolean.set(false);
7168

7269
try (MockedStatic<JLibrustzcash> mock = mockStatic(JLibrustzcash.class)) {
@@ -77,6 +74,8 @@ public void ZksnarkInitTest() throws IllegalAccessException, NoSuchFieldExceptio
7774
TronError thrown = assertThrows(TronError.class,
7875
ZksnarkInitService::librustzcashInitZksnarkParams);
7976
assertEquals(TronError.ErrCode.ZCASH_INIT, thrown.getErrCode());
77+
} finally {
78+
atomicBoolean.set(originalValue);
8079
}
8180
}
8281

0 commit comments

Comments
 (0)