From 730f895edee9a6201741b4e511c2572fcb56649d Mon Sep 17 00:00:00 2001 From: Pablo Carranza Date: Sat, 10 Jun 2017 23:45:51 +0200 Subject: Add one test, at least --- tests/test_connection.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 tests/test_connection.py (limited to 'tests/test_connection.py') diff --git a/tests/test_connection.py b/tests/test_connection.py new file mode 100644 index 0000000..1c0f4ea --- /dev/null +++ b/tests/test_connection.py @@ -0,0 +1,12 @@ +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_password_works(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) -- cgit v1.2.3 From b6ee32324517afe3f35e5c85b95b924718fff9af Mon Sep 17 00:00:00 2001 From: Pablo Carranza Date: Sat, 10 Jun 2017 23:56:09 +0200 Subject: Moar tests for get_connection --- tests/test_connection.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'tests/test_connection.py') diff --git a/tests/test_connection.py b/tests/test_connection.py index 1c0f4ea..d4fb134 100644 --- a/tests/test_connection.py +++ b/tests/test_connection.py @@ -5,8 +5,14 @@ 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_password_works(self, 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') -- cgit v1.2.3