} catch (IllegalArgumentException e) {
            assertEquals("name or dataStore is null.", e.getMessage());
        }
    }

    // Test getDataStore with existing name
    public void test_getDataStore_existingName() {
        TestDataStore dataStore = new TestDataStore("TestStore");
        dataStoreFactory.add("myStore", dataStore);

        DataStore retrieved = dataStoreFactory.getDataStore("myStore");