return connection.getInputStream();
    }

    @Override
    public OutputStream getOutputStream() throws IOException {
        try {
            connect();
        } catch (final IOException ex) {}
        final OutputStream output = connection.getOutputStream();
        cachedOutput = new ByteArrayOutputStream();
        return new CacheStream(output, cachedOutput);
    }

                    CopyUtil.copy(in, out);
                } catch (final IOException e) {
                    throw new WebApiException(HttpServletResponse.SC_BAD_REQUEST, e);
                }
            }
        }).execute()) {

            try (ServletOutputStream out = response.getOutputStream(); InputStream in = curlResponse.getContentAsStream()) {

        OutputStream initialOutputStream = mock(OutputStream.class);
        when(mockConnection.getOutputStream()).thenReturn(initialOutputStream);

        // Act
        // Get output stream and write data to it
        OutputStream os = ntlmConnection.getOutputStream();
        assertNotNull(os);
        byte[] testData = "test data".getBytes();
        os.write(testData);
        os.flush();

                }
            }, "ProcessCloser-error-" + sessionId).start();
            new Thread(() -> {
                try {
                    CloseableUtil.closeQuietly(process.getOutputStream());
                } catch (final Exception e) {
                    logger.warn("Could not close a process output stream.", e);
                } finally {
                    latch.countDown();
                }

        handshake();
        return this.connection.getInputStream();
    }

    @SuppressWarnings("resource")
    @Override
    public OutputStream getOutputStream() throws IOException {
        connect();
        final OutputStream output = this.connection.getOutputStream();
        this.cachedOutput = new ByteArrayOutputStream();
        return new CacheStream(output, this.cachedOutput);
    }

    @Override

                    0);
        }

        @Test
        void testGetOutputStream() throws IOException {
            // Act
            var outputStream = smbFile.getOutputStream();

            // Assert
            assertNotNull(outputStream);
            verify(smbFile).openUnshared(SmbConstants.O_CREAT | SmbConstants.O_WRONLY | SmbConstants.O_TRUNC, // 82

                    logger.fine(() -> ">>> " + body);
                    connection.setDoOutput(true);
                    try (BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(connection.getOutputStream(), encoding))) {
                        writer.write(body);
                        writer.flush();
                    }
                } else if (bodyStream != null) {

            }
            socket.connect(new InetSocketAddress(address.getHostAddress(), 139), CONN_TIMEOUT);
            socket.setSoTimeout(SO_TIMEOUT);

            out = socket.getOutputStream();
            in = socket.getInputStream();

            final SessionServicePacket ssp = new SessionRequestPacket(calledName, NbtAddress.getLocalName());
            out.write(sbuf, 0, ssp.writeWireFormat(sbuf, 0));

        lenient().when(request.getSession()).thenReturn(httpSession);
        lenient().when(request.getSession(anyBoolean())).thenReturn(httpSession);
        lenient().when(response.getOutputStream()).thenReturn(servletOutputStream);
    }

    @Test
    void testInit_success() throws ServletException {
        // Test successful initialization with valid configuration

            public void setWriteListener(WriteListener listener) {
            }
        };

        lenient().when(response.getWriter()).thenReturn(printWriter);
        lenient().when(response.getOutputStream()).thenReturn(servletOutputStream);

        // Setup servlet config
        lenient().when(servletConfig.getServletContext()).thenReturn(servletContext);