Add tests for FlexyPoolDataSource and increased test coverage