Add tests for the connection pool