Mocking fluent interfaces with Mockery, and vary return values based on input to methods in the middle of the chain

Bar.php

<?php
namespace Ramsey\Talks;

class Bar
{
    public function getSomething(Foo $foo, $arg1, $arg2)
    {
        $result = $foo->bar()->baz($arg1)->qux()->quux($arg2);

        return "Now, we're {$result}";
    }
}

FluentTest.php

<?php
namespace Ramsey\Talks\Test;

class FluentTest extends \PHPUnit_Framework_TestCase
{
    public function tearDown()
    {
        \Mockery::close();
    }

    public function testFluentInterfacesWithParameterAtEndOfChain()
    {
        $mock = \Mockery::mock('Ramsey\\Talks\\Foo');
        $mock->shouldReceive('bar->baz->qux->quux')
            ->with(123)
            ->andReturn('awesome!');

        $mock->shouldReceive('bar->baz->qux->quux')
            ->with(321)
            ->andReturn('cool!');

        $bar = new \Ramsey\Talks\Bar;

        $this->assertEquals("Now, we're awesome!", $bar->getSomething($mock, null, 123));
        $this->assertEquals("Now, we're cool!", $bar->getSomething($mock, null, 321));
    }

    public function testFluentInterfacesWithInterveningParameters()
    {
        $baz = \Mockery::mock('baz');
        $baz->shouldReceive('qux->quux')
            ->with(321)
            ->andReturn('rad!');

        $mock = \Mockery::mock('Ramsey\\Talks\\Foo');
        $mock->shouldReceive('bar->baz')
            ->with(123)
            ->andReturn($baz);

        $bar = new \Ramsey\Talks\Bar;

        $this->assertEquals("Now, we're rad!", $bar->getSomething($mock, 123, 321));
    }
}