Option to return ether on a contract call within a function

Is your feature request related to a problem? Please describe.
Consider a function:

function swap() external {
    // this function would transfer eth from somewhere to this contract.
    Interface(myAddress).getEthForMyTokenViaDex(....);
}

I want to mock incoming funds in such situations.

Describe the solution you'd like

await myAddress.getEthForMyTokenViaDex.returns(, {value:amount});
// OR 
await myAddress.getEthForMyTokenViaDex.returns().withValue(amount);