[New feature] Running code after mock method was called about mockery HOT 2 CLOSED

final class FakeRepository {
    private $records = [];
    public function get(int $id): ?string
    {
        return $this->records[$id] ?? null;
    }
    public function add(int $id, string $record) {
        $this->records[$id] = $record;
    }
}

final class CQRSCommandTest extends \PHPUnit\Framework\TestCase {
    use MockeryPHPUnitIntegration;

    public function testFetchRecord(): void
    {
        // Create a mock for the CQRS command
        $cqrsCommand = \Mockery::mock(CQRSCommandInterface::class);

        // Create an instance of the fake repository
        $repository = new FakeRepository();
        
        $id = 42;
        $returnValue = 'New record';

        // Set up the expectations for the mock CQRS command
        $cqrsCommand->shouldReceive('execute')->once()
            ->andReturnUsing(static function() use ($repository, $id, $returnValue) {               
                // Add a record to the repository when the CQRS command is called
                $repository->add($id, $returnValue);

                // return the desired value
                return $returnValue;
            });

        // Fetch record from the repository
        $record = $repository->get($id);

        // Assert that the record is initially not present
        self::assertNull($record);

        // Call the CQRS command, which should add the record to the repository
        $cqrsCommand->execute();

        // Fetch record again
        $record = $repository->get($id);

        // Assert that the record is now present
        self::assertSame($returnValue, $record);
    }
}