chore: add unit test for UtilPrngHandler (#13254)

Signed-off-by: Lev Povolotsky <lev@swirldslabs.com>
hashgraph · May 14, 2024 · b09fc8a · b09fc8a
1 parent 40b6658
commit b09fc8a
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/...rc/test/java/com/hedera/node/app/service/util/impl/test/handlers/UtilPrngHandlerTest.java b/...rc/test/java/com/hedera/node/app/service/util/impl/test/handlers/UtilPrngHandlerTest.java
@@ -21,8 +21,12 @@
 import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
 import static org.assertj.core.api.AssertionsForClassTypes.assertThatCode;
 import static org.assertj.core.api.AssertionsForClassTypes.assertThatThrownBy;
+import static org.mockito.ArgumentMatchers.anyLong;
+import static org.mockito.ArgumentMatchers.notNull;
 import static org.mockito.BDDMockito.given;
 import static org.mockito.Mock.Strictness.LENIENT;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
 
 import com.hedera.hapi.node.base.SubType;
 import com.hedera.hapi.node.transaction.TransactionBody;
@@ -31,6 +35,8 @@
 import com.hedera.node.app.service.util.impl.records.PrngRecordBuilder;
 import com.hedera.node.app.spi.fees.FeeAccumulator;
 import com.hedera.node.app.spi.fees.FeeCalculator;
+import com.hedera.node.app.spi.fees.FeeContext;
+import com.hedera.node.app.spi.fees.Fees;
 import com.hedera.node.app.spi.fixtures.fees.FakeFeeAccumulator;
 import com.hedera.node.app.spi.fixtures.fees.FakeFeeCalculator;
 import com.hedera.node.app.spi.records.BlockRecordInfo;
@@ -137,6 +143,26 @@ void followsHappyPathWithNoRange() {
         assertThat(recordBuilder.entropyBytes).isEqualTo(hash);
     }
 
+    @Test
+    void calculateFeesHappyPath() {
+        givenTxnWithoutRange();
+        final var body = TransactionBody.newBuilder().utilPrng(txn).build();
+
+        final var feeCtx = mock(FeeContext.class);
+        given(feeCtx.body()).willReturn(body);
+
+        final var feeCalc = mock(FeeCalculator.class);
+        given(feeCtx.feeCalculator(notNull())).willReturn(feeCalc);
+        given(feeCalc.addBytesPerTransaction(anyLong())).willReturn(feeCalc);
+        // The fees wouldn't be free in this scenario, but we don't care about the actual return
+        // value here since we're using a mock calculator
+        given(feeCalc.calculate()).willReturn(Fees.FREE);
+
+        subject.calculateFees(feeCtx);
+
+        verify(feeCalc).addBytesPerTransaction(0);
+    }
+
     @ParameterizedTest
     @ValueSource(
             ints = {
@@ -301,6 +327,18 @@ void emptyHashFromRunningHashReturnsAllZeros() {
         assertThat(recordBuilder.entropyBytes).isEqualTo(Bytes.wrap(new byte[48]));
     }
 
+    @Test
+    void verifyModThrowException() {
+        assertThatThrownBy(() -> UtilPrngHandler.mod(0, 0)).isInstanceOf(ArithmeticException.class);
+    }
+
+    @Test
+    void verifyModHappyPath() {
+        assertThatCode(() -> UtilPrngHandler.mod(2, 4)).doesNotThrowAnyException();
+
+        assertThat(UtilPrngHandler.mod(2, 4)).isEqualTo(2);
+    }
+
     private void givenTxnWithRange(int range) {
         txn = UtilPrngTransactionBody.newBuilder().range(range).build();
         given(handleContext.body())