DefaultDataStoreRegistry implements DataStoreRegistry { @Override public <E, K> DataStore<K, E> getDataStore(Class<E> entityType) { if (!dataStores.containsKey(entityType)) { log.error("There is no data store registered for entity type " + entityType); log.debug("Registered data types are " + dataStores.keySet()); throw new DataStoreNotFoundException(entityType); } return (DataStore<K, E>) dataStores.get(entityType); }  @Override void register(DataStore<K, E> dataStore); @Override DataStore<K, E> getDataStore(Class<E> entityType); @Override boolean has(Class<?> entityType);  }
@Test(expectedExceptions = DataStoreNotFoundException.class) public void testLookingForInvalidDataStore() { final DefaultDataStoreRegistry registry = new DefaultDataStoreRegistry(); registry.getDataStore(Person.class); }