// Mock config methods needed for SMB1
        when(config.getPid()).thenReturn(1234);

        SmbComTreeConnectAndXResponse response = mock(SmbComTreeConnectAndXResponse.class);
        when(response.getService()).thenReturn("A:");
        when(response.isValidTid()).thenReturn(true);
        when(response.getTid()).thenReturn(1);
        when(session.send(any(), any())).thenReturn(response);

        long mid = 12345L;
        mockResponse.setMid(mid);

        // Verify that the method was called with the correct argument
        verify(mockResponse, times(1)).setMid(mid);
    }

    @Test
    void testGetMid() {
        long mid = 54321L;
        when(mockResponse.getMid()).thenReturn(mid);
        assertEquals(mid, mockResponse.getMid());

        // Verify the method was called

                    th.getConfig().getListSize() - FIND_OVERHEAD), this.response);

            this.nextRequest = new Trans2FindNext2(th.getConfig(), this.response.getSid(), this.response.getResumeKey(),
                    this.response.getLastName(), th.getConfig().getListCount(), th.getConfig().getListSize() - FIND_OVERHEAD);
        } catch (final SmbException e) {

        // Should return dataCount
        assertEquals(5, result);
    }

    @Test
    void testGetSid() {
        // Test the getSid method
        // By default, it should be 0
        assertEquals(0, response.getSid());
    }

    @Test
    void testGetResumeKey() {
        // Test the getResumeKey method
        // By default, it should be 0

class PetOut(BaseModel):
    name: str
    owner: UserBase


@app.post("/users/", response_model=UserBase)
async def create_user(user: UserCreate):
    return user


@app.get("/pets/{pet_id}", response_model=PetOut)
async def read_pet(pet_id: int):
    user = UserDB(
        email="******@****.***",
        hashed_password="secrethashed",
    )
    pet = PetDB(name="Nibbler", owner=user)
    return pet

class PetOut(BaseModel):
    name: str
    owner: User


@app.post("/users/", response_model=User)
async def create_user(user: UserCreate):
    return user


@app.get("/pets/{pet_id}", response_model=PetOut)
async def read_pet(pet_id: int):
    user = UserDB(
        email="******@****.***",
        hashed_password="secrethashed",
    )
    pet = PetDB(name="Nibbler", owner=user)
    return pet

        super(config, SMB_COM_TRANSACTION2, SmbComTransaction.TRANS2_FIND_FIRST2);
    }

    /**
     * Gets the search ID for this response.
     *
     * @return the sid
     */
    public final int getSid() {
        return this.sid;
    }

    /**
     * Checks if this is the end of the search results.
     *
     * @return the isEndOfSearch
     */
    public final boolean isEndOfSearch() {

        this.largeFile = largeFile;
    }

    /**
     * Returns the process ID associated with this lock range.
     *
     * @return the process ID
     */
    public int getPid() {
        return this.pid;
    }

    /**
     * Returns the starting byte offset of the lock range.
     *
     * @return the starting byte offset
     */
    public long getByteOffset() {

    }

    /**
     *
     * @return a cancel request
     */
    @Override
    public CommonServerMessageBlockRequest createCancel() {
        return new SmbComNtCancel(getConfig(), (int) getMid());
    }

    @Override
    protected int writeParameterWordsWireFormat(final byte[] dst, int dstIndex) {
        final int start = dstIndex;

        if (this.getCommand() != SMB_COM_NT_TRANSACT_SECONDARY) {

                return ((SrvPipePeekResponse) resp.getOutputData()).getReadDataAvailable();
            }
            final TransPeekNamedPipe req = new TransPeekNamedPipe(th.getConfig(), this.handle.getUncPath(), fd.getFid());
            final TransPeekNamedPipeResponse resp = new TransPeekNamedPipeResponse(th.getConfig());
            th.send(req, resp, RequestParam.NO_RETRY);