}

    @Test
    void testInitSecContext_state1_type1Message() throws Exception {
        // Test the first step of context initialization (creating Type 1 message)
        when(mockAuth.getDomain()).thenReturn(domain);
        NtlmContext context = new NtlmContext(mockAuth, true);
        byte[] type1Token = context.initSecContext(new byte[0], 0, 0);

        assertNotNull(type1Token);

    void testGetDomain() {
        // Given
        String domain = "TESTDOMAIN";
        when(mockReferralData.getDomain()).thenReturn(domain);

        // When
        String result = mockReferralData.getDomain();

        // Then
        assertEquals(domain, result);
        verify(mockReferralData).getDomain();
    }

    @Test
    @DisplayName("Should get share")
    void testGetShare() {
        // Given

        assertEquals("user", auth.getUsername());
        assertEquals("password", auth.getPassword());
    }

    // Test constructor with user info string without password
    @Test
    void testConstructorWithUserInfoNoPassword() {
        NtlmPasswordAuthentication auth = new NtlmPasswordAuthentication("DOMAIN;user");
        assertEquals("DOMAIN", auth.getDomain());

     */
    public byte[] initSecContext(byte[] token, final int offset, final int len) throws SmbException {
        switch (state) {
        case 1:
            final Type1Message msg1 = new Type1Message(ntlmsspFlags, auth.getDomain(), workstation);
            token = msg1.toByteArray();

            if (LogStream.level >= 4) {
                log.println(msg1);
                if (LogStream.level >= 6) {

     * Gets the server name.
     *
     * @return the server name
     */
    String getServer();

    /**
     * Gets the domain name.
     *
     * @return the domain name
     */
    String getDomain();

    /**
     * Gets the full UNC path.
     *
     * @return the full UNC path
     */
    String getFullUNCPath();

    /**
     * Sets the path to the resource.
     *

     * @return the server this referral points to
     */
    String getServer();

    /**
     * Get the domain this referral is for
     *
     * @return the domain this referral is for
     */
    String getDomain();

    /**
     * Get the share this referral points to
     *
     * @return the share this referral points to
     */
    String getShare();

    /**

            this.serverAddress = serverAddress;
        }

        public String getUsername() {
            return username;
        }

        public String getDomain() {
            return domain;
        }

        public AuthType getAuthType() {
            return authType;
        }

        public long getTimestamp() {
            return timestamp;

                byte[] ntResponse = type3.getNTResponse();
                if (ntResponse == null) {
                    ntResponse = new byte[0];
                }
                return new NtlmPasswordAuthentication(type3.getDomain(), type3.getUser(), challenge, lmResponse, ntResponse);
            }
        } else {
            resp.setHeader("WWW-Authenticate", "NTLM");
        }
        resp.setStatus(HttpServletResponse.SC_UNAUTHORIZED);

                byte[] ntResponse = type3.getNTResponse();
                if (ntResponse == null) {
                    ntResponse = new byte[0];
                }
                return new NtlmPasswordAuthentication(type3.getDomain(), type3.getUser(), challenge, lmResponse, ntResponse);
            }
        } else {
            resp.setHeader("WWW-Authenticate", "NTLM");
        }
        resp.setStatus(HttpServletResponse.SC_UNAUTHORIZED);

        when(mockAuth.getUnicodeHash(any(byte[].class))).thenReturn(new byte[24]);
        when(mockAuth.getName()).thenReturn("testuser");
        when(mockAuth.getDomain()).thenReturn("TESTDOMAIN");

        setupAndX = new SmbComSessionSetupAndX(mockSession, mockAndx, mockAuth);
    }

    @Test
    void testConstructor() {
        assertNotNull(setupAndX);