blob: d4fb134e8a7c91eb379f775cf19a44381a8a73b2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
import unittest
from unittest.mock import patch, Mock
from prometheus_pgbouncer_exporter import utils
class ConnectionTest(unittest.TestCase):
@patch('prometheus_pgbouncer_exporter.utils.psycopg2.connect')
def test_get_connection_with_host_and_dbname(self, connect):
conn = utils.get_connection(host='/tmp/', dbname='template1')
connect.assert_called_once_with(host='/tmp/', dbname='template1')
connect.return_value.set_session.assert_called_once_with(autocommit=True)
@patch('prometheus_pgbouncer_exporter.utils.psycopg2.connect')
def test_get_connection_with_old_args_plus_password_works(self, connect):
conn = utils.get_connection(user='pablo', port=5432, host='localhost', password='mypassword')
connect.assert_called_once_with(dbname='pgbouncer', user='pablo', port=5432, host='localhost', password='mypassword')
|