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