Update tests for new data source classes.