cat ../ansible-role-pgbouncer/molecule/default/tests/test_default.py
import os

import testinfra.utils.ansible_runner

testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
    os.environ['MOLECULE_INVENTORY_FILE']).get_hosts('all')


def test_postgres_package(host):
    pkg = host.package('pgbouncer')
    assert pkg.is_installed


def test_postgres_service(host):
    pgbouncer_service = host.service('pgbouncer')
    assert pgbouncer_service.is_enabled
    assert pgbouncer_service.is_running


def test_config_files(host):
    pgbouncer_content = host.file('/etc/pgbouncer/pgbouncer.ini').content

    assert 'dbname=test_base1 user=postgres_test_user1 \
password=testpw1 host=localhost pool_size=50' in pgbouncer_content
    assert 'dbname=test_base2 user=postgres_test_user2 \
password=testpw2 host=localhost pool_size=50' in pgbouncer_content

    userlist_content = host.file('/etc/pgbouncer/userlist.txt').content

    assert '"pgbouncer_test_user1" "pgbouncer_testpw1"' in userlist_content
    assert '"pgbouncer_test_user2" "pgbouncer_testpw2"' in userlist_content