mirror of https://gitlab.com/litecord/litecord.git
132 lines
3.8 KiB
Python
132 lines
3.8 KiB
Python
"""
|
|
|
|
Litecord
|
|
Copyright (C) 2018-2019 Luna Mendes
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, version 3 of the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
|
|
import secrets
|
|
import sys
|
|
import os
|
|
|
|
import socket
|
|
import pytest
|
|
|
|
# this is very hacky.
|
|
sys.path.append(os.getcwd())
|
|
|
|
from tests.common import email, TestClient
|
|
|
|
from run import app as main_app, set_blueprints
|
|
|
|
from litecord.auth import create_user
|
|
from litecord.enums import UserFlags
|
|
from litecord.blueprints.auth import make_token
|
|
from litecord.blueprints.users import delete_user
|
|
|
|
|
|
@pytest.fixture(name='app')
|
|
def _test_app(unused_tcp_port, event_loop):
|
|
set_blueprints(main_app)
|
|
main_app.config['_testing'] = True
|
|
|
|
# reassign an unused tcp port for websockets
|
|
# since the config might give a used one.
|
|
ws_port = unused_tcp_port
|
|
|
|
main_app.config['IS_SSL'] = False
|
|
main_app.config['WS_PORT'] = ws_port
|
|
main_app.config['WEBSOCKET_URL'] = f'localhost:{ws_port}'
|
|
|
|
# testing user creations requires hardcoding this to true
|
|
# on testing
|
|
main_app.config['REGISTRATIONS'] = True
|
|
|
|
# make sure we're calling the before_serving hooks
|
|
event_loop.run_until_complete(main_app.startup())
|
|
|
|
# https://docs.pytest.org/en/latest/fixture.html#fixture-finalization-executing-teardown-code
|
|
yield main_app
|
|
|
|
# properly teardown
|
|
event_loop.run_until_complete(main_app.shutdown())
|
|
|
|
|
|
@pytest.fixture(name='test_cli')
|
|
def _test_cli(app):
|
|
"""Give a test client."""
|
|
return app.test_client()
|
|
|
|
# code shamelessly stolen from my elixire mr
|
|
# https://gitlab.com/elixire/elixire/merge_requests/52
|
|
async def _user_fixture_setup(app):
|
|
username = secrets.token_hex(6)
|
|
password = secrets.token_hex(6)
|
|
user_email = email()
|
|
|
|
user_id, pwd_hash = await create_user(
|
|
username, user_email, password, app.db, app.loop)
|
|
|
|
# generate a token for api access
|
|
user_token = make_token(user_id, pwd_hash)
|
|
|
|
return {'id': user_id, 'token': user_token,
|
|
'email': user_email, 'username': username,
|
|
'password': password}
|
|
|
|
|
|
async def _user_fixture_teardown(app, udata: dict):
|
|
await delete_user(udata['id'], db=app.db)
|
|
|
|
|
|
@pytest.fixture(name='test_user')
|
|
async def test_user_fixture(app):
|
|
"""Yield a randomly generated test user."""
|
|
udata = await _user_fixture_setup(app)
|
|
yield udata
|
|
await _user_fixture_teardown(app, udata)
|
|
|
|
|
|
@pytest.fixture
|
|
async def test_cli_user(test_cli, test_user):
|
|
"""Yield a TestClient instance that contains a randomly generated
|
|
user."""
|
|
yield TestClient(test_cli, test_user)
|
|
|
|
|
|
@pytest.fixture
|
|
async def test_cli_staff(test_cli):
|
|
"""Yield a TestClient with a staff user."""
|
|
# This does not use the test_user because if a given test uses both
|
|
# test_cli_user and test_cli_admin, test_cli_admin will just point to that
|
|
# same test_cli_user, which isn't acceptable.
|
|
app = test_cli.app
|
|
test_user = await _user_fixture_setup(app)
|
|
user_id = test_user['id']
|
|
|
|
# copied from manage.cmd.users.set_user_staff.
|
|
old_flags = await app.db.fetchval("""
|
|
SELECT flags FROM users WHERE id = $1
|
|
""", user_id)
|
|
|
|
new_flags = old_flags | UserFlags.staff
|
|
|
|
await app.db.execute("""
|
|
UPDATE users SET flags = $1 WHERE id = $2
|
|
""", new_flags, user_id)
|
|
|
|
yield TestClient(test_cli, test_user)
|
|
await _user_fixture_teardown(test_cli.app, test_user)
|