from os import environ
from mariadb import ConnectionPool, PoolError

host = environ.get("DB_HOST", "localhost")
port = int(environ.get("DB_PORT", "3306"))
user = environ.get("DB_USER", "root")
password = environ["DB_PASSWORD"]
database = environ.get("DB_NAME", "")

POOL_SIZE = 3

pool = ConnectionPool(
    host=host,
    port=port,
    user=user,
    password=password,
    database=database,
    pool_name="Test PoolError",
    pool_size=POOL_SIZE,
)

for k in range(POOL_SIZE):
    connection = pool.get_connection()

try:
    pool.get_connection()
except PoolError:
    print("PoolError raised")
    exit(0)

print("PoolError not raised")
exit(1)