when(mockReferralData.getKey()).thenReturn(key);
            doNothing().when(mockReferralData).setKey(key);

            mockReferralData.setKey(key);
            String retrievedKey = mockReferralData.getKey();

            assertEquals(key, retrievedKey);
            verify(mockReferralData, times(1)).setKey(key);
            verify(mockReferralData, times(1)).getKey();

            // readWR.setRdma().setRemote_addr(remoteAddress);
            // readWR.setRdma().setRkey(remoteKey);
            //
            // LinkedList<IbvSge> sgeList = new LinkedList<>();
            // IbvSge sge = new IbvSge();
            // sge.setAddr(disniRegion.getAddress());
            // sge.setLength(length);
            // sge.setLkey(disniRegion.getLocalKey());
            // sgeList.add(sge);
            //

        }

        @Test
        @DisplayName("Should get and set key")
        void testKey() {
            assertNull(referralData.getKey());
            referralData.setKey("cache-key-123");
            assertEquals("cache-key-123", referralData.getKey());
        }

        @Test
        @DisplayName("Should get server")
        void testGetServer() {
            assertNull(referralData.getServer());

                final CrawlingInfoParam crawlingInfoParam = new CrawlingInfoParam();
                crawlingInfoParam.setCrawlingInfoId(crawlingInfo.getId());
                crawlingInfoParam.setKey(entry.getKey());
                crawlingInfoParam.setValue(entry.getValue());
                crawlingInfoParamList.add(crawlingInfoParam);
            }
            getCrawlingInfoService().storeInfo(crawlingInfoParamList);

        this.link = link;
    }

    /**
     * @return the key
     */
    @Override
    public String getKey() {
        return this.key;
    }

    /**
     * @param key
     *            the key to set
     */
    @Override
    public void setKey(final String key) {
        this.key = key;
    }

    @Override
    public String getServer() {

      if (delegate != null) {
        return delegate.entrySet().contains(o);
      } else if (o instanceof Entry) {
        Entry<?, ?> entry = (Entry<?, ?>) o;
        int index = indexOf(entry.getKey());
        return index != -1 && Objects.equals(value(index), entry.getValue());
      }
      return false;
    }

    @Override
    public boolean remove(@Nullable Object o) {

        decodedContext.decode(buffer, 0, buffer.length);

        // Verify decoded values
        assertArrayEquals(originalKey.getKey(), decodedContext.getLeaseKey().getKey());
        assertEquals(originalLeaseState, decodedContext.getLeaseState());
        assertEquals(originalScope, decodedContext.getCacheScope());
        assertEquals(60000L, decodedContext.getMaxCacheAge());

        assertTrue(result);
        assertEquals(1, chain.changePasswordCalls.size());
        assertEquals("testuser", chain.changePasswordCalls.get(0).getKey());
        assertEquals("newpassword123", chain.changePasswordCalls.get(0).getValue());
    }

    // Test changePassword with failure
    public void test_changePassword_failure() {

        docMap.put("banana", "value3");

        Set<Map.Entry<String, Object>> entrySet = docMap.entrySet();
        assertEquals(4, entrySet.size());

        String[] keys = entrySet.stream().map(Map.Entry::getKey).toArray(String[]::new);

        boolean langFirst = "lang".equals(keys[0]);
        assertTrue("lang should be first or sorted normally", langFirst || Arrays.asList(keys).contains("lang"));

    RangeMapEntry(Range<K> range, V value) {
      super(range, value);
    }

    boolean contains(K value) {
      return getKey().contains(value);
    }

    Cut<K> getLowerBound() {
      return getKey().lowerBound;
    }

    Cut<K> getUpperBound() {
      return getKey().upperBound;
    }
  }

  @Override
  public @Nullable V get(K key) {
    Entry<Range<K>, V> entry = getEntry(key);