@DisplayName("LeaseManager Tests")
class LeaseManagerTest {

    private LeaseManager leaseManager;
    private CIFSContext mockContext;
    private Configuration mockConfig;

    @BeforeEach
    void setUp() {
        mockContext = mock(CIFSContext.class);
        mockConfig = mock(Configuration.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);
        when(mockConfig.getLeaseTimeout()).thenReturn(30000);

        Smb2IoctlRequest request = new Smb2IoctlRequest(mockConfig, TEST_CONTROL_CODE, TEST_FILE_ID, TEST_OUTPUT_BUFFER);

        Smb2IoctlResponse response = request.createResponse(mockContext, request);

        assertNotNull(response);
        verify(mockContext).getConfig();
    }

    @Test
    @DisplayName("Test setFlags method")
    void testSetFlags() {
        setupMockConfig();

    @Mock
    private Configuration mockConfig;

    @Mock
    private CIFSContext mockContext;

    private byte[] testFileId;
    private byte[] outputBuffer;
    private Smb2ReadRequest request;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        testFileId = new byte[16];

    private Configuration mockConfig;

    @Mock
    private CIFSContext mockContext;

    @Mock
    private Smb2Lock mockLock;

    private byte[] testFileId;
    private Smb2Lock[] testLocks;
    private Smb2LockRequest request;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        testFileId = new byte[16];

@ExtendWith(MockitoExtension.class)
class Smb2ChangeNotifyRequestTest {

    private Configuration mockConfig;
    private CIFSContext mockContext;
    private byte[] testFileId;
    private Smb2ChangeNotifyRequest request;

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);
        mockContext = mock(CIFSContext.class);
        testFileId = new byte[16];
        Arrays.fill(testFileId, (byte) 0xAB);

class Smb2WriteRequestTest {

    @Mock
    private Configuration mockConfig;

    @Mock
    private CIFSContext mockContext;

    private byte[] testFileId;
    private Smb2WriteRequest request;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        testFileId = new byte[16];
        new SecureRandom().nextBytes(testFileId);

class Smb2QueryInfoRequestTest {

    private Configuration mockConfig;
    private CIFSContext mockContext;
    private Smb2QueryInfoRequest request;
    private byte[] testFileId;

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);
        mockContext = mock(CIFSContext.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);
        when(mockConfig.getMaximumBufferSize()).thenReturn(65536);

    private Configuration mockConfig;
    private CIFSContext mockContext;
    private SecureRandom mockRandom;
    private Smb2NegotiateRequest request;
    private byte[] testMachineId = new byte[16];
    private byte[] testSalt = new byte[32];

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);
        mockContext = mock(CIFSContext.class);
        mockRandom = mock(SecureRandom.class);

            SmbResource resource = createMockResource();
            when(resource.getContext()).thenReturn(mockContext);

            // When
            CIFSContext context = resource.getContext();

            // Then
            assertNotNull(context, "Context should not be null");
            assertSame(mockContext, context, "Should return the expected context");
        }

        @Test

    @Mock
    private CIFSContext mockContext;

    @Mock
    private Configuration mockConfig;

    private MockWitnessService mockService;

    @BeforeEach
    void setUp() throws Exception {
        mockService = new MockWitnessService();
        mockService.start();

        // Setup mock configuration with lenient stubbing
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);