class AddressTest extends BaseTest {

    @Mock
    private CIFSContext mockContext;

    @Mock
    private Address mockAddress;

    @Test
    @DisplayName("Address interface should define correct method signatures")
    void testInterfaceContract() {
        // Given
        Address address = mockAddress;

        // When & Then - verify interface methods exist and can be called
        assertDoesNotThrow(() -> {

 * Tests the batch limit configuration for various SMB commands.
 */
public class SmbConnectionTest {

    /**
     * Test that getBatchLimit returns correct values for different commands
     */
    @Test
    @DisplayName("getBatchLimit returns correct values for different SMB commands")
    public void testBatchLimitForDifferentCommands() throws CIFSException {
        Properties props = new Properties();

        byte[] buffer = new byte[100];
        int dstIndex = 10;
        int bytesWritten = info.encode(buffer, dstIndex);

        // Verify replaceIfExists flag at correct position
        assertEquals(1, buffer[dstIndex]);

        // Verify file name length at correct position
        int nameLength = SMBUtil.readInt4(buffer, dstIndex + 16);
        assertEquals(fileName.getBytes(StandardCharsets.UTF_16LE).length, nameLength);

        assertTrue(resp instanceof ServerMessageBlock2);
        assertTrue(resp instanceof TreeConnectResponse);
    }

    @Test
    @DisplayName("Should return correct share type constants")
    void testShareTypeConstants() {
        // Then
        assertEquals((byte) 0x1, Smb2TreeConnectResponse.SMB2_SHARE_TYPE_DISK);

    class InterfaceConstantsTests {

        @Test
        @DisplayName("Should have correct SMB_INFO_ALLOCATION constant value")
        void testSmbInfoAllocationConstant() {
            assertEquals((byte) -1, FileSystemInformation.SMB_INFO_ALLOCATION);
        }

        @Test
        @DisplayName("Should have correct FS_SIZE_INFO constant value")
        void testFsSizeInfoConstant() {

        verify(mockResponse, times(1)).getErrorCode();
    }

    @Test
    void testSetMid() {
        long mid = 12345L;
        mockResponse.setMid(mid);

        // Verify that the method was called with the correct argument
        verify(mockResponse, times(1)).setMid(mid);
    }

    @Test
    void testGetMid() {
        long mid = 54321L;
        when(mockResponse.getMid()).thenReturn(mid);

        assertEquals(32, derived1.length, "Derived key should have correct length");
        assertEquals(32, derived2.length, "Derived key should have correct length");
        assertEquals(32, derived3.length, "Derived key should have correct length");

        assertFalse(Arrays.equals(derived1, derived2), "Different labels should produce different keys");

        SamrAliasHandle aliasHandle = new SamrAliasHandle(mockDcerpcHandle, mockSamrDomainHandle, access, rid);

        // Assert
        assertNotNull(aliasHandle);
        // Verify that sendrecv was called with the correct MsrpcSamrOpenAlias object
        ArgumentCaptor<MsrpcSamrOpenAlias> rpcCaptor = ArgumentCaptor.forClass(MsrpcSamrOpenAlias.class);
        verify(mockDcerpcHandle).sendrecv(rpcCaptor.capture());

        long expectedValue = 123456789012345L;
        NdrHyper ndrHyper = new NdrHyper(expectedValue);
        assertEquals(expectedValue, ndrHyper.value, "Constructor should set the correct value.");
    }

    @Test
    void testEncode() throws NdrException {
        // Test the encode method
        long testValue = 987654321098765L;
        NdrHyper ndrHyper = new NdrHyper(testValue);

    class FlagConstantsTests {

        @Test
        @DisplayName("Should have correct shared lock flag value")
        void testSharedLockFlag() {
            assertEquals(0x1, Smb2Lock.SMB2_LOCKFLAG_SHARED_LOCK);
        }

        @Test
        @DisplayName("Should have correct exclusive lock flag value")
        void testExclusiveLockFlag() {