func (ahs *allHealState) setDiskHealingStatus(ep Endpoint, healing bool) {
	ahs.Lock()
	defer ahs.Unlock()

	ahs.healLocalDisks[ep] = healing
}

func (ahs *allHealState) pushHealLocalDisks(healLocalDisks ...Endpoint) {
	ahs.Lock()
	defer ahs.Unlock()

	for _, ep := range healLocalDisks {
		ahs.healLocalDisks[ep] = false
	}
}

 */
@ExtendWith(MockitoExtension.class)
@MockitoSettings(strictness = Strictness.LENIENT)
class SmbComQueryInformationTest {

    @Mock
    Configuration mockConfig;

    private SmbComQueryInformation cmd;

    @BeforeEach
    void setUp() {
        // Configure the mock to return the OEM encoding which is required for writeString
        when(mockConfig.getOemEncoding()).thenReturn("windows-1252");

import org.junit.jupiter.params.provider.MethodSource;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import jcifs.CIFSException;
import jcifs.ResourceFilter;
import jcifs.SmbConstants;
import jcifs.SmbResource;

@ExtendWith(MockitoExtension.class)
class ShareEnumIteratorTest {

    @Mock
    ResourceFilter filter;

@DisplayName("Type1Message Tests")
class Type1MessageTest {

    @Mock
    private CIFSContext mockContext;

    @Mock
    private Configuration mockConfig;

    @Mock
    private NameServiceClient mockNameServiceClient;

    @Mock
    private NetbiosAddress mockLocalHost;

    @BeforeEach
    void setUp() {
        // Setup mock context with lenient stubbing to avoid UnnecessaryStubbingException

import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import jcifs.CIFSContext;

/**
 * Test class for Request interface
 */
@ExtendWith(MockitoExtension.class)
class RequestTest {

    @Mock
    private Request<CommonServerMessageBlockResponse> request;

    @Mock
    private CommonServerMessageBlockResponse response;

    @Mock

import jcifs.internal.util.SMBUtil;

class Trans2SetFileInformationTest {

    private Trans2SetFileInformation trans2SetFileInfo;
    private Configuration config;

    @Mock
    private Configuration mockConfig;

    @Mock
    private FileInformation mockFileInfo;

    private static final int TEST_FID = 0x1234;
    private static final int TEST_ATTRIBUTES = 0x20; // FILE_ATTRIBUTE_ARCHIVE

import jcifs.internal.smb1.com.SmbComClose;
import jcifs.internal.smb2.create.Smb2CloseRequest;

@ExtendWith(MockitoExtension.class)
class SmbFileHandleImplTest {

    @Mock
    Configuration cfg;

    @Mock
    SmbTreeHandleImpl tree;

    private void stubValidTree(long treeId, boolean connected, boolean smb2) {
        lenient().when(tree.acquire()).thenReturn(tree);
        lenient().when(tree.getTreeId()).thenReturn(treeId);

import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import jcifs.CIFSException;
import jcifs.Configuration;

/**
 * Tests for SpnegoContext focusing on delegation, error handling, and edge cases.
 */
@ExtendWith(MockitoExtension.class)
class SpnegoContextTest {

    @Mock
    Configuration config;

    @Mock
    SSPContext mechContext;

        final int[] searchLogDeletedDays = { 0 };
        final int[] jobLogDeletedDays = { 0 };
        final int[] userInfoDeletedDays = { 0 };
        final long[] expectedTime = { 0 };

        // Create mock services
        CrawlingInfoService crawlingInfoService = new CrawlingInfoService() {
            @Override
            public void deleteBefore(long time) {
                deleteCrawlingInfoCalled[0] = true;

import static org.mockito.Mockito.mock;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

/**
 * Test class for SmbNegotiation
 */
class SmbNegotiationTest {

    @Mock
    private SmbNegotiationRequest mockRequest;

    @Mock
    private SmbNegotiationResponse mockResponse;