@Test public void getKeyForIteratorReturnsKey() { RegionEntryFactory regionEntryFactory = new RegionEntryFactoryBuilder().create(false, false, false, false, false); RegionEntry regionEntry = regionEntryFactory.createEntry(region, key, null); KeyInfo stringKeyInfo = new KeyInfo(key, null, null); KeyInfo regionEntryKeyInfo = new KeyInfo(regionEntry, null, null); boolean allowTombstones = false; boolean rememberReads = true; when(region.getSharedDataView()).thenReturn(view); when(view.getEntry(stringKeyInfo, region, allowTombstones)).thenReturn(mock(NonTXEntry.class)); when(view.getKeyForIterator(stringKeyInfo, region, rememberReads, allowTombstones)) .thenCallRealMethod(); when(view.getKeyForIterator(regionEntryKeyInfo, region, rememberReads, allowTombstones)) .thenCallRealMethod(); TXStateProxyImpl tx = new TXStateProxyImpl(cache, txManager, txId, false); Object key1 = tx.getKeyForIterator(regionEntryKeyInfo, region, rememberReads, allowTombstones); assertThat(key1.equals(key)).isTrue(); Object key2 = tx.getKeyForIterator(stringKeyInfo, region, rememberReads, allowTombstones); assertThat(key2.equals(key)).isTrue(); }