@Test
public void allocateId() throws IOException {

  when(datastore.allocateId(incompleteKey)).thenReturn(key);
  assertThat(sut.allocateId(incompleteKey), is(key));
  verify(datastore).allocateId(incompleteKey);
}