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(flask_app, 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)