How to assert input on a chained mock method - php

Question

Im using PHPunit 8.3.5 and I'm trying to check wether a method gets the right parameter.
Example code:
$this->registry->get($thing)->apply(EXAMPLE::SUBMIT);
$this->registry->get($thing)->apply(EXAMPLE::CANCEL);
I have two functions, functionA uses the first example line, functionB the second. I need to make sure functionA uses SUBMIT and nothing else, and the same for Bs case.
The problem:
I can use a ->method('apply')->with() with a callback to test wether it gets the right input
I can create a willReturn for ->method('get')->with() to return a simple class with apply as function
I can't figure out how to combine the two
$registryMock = $this->createMock(Registry::class);
$registryMock->method('get')->willReturn(new class {
public function apply(){} // <-- I need to assert the input of this method
});
$registryMock->method('apply')->with(self::callback(function($order, $status){
return $status === EXAMPLE::SUBMIT;
}));
How can I combine those two methods? I've also tried get->apply, but that wasnt it.
Please note: Rewriting the actual code is not an option.

Answer 1

Based on the comment of Dirk:
You create a mock first the 2nd function like you would normally. You then create a mock that returns the previous mock:
// first we create a mock for the last in the chain, here '->apply()'
$registryMockApply = $this->createMock(Registry::class);
$registryMockApply->expects(self::once())->method('apply')->with(
self::equalTo(EXAMPLE::SUBMIT),
);
// Then the one before that, here '->get()', which returns the previous mock
$registryMock = $this->createMock(Registry::class);
$registryMock->method('get')->willReturn($registryMockApply);
// Together resulting in '->get()->apply()'