Simplify data source test to only test getConnection method