mockAuth.username = "testuser";
        mockAuth.domain = "TESTDOMAIN";
        mockAuth.password = "testpass";

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

    private final String testDomain = "TEST_DOMAIN";
    private final int testServerTypes = NetServerEnum2.SV_TYPE_ALL;

    @BeforeEach
    void setUp() {
        netServerEnum2 = new NetServerEnum2(testDomain, testServerTypes);
    }

    /**
     * Test the constructor of NetServerEnum2.
     */
    @Test
    void testConstructor() {
        assertEquals(testDomain, netServerEnum2.domain);

     * Test basic session lifecycle management
     */
    @Test
    @Timeout(5)
    public void testSessionLifecycle() throws Exception {
        SmbSessionImpl session = new SmbSessionImpl(mockContext, "testhost", "testdomain", mockTransport);

        // Test initial state - session starts with usage count > 0 due to transport.acquire()
        assertTrue(session.isInUse(), "New session should be in use due to transport acquisition");

    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() {

        verify(mockCredentials).unwrap(Credentials.class);
    }

    @Test
    @DisplayName("Should get user domain")
    void testGetUserDomain() {
        // Given
        String domain = "TESTDOMAIN";
        when(mockCredentials.getUserDomain()).thenReturn(domain);

        // When
        String result = mockCredentials.getUserDomain();

        // Then
        assertEquals(domain, result);

        @Override
        public boolean isAnonymous() {
            return false; // not anonymous for test purposes
        }

        @Override
        public String getUserDomain() {
            return "TESTDOMAIN"; // test domain for test purposes
        }

        @Override
        public <T extends Credentials> T unwrap(Class<T> type) {
            if (type.isInstance(this)) {
                return type.cast(this);

import jcifs.smb1.ntlmssp.Type2Message;

@ExtendWith(MockitoExtension.class)
class NtlmContextTest {

    @Mock
    private NtlmPasswordAuthentication mockAuth;

    private final String domain = "TEST_DOMAIN";
    private final String username = "testUser";
    private final String password = "testPassword";
    private final String workstation = "TEST_WORKSTATION";

    @BeforeEach
    void setUp() {

    }

    /**
     * Returns the domain.
     * @return The domain.
     */
    public String getDomain() {
        return domain;
    }

    /**
     * Sets the domain.
     * @param domain The domain to set.
     */
    public void setDomain(final String domain) {
        this.domain = domain;
    }

    /**
     * Returns the domain for SMB authentication.
     * @return the domain
     */
    public String getDomain() {
        return domain;
    }

    /**
     * Sets the domain for SMB authentication.
     * @param domain the domain to set
     */
    public void setDomain(final String domain) {
        this.domain = domain;
    }

    /**

     * @throws CIFSException
     */
    @Test
    void testGetOEMDomainName() throws CIFSException {
        when(smbTreeHandle.getOEMDomainName()).thenReturn("TEST_DOMAIN");
        assertEquals("TEST_DOMAIN", smbTreeHandle.getOEMDomainName(), "OEM domain name should be TEST_DOMAIN");
    }

    /**
     * Test for getOEMDomainName() method throwing CIFSException.
     * @throws CIFSException
     */
    @Test