request = new Smb2NegotiateRequest(mockConfig, 0);

        // Then
        int[] dialects = request.getDialects();
        assertNotNull(dialects);
        assertEquals(4, dialects.length); // SMB210, SMB300, SMB302, SMB311
        assertEquals(0x0210, dialects[0]);
        assertEquals(0x0300, dialects[1]);
        assertEquals(0x0302, dialects[2]);
        assertEquals(0x0311, dialects[3]);
    }

    @Test

    void testSMB302Dialect() {
        // When
        Smb2EncryptionContext context = new Smb2EncryptionContext(1, DialectVersion.SMB302, testEncryptionKey, testDecryptionKey);

        // Then
        assertEquals(DialectVersion.SMB302, context.getDialect(), "Should support SMB 3.0.2 dialect");
    }

    @Test
    @DisplayName("Should handle different cipher IDs")

    void testWithDifferentConfigurations() throws Exception {
        // Test with different configuration settings
        Properties props = new Properties();
        props.setProperty("jcifs.smb.client.maxVersion", "SMB302");
        Configuration customConfig = new PropertyConfiguration(props);

        Trans2SetFileInformationResponse customResponse = new Trans2SetFileInformationResponse(customConfig);
        assertNotNull(customResponse);

    void testAllDialectVersions() {
        // Test each dialect version
        DialectVersion[] dialects = { DialectVersion.SMB1, DialectVersion.SMB202, DialectVersion.SMB210, DialectVersion.SMB300,
                DialectVersion.SMB302, DialectVersion.SMB311 };

        for (DialectVersion dialect : dialects) {
            when(negotiationResponse.getSelectedDialect()).thenReturn(dialect);

        return Stream.of(Arguments.of(0x0202, DialectVersion.SMB202, true), Arguments.of(0x0210, DialectVersion.SMB210, true),
                Arguments.of(0x0300, DialectVersion.SMB300, true), Arguments.of(0x0302, DialectVersion.SMB302, true),
                Arguments.of(0x0311, DialectVersion.SMB311, true), Arguments.of(0xFFFF, null, false), // SMB2_DIALECT_ANY
                Arguments.of(0x9999, null, false) // Unknown dialect
        );
    }