PHPSpec, testing code that uses fluent mock

Question

I've just completed writing a spec for a Doctrine2 repository. The repo contains only one method so far, the spec for which takes in an OAuth Identity token object (that contains an ID that matches up with a database table in my system) and returns either NULL or a UserAccount entity.

Due to the fluent nature of Doctrine query building, my test seems quite verbose, and I have used a lot of lines in my test wiring the 'fluent' aspect of the Query api together.

Repository:

namespace Users\Repository;

use Users\Entity\UserAccount;
use ZF\MvcAuth\Identity;

class UserAccountRepository extends \Doctrine\ORM\EntityRepository
{
    public function getFromIdentity(Identity\IdentityInterface $identity)
    {
        if ($identity instanceof Identity\GuestIdentity) {
            return null;
        }

        if (!$identity instanceof Identity\AuthenticatedIdentity) {
            return null;
        }

        /** @var $identity Identity\AuthenticatedIdentity */
        $id = $identity->getRoleId();

        if (!is_int($id)) {
            return null;
        }

        return $this->createNamedQuery('u')
            ->where('u.id = :id')
            ->setParameter(':id', $id)
            ->getQuery()
            ->getResult();
    }
}

Spec:

/**
 * @mixin \Users\Repository\UserAccountRepository
 */
class UserAccountRepositorySpec extends ObjectBehavior
{
    function let(
        ORM\EntityManager $em,
        ORM\Mapping\ClassMetadata $classMetadata
    )
    {
        $this->beConstructedWith($em, $classMetadata);
    }

    function it_is_initializable()
    {
        $this->shouldHaveType('Users\Repository\UserAccountRepository');
    }

    function it_get_a_user_account_from_an_identity(
        ORM\EntityManager $em,
        Identity $identity,
        TestQuery $query
    )
    {
        $identity->getRoleId()->willReturn(4);

        $em->createQuery(Argument::any())->willReturn($query);
        $query->where('u.id = :id')->willReturn($query);
        $query->setParameter(':id', 4)->willReturn($query);
        $query->getQuery(Argument::any())->willReturn($query);
        $query->getResult(Argument::any())->shouldBeCalled();

        $this->getFromIdentity($identity);
    }

    function it_should_always_return_null_when_passed_a_guest_identity(
        GuestIdentity $identity
    )
    {
        $this->getFromIdentity($identity)->shouldBeNull();
    }
}

/** concrete mock due to ORM\Query being final */
class TestQuery extends ORM\AbstractQuery
{
    public function getSql(){}
    public function _doExecute(){}
    public function where(){}
    public function getQuery(){}
}