// When/Then: Should throw last exception
        IOException thrown = assertThrows(IOException.class, () -> poolSpy.getSmbTransport(ctx, "failing.server", 445, false, false));
        assertSame(secondException, thrown);

        // Verify fail counts incremented
        assertEquals(1, pool.failCounts.get("10.0.0.1"));

        when(dest.getAttributes()).thenReturn(SmbConstants.ATTR_ARCHIVE); // no READONLY bit

        // Act + Assert
        SmbAuthException thrown =
                assertThrows(SmbAuthException.class, () -> SmbCopyUtil.openCopyTargetFile(dest, SmbConstants.ATTR_NORMAL, false));
        assertSame(authEx, thrown, "Should rethrow the same SmbAuthException instance");
        verify(dest, times(1)).getAttributes();

        void testGetTransportContext() throws Exception {
            DcerpcPipeHandle handle = createMockedDcerpcPipeHandle();
            assertEquals(mockContext, handle.getTransportContext());
        }

        @Test
        @DisplayName("Should return correct server name")
        void testGetServer() throws Exception {
            DcerpcPipeHandle handle = createMockedDcerpcPipeHandle();

        @Override
        public byte[] getSessionKey() throws CIFSException {
            return sessionKey;
        }

        @Override
        protected void doSendFragment(byte[] buf, int off, int length) throws IOException {
            // Mock implementation for testing
        }

        @Override
        protected int doReceiveFragment(byte[] buf) throws IOException {
            return doReceiveFragmentReturn;

   *
   * @param timeout the maximum time to wait
   * @throws TimeoutException if the service has not reached the given state within the deadline
   * @throws IllegalStateException if the service {@linkplain State#FAILED fails}.
   * @since 28.0
   */
  default void awaitTerminated(Duration timeout) throws TimeoutException {
    awaitTerminated(toNanosSaturated(timeout), TimeUnit.NANOSECONDS);
  }

  /**

     * @return The object bound to <code>jndiName</code>.
     * @throws NamingRuntimeException
     *             Thrown if the initial context cannot be created.
     */
    public static Object lookup(final InitialContext ctx, final String jndiName) throws NamingRuntimeException {
        assertArgumentNotNull("ctx", ctx);
        assertArgumentNotEmpty("jndiName", jndiName);

    @Test
    void testDfsImplConstructor() {
        // The constructor is called in setUp(), so if no exception is thrown, this test passes.
        assertNotNull(dfsImpl);
    }

    // Tests for isTrustedDomain
    @Test
    void testIsTrustedDomain_DfsDisabled() throws SmbAuthException {
        // Scenario: DFS is disabled in the configuration.
        when(mockConfig.isDfsDisabled()).thenReturn(true);

     * any one time. If this method throws an exception or the connect timeout
     * expires an encapsulating TransportException will be thrown from connect
     * and the transport will be in error.
     */

    /**
     * Establishes a connection to the remote endpoint.
     *
     * @throws Exception if the connection fails
     */
    protected abstract void doConnect() throws Exception;

  @Override
  public final void tearDown() {
    try {
      sloppyTearDown();
    } catch (Throwable t) {
      logger.log(Level.INFO, "exception thrown during tearDown: " + t.getMessage(), t);
    }
  }

  public abstract void sloppyTearDown() throws Exception;

  public void testSingleTearDown() throws Exception {
    TearDownStack stack = buildTearDownStack();

    SimpleTearDown tearDown = new SimpleTearDown();
    stack.addTearDown(tearDown);

    assertEquals(false, tearDown.ran);

    stack.runTearDown();

    assertEquals("tearDown should have run", true, tearDown.ran);
  }

  public void testMultipleTearDownsHappenInOrder() throws Exception {