79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
import pytest
|
|
import os
|
|
|
|
from click.testing import CliRunner
|
|
from ceo.cli import cli
|
|
|
|
from mysql.connector import connect
|
|
from mysql.connector.errors import ProgrammingError
|
|
|
|
|
|
def mysql_attempt_connection(host, username, password):
|
|
with connect(
|
|
host=host,
|
|
user=username,
|
|
password=password,
|
|
) as con, con.cursor() as cur:
|
|
cur.execute("SHOW DATABASES")
|
|
response = cur.fetchall()
|
|
assert len(response) == 2
|
|
|
|
with pytest.raises(ProgrammingError):
|
|
cur.execute("CREATE DATABASE new_db")
|
|
|
|
|
|
def test_mysql(cli_setup, cfg, ldap_user):
|
|
runner = CliRunner()
|
|
|
|
username = ldap_user.uid
|
|
host = cfg.get("mysql_host")
|
|
info_file_path = os.path.join(ldap_user.home_directory, "ceo-mysql-info")
|
|
assert not os.path.isfile(info_file_path)
|
|
|
|
# create database for user
|
|
result = runner.invoke(cli, ['mysql', '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"""MySQL 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
|
|
mysql_attempt_connection(host, username, passwd)
|
|
|
|
# perform password reset for user
|
|
result = runner.invoke(cli, ['mysql', '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
|
|
mysql_attempt_connection(host, username, new_passwd)
|
|
|
|
# delete database and file
|
|
result = runner.invoke(cli, ['mysql', 'delete', username], input="y\n")
|
|
assert result.exit_code == 0
|
|
|
|
# user should be deleted
|
|
with pytest.raises(ProgrammingError):
|
|
mysql_attempt_connection(host, username, passwd)
|
|
|
|
os.remove(info_file_path)
|