def test_instance_names(self, publish, _get_stats_by_database): def side_effect(host, port, user, password): if (host, port) == ('127.0.0.1', '6432'): return {'foo': {'bar': 42}} elif (host, port) == ('localhost', '6433'): return {'foo': {'baz': 24}} _get_stats_by_database.side_effect = side_effect config = get_collector_config('PgbouncerCollector', { 'instances': { 'alpha': { 'host': '127.0.0.1', 'port': '6432', }, 'beta': { 'host': 'localhost', 'port': '6433', }, } }) collector = PgbouncerCollector(config, None) collector.collect() self.assertPublished(publish, 'alpha.foo.bar', 42) self.assertPublished(publish, 'beta.foo.baz', 24)
def test_override_user_password(self, _get_stats_by_database): _get_stats_by_database.return_value = {} config = get_collector_config( 'PgbouncerCollector', { 'instances': { 'test1': { 'host': '127.0.0.1', 'port': '6433', 'password': '******', }, 'test2': { 'host': '127.0.0.2', 'port': '6432', 'user': '******', } } }) collector = PgbouncerCollector(config, None) collector.collect() _get_stats_by_database.assert_any_call('127.0.0.1', '6433', 'postgres', 'foobar') _get_stats_by_database.assert_any_call('127.0.0.2', '6432', 'pgbouncer', '')
def test_instance_names(self, publish, _get_stats_by_database): def side_effect(host, port, user, password): if (host, port) == ('127.0.0.1', '6432'): return {'foo': {'bar': 42}} elif (host, port) == ('localhost', '6433'): return {'foo': {'baz': 24}} _get_stats_by_database.side_effect = side_effect config = get_collector_config( 'PgbouncerCollector', { 'instances': { 'alpha': { 'host': '127.0.0.1', 'port': '6432', }, 'beta': { 'host': 'localhost', 'port': '6433', }, } }) collector = PgbouncerCollector(config, None) collector.collect() self.assertPublished(publish, 'alpha.foo.bar', 42) self.assertPublished(publish, 'beta.foo.baz', 24)
def test_override_user_password(self, _get_stats_by_database): _get_stats_by_database.return_value = {} config = get_collector_config( "PgbouncerCollector", { "instances": { "test1": {"host": "127.0.0.1", "port": "6433", "password": "******"}, "test2": {"host": "127.0.0.2", "port": "6432", "user": "******"}, } }, ) collector = PgbouncerCollector(config, None) collector.collect() _get_stats_by_database.assert_any_call("127.0.0.1", "6433", "postgres", "foobar") _get_stats_by_database.assert_any_call("127.0.0.2", "6432", "pgbouncer", "")
def test_instance_names(self, publish, _get_stats_by_database): def side_effect(host, port, user, password): if (host, port) == ("127.0.0.1", "6432"): return {"foo": {"bar": 42}} elif (host, port) == ("localhost", "6433"): return {"foo": {"baz": 24}} _get_stats_by_database.side_effect = side_effect config = get_collector_config( "PgbouncerCollector", { "instances": { "alpha": {"host": "127.0.0.1", "port": "6432"}, "beta": {"host": "localhost", "port": "6433"}, } }, ) collector = PgbouncerCollector(config, None) collector.collect() self.assertPublished(publish, "alpha.foo.bar", 42) self.assertPublished(publish, "beta.foo.baz", 24)
def test_override_user_password(self, _get_stats_by_database): _get_stats_by_database.return_value = {} config = get_collector_config('PgbouncerCollector', { 'instances': { 'test1': { 'host': '127.0.0.1', 'port': '6433', 'password': '******', }, 'test2': { 'host': '127.0.0.2', 'port': '6432', 'user': '******', } } }) collector = PgbouncerCollector(config, None) collector.collect() _get_stats_by_database.assert_any_call( '127.0.0.1', '6433', 'postgres', 'foobar') _get_stats_by_database.assert_any_call( '127.0.0.2', '6432', 'pgbouncer', '')
def setUp(self): config = get_collector_config('PgbouncerCollector', {}) self.collector = PgbouncerCollector(config, None)
class TestPgbouncerCollector(CollectorTestCase): def setUp(self): config = get_collector_config('PgbouncerCollector', {}) self.collector = PgbouncerCollector(config, None) def test_import(self): self.assertTrue(PgbouncerCollector) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, '_get_stats_by_database') @patch.object(PgbouncerCollector, 'publish') def test_default(self, publish, _get_stats_by_database): _get_stats_by_database.return_value = {'foo': {'bar': 42}} self.collector.collect() _get_stats_by_database.assert_called_with( 'localhost', '6432', 'postgres', '') self.assertPublished(publish, 'default.foo.bar', 42) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, '_get_stats_by_database') @patch.object(PgbouncerCollector, 'publish') def test_instance_names(self, publish, _get_stats_by_database): def side_effect(host, port, user, password): if (host, port) == ('127.0.0.1', '6432'): return {'foo': {'bar': 42}} elif (host, port) == ('localhost', '6433'): return {'foo': {'baz': 24}} _get_stats_by_database.side_effect = side_effect config = get_collector_config('PgbouncerCollector', { 'instances': { 'alpha': { 'host': '127.0.0.1', 'port': '6432', }, 'beta': { 'host': 'localhost', 'port': '6433', }, } }) collector = PgbouncerCollector(config, None) collector.collect() self.assertPublished(publish, 'alpha.foo.bar', 42) self.assertPublished(publish, 'beta.foo.baz', 24) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, '_get_stats_by_database') def test_override_user_password(self, _get_stats_by_database): _get_stats_by_database.return_value = {} config = get_collector_config('PgbouncerCollector', { 'instances': { 'test1': { 'host': '127.0.0.1', 'port': '6433', 'password': '******', }, 'test2': { 'host': '127.0.0.2', 'port': '6432', 'user': '******', } } }) collector = PgbouncerCollector(config, None) collector.collect() _get_stats_by_database.assert_any_call( '127.0.0.1', '6433', 'postgres', 'foobar') _get_stats_by_database.assert_any_call( '127.0.0.2', '6432', 'pgbouncer', '')
class TestPgbouncerCollector(CollectorTestCase): def setUp(self): config = get_collector_config("PgbouncerCollector", {}) self.collector = PgbouncerCollector(config, None) def test_import(self): self.assertTrue(PgbouncerCollector) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, "_get_stats_by_database") @patch.object(PgbouncerCollector, "publish") def test_default(self, publish, _get_stats_by_database): _get_stats_by_database.return_value = {"foo": {"bar": 42}} self.collector.collect() _get_stats_by_database.assert_called_with("localhost", "6432", "postgres", "") self.assertPublished(publish, "default.foo.bar", 42) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, "_get_stats_by_database") @patch.object(PgbouncerCollector, "publish") def test_instance_names(self, publish, _get_stats_by_database): def side_effect(host, port, user, password): if (host, port) == ("127.0.0.1", "6432"): return {"foo": {"bar": 42}} elif (host, port) == ("localhost", "6433"): return {"foo": {"baz": 24}} _get_stats_by_database.side_effect = side_effect config = get_collector_config( "PgbouncerCollector", { "instances": { "alpha": {"host": "127.0.0.1", "port": "6432"}, "beta": {"host": "localhost", "port": "6433"}, } }, ) collector = PgbouncerCollector(config, None) collector.collect() self.assertPublished(publish, "alpha.foo.bar", 42) self.assertPublished(publish, "beta.foo.baz", 24) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, "_get_stats_by_database") def test_override_user_password(self, _get_stats_by_database): _get_stats_by_database.return_value = {} config = get_collector_config( "PgbouncerCollector", { "instances": { "test1": {"host": "127.0.0.1", "port": "6433", "password": "******"}, "test2": {"host": "127.0.0.2", "port": "6432", "user": "******"}, } }, ) collector = PgbouncerCollector(config, None) collector.collect() _get_stats_by_database.assert_any_call("127.0.0.1", "6433", "postgres", "foobar") _get_stats_by_database.assert_any_call("127.0.0.2", "6432", "pgbouncer", "")
class TestPgbouncerCollector(CollectorTestCase): def setUp(self): config = get_collector_config('PgbouncerCollector', {}) self.collector = PgbouncerCollector(config, None) def test_import(self): self.assertTrue(PgbouncerCollector) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, '_get_stats_by_database') @patch.object(PgbouncerCollector, 'publish') def test_default(self, publish, _get_stats_by_database): _get_stats_by_database.return_value = {'foo': {'bar': 42}} self.collector.collect() _get_stats_by_database.assert_called_with('localhost', '6432', 'postgres', '') self.assertPublished(publish, 'default.foo.bar', 42) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, '_get_stats_by_database') @patch.object(PgbouncerCollector, 'publish') def test_instance_names(self, publish, _get_stats_by_database): def side_effect(host, port, user, password): if (host, port) == ('127.0.0.1', '6432'): return {'foo': {'bar': 42}} elif (host, port) == ('localhost', '6433'): return {'foo': {'baz': 24}} _get_stats_by_database.side_effect = side_effect config = get_collector_config( 'PgbouncerCollector', { 'instances': { 'alpha': { 'host': '127.0.0.1', 'port': '6432', }, 'beta': { 'host': 'localhost', 'port': '6433', }, } }) collector = PgbouncerCollector(config, None) collector.collect() self.assertPublished(publish, 'alpha.foo.bar', 42) self.assertPublished(publish, 'beta.foo.baz', 24) @run_only_if_psycopg2_is_available @patch.object(PgbouncerCollector, '_get_stats_by_database') def test_override_user_password(self, _get_stats_by_database): _get_stats_by_database.return_value = {} config = get_collector_config( 'PgbouncerCollector', { 'instances': { 'test1': { 'host': '127.0.0.1', 'port': '6433', 'password': '******', }, 'test2': { 'host': '127.0.0.2', 'port': '6432', 'user': '******', } } }) collector = PgbouncerCollector(config, None) collector.collect() _get_stats_by_database.assert_any_call('127.0.0.1', '6433', 'postgres', 'foobar') _get_stats_by_database.assert_any_call('127.0.0.2', '6432', 'pgbouncer', '')