Add associate state to test data sources