public function testAllocateIds()
{
$this->operation->allocateIds(Argument::type('array'), Argument::type('array'))->shouldBeCalled()->willReturn([]);
$this->datastore->setOperation($this->operation->reveal());
$key = $this->prophesize(Key::class);
$keys = [$key->reveal(), $key->reveal()];
$res = $this->datastore->allocateIds($keys);
$this->assertTrue(is_array($res));
}