def test_connect_with_pgpass(self, psycopg2_mock):
config = get_collector_config('PostgresqlCollector',
{'password_provider': 'pgpass'})
self.collector = PostgresqlCollector(config, None)
conn_mock = Mock()
psycopg2_mock.connect.return_value = conn_mock
ret = self.collector._connect('test_db')
self.assertTrue(conn_mock.set_isolation_level.called)
self.assertEqual(ret, conn_mock)
psycopg2_mock.connect.assert_called_once_with(database='test_db',
host='localhost',
port=5432,
sslmode='disable',
user='******')
def setUp(self, allowed_names=None):
if not allowed_names:
allowed_names = []
config = get_collector_config('PostgresqlCollector', {})
self.collector = PostgresqlCollector(config, None)
