public void setUp() throws Exception {
        mocks = MockitoAnnotations.openMocks(this);

        // Set up mock provider to return TCP memory regions
        when(mockProvider.registerMemory(any(ByteBuffer.class), any())).thenAnswer(invocation -> {
            ByteBuffer buffer = invocation.getArgument(0);
            EnumSet<RdmaAccess> access = invocation.getArgument(1);
            return new TcpMemoryRegion(buffer, access);
        });

                entry("keep2", SmbConstants.TYPE_SHARE));

        // Filter accepts names starting with "keep"
        when(filter.accept(any())).thenAnswer(inv -> {
            SmbResource res = inv.getArgument(0);
            return res.getName().startsWith("keep");
        });

        ShareEnumIterator it = new ShareEnumIterator(parent, entries.iterator(), filter);

                // Configure mock to do nothing when called
                mockedSMBUtil.when(() -> SMBUtil.writeInt2(anyLong(), any(byte[].class), anyInt())).thenAnswer(invocation -> null);

                byte[] dst = new byte[buffer.size()];
                buffer.encode(dst, 0);

                // Verify SMBUtil.writeInt2 was called for referral level (with long parameter)

        when(tree.getConfig()).thenReturn(cfg);
        when(tree.getReceiveBufferSize()).thenReturn(1024);
        when(tree.getSendBufferSize()).thenReturn(1024);
        when(tree.hasCapability(anyInt())).thenAnswer(inv -> ntSmbsCap && inv.getArgument(0).equals(SmbConstants.CAP_NT_SMBS));
        when(tree.areSignaturesActive()).thenReturn(false);
        when(tree.isSMB2()).thenReturn(smb2);