Add test for distributed transaction for data sources