public function testUpsertBatch()
{
$this->operation->commit(Argument::type('array'), Argument::type('array'))->shouldBeCalled()->willReturn(['mutationResults' => [['version' => '1234']]]);
$this->operation->mutation(Argument::exact('upsert'), Argument::type(Entity::class), Argument::exact(Entity::class), Argument::exact(null))->shouldBeCalled();
$this->datastore->setOperation($this->operation->reveal());
$e = $this->prophesize(Entity::class);
$res = $this->datastore->upsertBatch([$e->reveal()]);
$this->assertEquals($res, ['mutationResults' => [['version' => '1234']]]);
}