mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
fa43f6bc3e
This change utilizes pytest-xdist to perform a multiproc test
run and reworks aurweb.db's code. We no longer use a global
engine, session or Session, but we now use a memo of engines
and sessions as they are requested, based on the PYTEST_CURRENT_TEST
environment variable, which is available during testing.
Additionally, this change strips several SQLite components
out of the Python code-base.
SQLite is still compatible with PHP and sharness tests, but
not with our FastAPI implementation.
More changes:
------------
- Remove use of aurweb.db.session global in other code.
- Use new aurweb.db.name() dynamic db name function in env.py.
- Added 'addopts' to pytest.ini which utilizes multiprocessing.
- Highly recommended to leave this be or modify `-n auto` to
`-n {cpu_threads}` where cpu_threads is at least 2.
Signed-off-by: Kevin Morris <kevr@0cost.org>
55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
import pytest
|
|
|
|
from aurweb import db
|
|
from aurweb.models.account_type import USER_ID
|
|
from aurweb.models.ssh_pub_key import SSHPubKey, get_fingerprint
|
|
from aurweb.models.user import User
|
|
|
|
TEST_SSH_PUBKEY = """
|
|
ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQCycoCi5yGCvSclH2wmNBUuwsYEzRZZBJaQquRc4ysl+Tg+/jiDkR3Zn9fIznC4KnFoyrIHzkKuePZ3bNDYwkZxkJKoWBCh4hXKDXSm87FMN0+VDC+1QxF/z0XaAGr/P6f4XukabyddypBdnHcZiplbw+YOSqcAE2TCqOlSXwNMOcF9U89UsR/Q9i9I52hlvU0q8+fZVGhou1KCowFSnHYtrr5KYJ04CXkJ13DkVf3+pjQWyrByvBcf1hGEaczlgfobrrv/y96jDhgfXucxliNKLdufDPPkii3LhhsNcDmmI1VZ3v0irKvd9WZuauqloobY84zEFcDTyjn0hxGjVeYFejm4fBnvjga0yZXORuWksdNfXWLDxFk6MDDd1jF0ExRbP+OxDuU4IVyIuDL7S3cnbf2YjGhkms/8voYT2OBE7FwNlfv98Kr0NUp51zpf55Arxn9j0Rz9xTA7FiODQgCn6iQ0SDtzUNL0IKTCw26xJY5gzMxbfpvzPQGeulx/ioM= kevr@volcano
|
|
"""
|
|
|
|
user = ssh_pub_key = None
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup(db_test):
|
|
global user, ssh_pub_key
|
|
|
|
with db.begin():
|
|
user = db.create(User, Username="test", Email="test@example.org",
|
|
RealName="Test User", Passwd="testPassword",
|
|
AccountTypeID=USER_ID)
|
|
|
|
with db.begin():
|
|
ssh_pub_key = db.create(SSHPubKey, UserID=user.ID,
|
|
Fingerprint="testFingerprint",
|
|
PubKey="testPubKey")
|
|
|
|
|
|
def test_ssh_pub_key():
|
|
assert ssh_pub_key.UserID == user.ID
|
|
assert ssh_pub_key.User == user
|
|
assert ssh_pub_key.Fingerprint == "testFingerprint"
|
|
assert ssh_pub_key.PubKey == "testPubKey"
|
|
|
|
|
|
def test_ssh_pub_key_cs():
|
|
""" Test case sensitivity of the database table. """
|
|
with db.begin():
|
|
ssh_pub_key_cs = db.create(SSHPubKey, UserID=user.ID,
|
|
Fingerprint="TESTFINGERPRINT",
|
|
PubKey="TESTPUBKEY")
|
|
|
|
assert ssh_pub_key_cs.Fingerprint == "TESTFINGERPRINT"
|
|
assert ssh_pub_key_cs.PubKey == "TESTPUBKEY"
|
|
assert ssh_pub_key.Fingerprint == "testFingerprint"
|
|
assert ssh_pub_key.PubKey == "testPubKey"
|
|
|
|
|
|
def test_ssh_pub_key_fingerprint():
|
|
assert get_fingerprint(TEST_SSH_PUBKEY) is not None
|
|
|
|
|
|
def test_ssh_pub_key_invalid_fingerprint():
|
|
assert get_fingerprint("ssh-rsa fake and invalid") is None
|