Key key1 = new Key(1);
      String value1 = key1.toString();
      Key key2 = new Key(2);
      String value2 = key2.toString();
      assertThat(cache.getUnchecked(key1)).isSameInstanceAs(value1);
      assertThat(cache.getUnchecked(key2)).isSameInstanceAs(value2);
      assertThat(cache.asMap().keySet()).isEqualTo(ImmutableSet.of(key1, key2));
      assertThat(cache.asMap().values()).containsExactly(value1, value2);

      Key key1 = new Key(1);
      String value1 = key1.toString();
      Key key2 = new Key(2);
      String value2 = key2.toString();
      assertThat(cache.getUnchecked(key1)).isSameInstanceAs(value1);
      assertThat(cache.getUnchecked(key2)).isSameInstanceAs(value2);
      assertThat(cache.asMap().keySet()).isEqualTo(ImmutableSet.of(key1, key2));
      assertThat(cache.asMap().values()).containsExactly(value1, value2);

        assertTrue(dataStoreParams.containsKey("key1"));

        dataStoreParams.put("key2", null);
        assertTrue(dataStoreParams.containsKey("key2"));
    }

    // Test putAll
    @Test
    public void test_putAll() {
        Map<String, String> map = new HashMap<>();
        map.put("key1", "value1");
        map.put("key2", "value2");
        map.put("key3", "value3");

        dataStoreParams.putAll(map);

        DocMap docMap = new DocMap(parentMap);

        docMap.put("key1", "value1");
        docMap.put("key2", 123);
        docMap.put("key3", true);

        assertEquals(3, docMap.size());
        assertEquals("value1", docMap.get("key1"));
        assertEquals(123, docMap.get("key2"));
        assertEquals(true, docMap.get("key3"));

        assertTrue(docMap.containsKey("key1"));

        Smb2LeaseKey key2 = leaseManager.requestLease(path2, Smb2LeaseState.SMB2_LEASE_WRITE_CACHING);

        assertNotNull(leaseManager.getLease(key1));
        assertNotNull(leaseManager.getLease(key2));

        leaseManager.releaseAll();

        assertNull(leaseManager.getLease(key1));
        assertNull(leaseManager.getLease(key2));
        assertTrue(leaseManager.getAllLeases().isEmpty());

        Smb2LeaseKey key2 = new Smb2LeaseKey(testBytes2);

        assertEquals(key1.hashCode(), key2.hashCode());
    }

    @Test
    @DisplayName("Should generate different random keys")
    void testRandomKeyUniqueness() {
        Smb2LeaseKey key1 = new Smb2LeaseKey();
        Smb2LeaseKey key2 = new Smb2LeaseKey();

        assertNotEquals(key1, key2);

	}
	if !ok {
		t.Fatalf("should be true")
	}

	if !lc.Contains("key1") {
		t.Fatalf("should contain key1")
	}
	if lc.Contains("key2") {
		t.Fatalf("should not contain key2")
	}

	v, ok = lc.Peek("key2")
	if v != "" {
		t.Fatalf("should be empty")
	}
	if ok {
		t.Fatalf("should be false")
	}

	if !reflect.DeepEqual(lc.Keys(), []string{"key1"}) {

        sourceMap.put("key1", "value1");
        sourceMap.put("key2", 123);
        sourceMap.put("key3", true);
        event.setSourceMap(sourceMap);

        Map<String, Object> result = event.toSource();
        assertNotNull(result);
        assertEquals(3, result.size());
        assertEquals("value1", result.get("key1"));
        assertEquals(123, result.get("key2"));
        assertEquals(true, result.get("key3"));

        byte[] key2 = keyManager.generateRandomKey(32);

        assertNotNull(key1, "Should generate key 1");
        assertNotNull(key2, "Should generate key 2");
        assertEquals(32, key1.length, "Key 1 should have correct length");
        assertEquals(32, key2.length, "Key 2 should have correct length");
        assertFalse(Arrays.equals(key1, key2), "Random keys should be different");
    }

    @Test
    void testGetKey_KeyExists() throws LoginException {
        when(key1.getKeyType()).thenReturn(KEY_TYPE_1);
        when(key2.getKeyType()).thenReturn(KEY_TYPE_2);
        Set<Object> privateCredentials = new HashSet<>();
        privateCredentials.add(key1);
        privateCredentials.add(key2);
        when(subject.getPrivateCredentials()).thenReturn(privateCredentials);