mirror of
https://github.com/chatmail/relay.git
synced 2026-05-16 03:24:48 +00:00
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
import pytest
|
|
import threading
|
|
import queue
|
|
|
|
|
|
def test_login_basic_functioning(imap_or_smtp, gencreds, lp):
|
|
"""Test a) that an initial login creates a user automatically
|
|
and b) verify we can also login a second time with the same password
|
|
and c) that using a different password fails the login."""
|
|
user, password = gencreds()
|
|
lp.sec(f"login first time with {user} {password}")
|
|
imap_or_smtp.connect()
|
|
imap_or_smtp.login(user, password)
|
|
lp.indent("success")
|
|
|
|
lp.sec(f"reconnect and login second time {user} {password}")
|
|
imap_or_smtp.connect()
|
|
imap_or_smtp.login(user, password)
|
|
imap_or_smtp.connect()
|
|
lp.sec("success")
|
|
|
|
lp.sec(f"reconnect and verify wrong password fails {user} ")
|
|
imap_or_smtp.connect()
|
|
with pytest.raises(imap_or_smtp.AuthError):
|
|
imap_or_smtp.login(user, password + "wrong")
|
|
|
|
lp.sec("creating users with a short password is not allowed")
|
|
user, _password = gencreds()
|
|
with pytest.raises(imap_or_smtp.AuthError):
|
|
imap_or_smtp.login(user, "admin")
|
|
|
|
|
|
def test_login_same_password(imap_or_smtp, gencreds):
|
|
"""Test two different users logging in with the same password
|
|
to ensure that authentication process does not confuse the users
|
|
by using only the password hash as a key.
|
|
"""
|
|
user1, password1 = gencreds()
|
|
user2, _ = gencreds()
|
|
imap_or_smtp.connect()
|
|
imap_or_smtp.login(user1, password1)
|
|
imap_or_smtp.connect()
|
|
imap_or_smtp.login(user2, password1)
|
|
|
|
|
|
def test_concurrent_logins_same_account(
|
|
make_imap_connection, make_smtp_connection, gencreds
|
|
):
|
|
"""Test concurrent smtp and imap logins
|
|
and check remote server succeeds on each connection.
|
|
"""
|
|
user1, password1 = gencreds()
|
|
login_results = queue.Queue()
|
|
|
|
def login_smtp_imap(smtp, imap):
|
|
try:
|
|
imap.login(user1, password1)
|
|
except Exception:
|
|
login_results.put(False)
|
|
else:
|
|
login_results.put(True)
|
|
|
|
conns = [(make_smtp_connection(), make_imap_connection()) for i in range(10)]
|
|
|
|
for args in conns:
|
|
thread = threading.Thread(target=login_smtp_imap, args=args, daemon=True)
|
|
thread.start()
|
|
|
|
for _ in conns:
|
|
assert login_results.get()
|