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);

        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");

     * 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");

        assertEquals(server, result);
        verify(mockReferralData).getServer();
    }

    @Test
    @DisplayName("Should get domain")
    void testGetDomain() {
        // Given
        String domain = "TESTDOMAIN";
        when(mockReferralData.getDomain()).thenReturn(domain);

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

        // Then
        assertEquals(domain, result);

        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);

     * @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

        serverData.serverTimeZone = 300;
        serverData.encryptionKeyLength = 8;
        response.byteCount = 10;
        serverData.oemDomainName = "TEST_DOMAIN";

        String result = response.toString();

        assertTrue(result.contains("dialectIndex=5"));
        assertTrue(result.contains("securityMode=0x3"));

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