}

    @Test
    @DisplayName("Should perform DES encryption/decryption")
    void testDES() throws Exception {
        // Given
        byte[] key = "testkey1".getBytes(); // 8 bytes for DES
        byte[] plaintext = "12345678".getBytes(); // 8 bytes (DES block size)

        // When
        Cipher desCipher = Crypto.getDES(key);

import org.mockito.Mock;
import org.mockito.MockedStatic;
import org.mockito.junit.jupiter.MockitoExtension;

@ExtendWith(MockitoExtension.class)
class HMACT64Test {

    private static final byte[] TEST_KEY = "testkey".getBytes();
    private static final byte[] LONG_TEST_KEY = "thisisalongtestkeythatislongerthan64bytesandshouldbetruncated".getBytes();
    private static final byte[] SHORT_TEST_KEY = "short".getBytes();

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;

import org.junit.jupiter.api.Test;

class HMACT64Test {

    private static final byte[] TEST_KEY = "testkey".getBytes();
    private static final byte[] LONG_TEST_KEY = new byte[80]; // Longer than 64 bytes
    private static final byte[] SHORT_TEST_KEY = "short".getBytes();

 */
public class SecureKeyManagerTest {

    private SecureKeyManager keyManager;
    private byte[] testKey;

    @BeforeEach
    public void setUp() {
        keyManager = new SecureKeyManager();
        testKey = new byte[16];
        new SecureRandom().nextBytes(testKey);
    }

    @AfterEach
    public void tearDown() {
        if (keyManager != null) {
            keyManager.close();

    private LeaseV1CreateContextRequest leaseContext;
    private Smb2LeaseKey testKey;
    private int testState;

    @BeforeEach
    void setUp() {
        byte[] keyBytes = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
        testKey = new Smb2LeaseKey(keyBytes);
        testState = Smb2LeaseState.SMB2_LEASE_READ_WRITE;

        for (int size : keySizes) {
            // Given
            byte[] testKey = new byte[size];
            new SecureRandom().nextBytes(testKey);

            // When
            byte[] derivedKey = Smb3KeyDerivation.deriveSigningKey(dialect, testKey, preauthIntegrity);

            // Then
            assertNotNull(derivedKey, "Should handle " + size + "-byte session key");

        byte[] testKey = new byte[16];
        new SecureRandom().nextBytes(testKey);

        // When - Create context with SecureKeyManager using same key for both operations
        Smb2EncryptionContext context = new Smb2EncryptionContext(EncryptionNegotiateContext.CIPHER_AES128_GCM, DialectVersion.SMB311,
                testKey, testKey, keyManager);

            assertEquals("\\\\server\\share\\link", referralData.getLink());
        }

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

        @Test