public function testTransaction()
{
$snippet = $this->snippetFromMethod(DatastoreClient::class, 'transaction');
$snippet->addLocal('datastore', $this->client);
$this->connection->beginTransaction(Argument::any())->shouldBeCalled()->willReturn(['transaction' => 'foo']);
$this->client->setConnection($this->connection->reveal());
$res = $snippet->invoke('transaction');
$this->assertInstanceOf(Transaction::class, $res->returnVal());
}