40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
import pytest
|
|
from sachet.server.commands import create_db, drop_db, create_user, delete_user
|
|
from sachet.server import app, db
|
|
from sqlalchemy import inspect
|
|
|
|
from sachet.server.models import User
|
|
|
|
def test_db(flask_app_bare, cli):
|
|
"""Test the CLI's ability to create and drop the DB."""
|
|
# make tables
|
|
result = cli.invoke(create_db)
|
|
assert result.exit_code == 0
|
|
assert "users" in inspect(db.engine).get_table_names()
|
|
|
|
# tear down
|
|
result = cli.invoke(drop_db, ["--yes"])
|
|
assert result.exit_code == 0
|
|
assert "users" not in inspect(db.engine).get_table_names()
|
|
|
|
|
|
def test_user(client, cli):
|
|
"""Test the CLI's ability to create then delete a user."""
|
|
# create user
|
|
result = cli.invoke(create_user, ["--username", "jeff", "--password", "1234"])
|
|
assert result.exit_code == 0
|
|
assert User.query.filter_by(username="jeff").first() is not None
|
|
|
|
# create duplicate user
|
|
result = cli.invoke(create_user, ["--username", "jeff", "--password", "1234"])
|
|
assert isinstance(result.exception, KeyError)
|
|
|
|
# delete user
|
|
result = cli.invoke(delete_user, ["--yes", "jeff"])
|
|
assert result.exit_code == 0
|
|
assert User.query.filter_by(username="jeff").first() is None
|
|
|
|
# delete non-existent user
|
|
result = cli.invoke(delete_user, ["--yes", "jeff"])
|
|
assert isinstance(result.exception, KeyError)
|