@Test
public void shouldWriteActiveStatesWithSamePartitionAsSeparateEntities() throws Exception {

  storage.writeActiveState(WORKFLOW_INSTANCE1, RUN_STATE);
  storage.writeActiveState(WORKFLOW_INSTANCE2, RUN_STATE2);
  assertThat(entitiesOfKind(DatastoreStorage.KIND_ACTIVE_WORKFLOW_INSTANCE), hasSize(2));
}