80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
import pytest, os
|
|
|
|
from click.testing import CliRunner
|
|
from ceo.cli import cli
|
|
|
|
from psycopg2 import connect, OperationalError, ProgrammingError
|
|
|
|
|
|
def psql_attempt_connection(host, username, password):
|
|
con = connect(
|
|
host=host,
|
|
user=username,
|
|
password=password,
|
|
)
|
|
con.autocommit = True
|
|
with con.cursor() as cur:
|
|
cur.execute("SELECT datname FROM pg_database")
|
|
response = cur.fetchall()
|
|
# 3 of the 4 are postgres, template0, template1
|
|
assert len(response) == 4
|
|
with pytest.raises(ProgrammingError):
|
|
cur.execute("CREATE DATABASE new_db")
|
|
con.close()
|
|
|
|
|
|
def test_postgresql(cli_setup, cfg, ldap_user):
|
|
runner = CliRunner()
|
|
|
|
username = ldap_user.uid
|
|
host = cfg.get("postgresql_host")
|
|
info_file_path = os.path.join(ldap_user.home_directory, "ceo-psql-info")
|
|
assert not os.path.isfile(info_file_path)
|
|
|
|
# create database for user
|
|
result = runner.invoke(cli, ['postgresql', 'create', username])
|
|
assert result.exit_code == 0
|
|
assert os.path.isfile(info_file_path)
|
|
|
|
response_arr = result.output.split()
|
|
passwd = response_arr[response_arr.index("Password:") + 1]
|
|
with open(info_file_path, 'r') as file:
|
|
old_info = file.read()
|
|
|
|
expected = f"""PostgreSQL database created
|
|
|
|
Connection Information:
|
|
|
|
Database: {username}
|
|
Username: {username}
|
|
Password: {passwd}
|
|
Host: {host}
|
|
|
|
Settings and more info has been written to {info_file_path}"""
|
|
|
|
assert result.output == expected
|
|
psql_attempt_connection(host, username, passwd)
|
|
|
|
# perform password reset for user
|
|
result = runner.invoke(cli, ['postgresql', 'pwreset', username], input="y\n")
|
|
assert result.exit_code == 0
|
|
|
|
response_arr = result.output.split()
|
|
new_passwd = response_arr[response_arr.index("Password:") + 1]
|
|
with open(info_file_path, 'r') as file:
|
|
new_info = file.read()
|
|
|
|
assert new_passwd != passwd
|
|
assert old_info != new_info
|
|
psql_attempt_connection(host, username, new_passwd)
|
|
|
|
# delete database and file
|
|
result = runner.invoke(cli, ['postgresql', 'delete', username], input="y\n")
|
|
assert result.exit_code == 0
|
|
|
|
# user should be deleted
|
|
with pytest.raises(OperationalError):
|
|
psql_attempt_connection(host, username, passwd)
|
|
|
|
os.remove(info_file_path)
|