if err := DB.Model(&user2).Association("Company").Append(&company); err != nil {
		t.Fatalf("Error happened when append Company, got %v", err)
	}

	if company.ID == 0 {
		t.Fatalf("Company's ID should be created")
	}

	if err := DB.Model(&user2).Association("Manager").Append(manager); err != nil {
		t.Fatalf("Error happened when append Manager, got %v", err)
	}

	if manager.ID == 0 {
		t.Fatalf("Manager's ID should be created")

    call: Call,
  ) {}

  /**
   * Invoked when a connection is released as no longer required.
   */
  open fun connectionClosed(connection: Connection) {}

  /**
   * Invoked when a call is assigned a particular connection.
   */
  open fun connectionAcquired(
    connection: Connection,
    call: Call,
  ) {}

  /**
   * Invoked when a call no longer uses a connection.
   */

    }

    @Test
    void testSingleChannelSelection() {
        when(mockChannelManager.getHealthyChannels()).thenReturn(Collections.singletonList(channel1));

        ChannelInfo selected = loadBalancer.selectChannel(mockMessage);
        assertEquals(channel1, selected);
    }

    @Test
    void testNoChannelsAvailable() {
        when(mockChannelManager.getHealthyChannels()).thenReturn(Collections.emptyList());

            rpc.retval = 0;
            return null;
        }).when(mockDcerpcHandle).sendrecv(any(MsrpcLsarOpenPolicy2.class));

        LsaPolicyHandle handle = new LsaPolicyHandle(mockDcerpcHandle, "server", 123);

        // Now, mock the behavior for MsrpcLsarClose to throw IOException
        doThrow(new IOException("Close network error")).when(mockDcerpcHandle).sendrecv(any(MsrpcLsarClose.class));

        // Act & Assert

    SmbTreeHandleImpl tree;

    private void stubValidTree(long treeId, boolean connected, boolean smb2) {
        lenient().when(tree.acquire()).thenReturn(tree);
        lenient().when(tree.getTreeId()).thenReturn(treeId);
        lenient().when(tree.isConnected()).thenReturn(connected);
        lenient().when(tree.isSMB2()).thenReturn(smb2);
    }

    @Test
    @DisplayName("Constructor with null tree throws NPE (byte[] fid variant)")

        when(ctx.getConfig()).thenReturn(config);
        when(config.getCapabilities()).thenReturn(0);
        when(config.getFlags2()).thenReturn(0);
        when(config.getMaxMpxCount()).thenReturn(1);
        when(config.getSendBufferSize()).thenReturn(4096);
        when(config.getReceiveBufferSize()).thenReturn(4096);
        when(config.getTransactionBufferSize()).thenReturn(4096);

    public Integer crawlingThreadCount;

    /**
     * Enable or disable search query logging.
     * When enabled, user search queries are logged for analysis.
     */
    @Size(max = 10)
    public String searchLog;

    /**
     * Enable or disable user information tracking.
     * When enabled, user information is stored and tracked.
     */
    @Size(max = 10)
    public String userInfo;

    /**

        @Test
        @DisplayName("Should use special name when no expanded names")
        void testFromReferralWithSpecialName() {
            when(mockReferral.getTtl()).thenReturn(300);
            when(mockReferral.getRFlags()).thenReturn(Trans2GetDfsReferralResponse.FLAGS_NAME_LIST_REFERRAL);
            when(mockReferral.getExpandedNames()).thenReturn(new String[0]);
            when(mockReferral.getSpecialName()).thenReturn("\\SPECIAL");

        }
    }

    @BeforeEach
    void setUp() {
        lenient().when(fileEntry.getName()).thenReturn("test");
    }

    @Test
    @DisplayName("Iterator without filter - simple case")
    void iteratorWithoutFilter() {
        // Setup: constructor calls advance() once
        when(delegate.hasNext()).thenReturn(true, false);
        when(delegate.next()).thenReturn(fileEntry);

        lenient().when(UniAddress.getByName("domain.com", true)).thenReturn(uniAddress);
        lenient().when(SmbTransport.getSmbTransport(uniAddress, 0)).thenReturn(smbTransport);

        DfsReferral referral = new DfsReferral();
        referral.server = "dc1.domain.com";
        referral.next = referral;