response.prepare(mockNext);

        // Then
        verify(mockNext).setTreeId(999);
    }

    @Test
    @DisplayName("Should not set tree ID when not received")
    void testPrepareWhenNotReceived() {
        // Given
        ServerMessageBlock2Request mockNext = mock(ServerMessageBlock2Request.class);

        // When
        response.prepare(mockNext);

        // Then