Increase coverage

Signed-off-by: Bilyana Gospodinova <bilyana.gospodinova14@gmail.com>
hashgraph · Dec 12, 2024 · 2e5f39f · 2e5f39f
1 parent 4c17f47
commit 2e5f39f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/hedera-mirror-web3/src/test/java/com/hedera/mirror/web3/state/FileReadableKVStateTest.java b/hedera-mirror-web3/src/test/java/com/hedera/mirror/web3/state/FileReadableKVStateTest.java
@@ -192,6 +192,17 @@ void readFromDataSourceWhenThereIsContext() {
         verify(fileDataRepository, times(0)).getFileAtTimestamp(anyLong(), anyLong());
     }
 
+    @Test
+    void readFromDataSourceWhenThereIsContextButDoesNotMatchTheKey() {
+        when(ContractCallContext.get().getFile()).thenReturn(Optional.of(FILE));
+
+        File result = fileReadableKVState.readFromDataSource(
+                FileID.newBuilder().fileNum(321L).build());
+
+        assertThat(result).isNull();
+        verify(fileDataRepository, times(1)).getFileAtTimestamp(anyLong(), anyLong());
+    }
+
     @Test
     void sizeIsAlwaysZero() {
         assertThat(fileReadableKVState.size()).isZero();