def test_check(instance, aggregator, datadog_agent):
# add some stats
connection = psycopg2.connect(
host=common.HOST,
port=common.PORT,
user=common.USER,
password=common.PASS,
database=common.DB,
connect_timeout=1,
)
connection.set_isolation_level(
psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
cur = connection.cursor()
cur.execute('SELECT * FROM persons;')
# run the check
check = PgBouncer('pgbouncer', {}, [instance])
check.check_id = 'test:123'
check.check(instance)
version = _get_version_from_env()
assert_metric_coverage(version, aggregator)
patch = '1' if version[1] == '8' else '2'
version_metadata = {
'version.raw': '.'.join(version) + '.' + patch,
'version.scheme': 'semver',
'version.major': version[0],
'version.minor': version[1],
'version.patch': patch,
}
datadog_agent.assert_metadata('test:123', version_metadata)
def test_critical_service_check(instance, aggregator):
instance['port'] = '123' # Bad port
check = PgBouncer('pgbouncer', {}, [instance])
with pytest.raises(Exception):
check.check(instance)
aggregator.assert_service_check(PgBouncer.SERVICE_CHECK_NAME,
status=PgBouncer.CRITICAL)
def test_check(instance, aggregator, datadog_agent, dd_run_check):
# add some stats
connection = psycopg2.connect(
host=common.HOST,
port=common.PORT,
user=common.USER,
password=common.PASS,
database=common.DB,
connect_timeout=1,
)
connection.set_isolation_level(
psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
cur = connection.cursor()
cur.execute('SELECT * FROM persons;')
# run the check
check = PgBouncer('pgbouncer', {}, [instance])
check.check_id = 'test:123'
dd_run_check(check)
env_version = common.get_version_from_env()
assert_metric_coverage(env_version, aggregator)
version_metadata = {
'version.raw': str(env_version),
'version.scheme': 'semver',
'version.major': str(env_version.major),
'version.minor': str(env_version.minor),
'version.patch': str(env_version.micro),
}
datadog_agent.assert_metadata('test:123', version_metadata)
def test_check(instance, aggregator):
# add some stats
connection = psycopg2.connect(
host=common.HOST,
port=common.PORT,
user=common.USER,
password=common.PASS,
database=common.DB,
connect_timeout=1,
)
connection.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
cur = connection.cursor()
cur.execute('SELECT * FROM persons;')
# run the check
check = PgBouncer('pgbouncer', {}, {})
check.check(instance)
aggregator.assert_metric('pgbouncer.pools.cl_active')
aggregator.assert_metric('pgbouncer.pools.cl_waiting')
aggregator.assert_metric('pgbouncer.pools.sv_active')
aggregator.assert_metric('pgbouncer.pools.sv_idle')
aggregator.assert_metric('pgbouncer.pools.sv_used')
aggregator.assert_metric('pgbouncer.pools.sv_tested')
aggregator.assert_metric('pgbouncer.pools.sv_login')
aggregator.assert_metric('pgbouncer.pools.maxwait')
aggregator.assert_metric('pgbouncer.stats.avg_recv')
aggregator.assert_metric('pgbouncer.stats.avg_sent')
# get PgBouncer version
ver = common.get_version()
assert ver is not None, "Unable to retrieve PgBouncer version"
pgbouncer_pre18 = ver[1] < 8
if pgbouncer_pre18:
aggregator.assert_metric('pgbouncer.stats.avg_req')
aggregator.assert_metric('pgbouncer.stats.avg_query')
aggregator.assert_metric('pgbouncer.stats.requests_per_second')
else:
aggregator.assert_metric('pgbouncer.stats.avg_transaction_time')
aggregator.assert_metric('pgbouncer.stats.avg_query_time')
aggregator.assert_metric('pgbouncer.stats.avg_transaction_count')
aggregator.assert_metric('pgbouncer.stats.avg_query_count')
aggregator.assert_metric('pgbouncer.stats.queries_per_second')
aggregator.assert_metric('pgbouncer.stats.transactions_per_second')
aggregator.assert_metric('pgbouncer.stats.total_transaction_time')
aggregator.assert_metric('pgbouncer.stats.total_query_time')
aggregator.assert_metric('pgbouncer.stats.bytes_received_per_second')
aggregator.assert_metric('pgbouncer.stats.bytes_sent_per_second')
aggregator.assert_metric('pgbouncer.databases.pool_size')
aggregator.assert_metric('pgbouncer.databases.max_connections')
aggregator.assert_metric('pgbouncer.databases.current_connections')
# Service checks
sc_tags = ['host:{}'.format(common.HOST), 'port:{}'.format(common.PORT), 'db:pgbouncer', 'optional:tag1']
aggregator.assert_service_check('pgbouncer.can_connect', status=PgBouncer.OK, tags=sc_tags)
def test_check_with_url(instance_with_url, aggregator, datadog_agent):
# run the check
check = PgBouncer('pgbouncer', {}, [instance_with_url])
check.check_id = 'test:123'
check.check(instance_with_url)
env_version = common.get_version_from_env()
assert_metric_coverage(env_version, aggregator)
version_metadata = {
'version.raw': str(env_version),
'version.scheme': 'semver',
'version.major': str(env_version.major),
'version.minor': str(env_version.minor),
'version.patch': str(env_version.micro),
}
datadog_agent.assert_metadata('test:123', version_metadata)
def test_check(instance, aggregator):
# add some stats
connection = psycopg2.connect(
host=common.HOST,
port=common.PORT,
user=common.USER,
password=common.PASS,
database=common.DB,
connect_timeout=1,
)
connection.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
cur = connection.cursor()
cur.execute('SELECT * FROM persons;')
# run the check
check = PgBouncer('pgbouncer', {}, [instance])
check.check(instance)
assert_metric_coverage(aggregator)
def test_config_missing_user(instance):
with pytest.raises(ConfigurationError):
del instance['username']
PgBouncer('pgbouncer', {}, [instance])