import smtplib
import pytest


def test_remote(remote, imap_or_smtp):
    lineproducer = remote.iter_output(imap_or_smtp.logcmd)
    imap_or_smtp.connect()
    assert imap_or_smtp.name in next(lineproducer)


def test_use_two_chatmailservers(cmfactory, maildomain2):
    ac1 = cmfactory.new_online_configuring_account(cache=False)
    cmfactory.switch_maildomain(maildomain2)
    ac2 = cmfactory.new_online_configuring_account(cache=False)
    cmfactory.bring_accounts_online()
    cmfactory.get_accepted_chat(ac1, ac2)
    domain1 = ac1.get_config("addr").split("@")[1]
    domain2 = ac2.get_config("addr").split("@")[1]
    assert domain1 != domain2


@pytest.mark.parametrize("internal", [True, False])
def test_reject_forged_from(cmsetup, mailgen, lp, remote, internal):
    user1, user2, user3 = cmsetup.gen_users(3)

    lp.sec("send encrypted message with forged from")
    print("envelope_from", user1.addr)
    if internal:
        user_to_forge = user2.addr
    else:
        user_to_forge = "someone@example.org"

    print("message to inject:")
    msg = mailgen.get_encrypted(from_addr=user_to_forge, to_addr=user3.addr)
    for line in msg.split("\n")[:4]:
        print(f"   {line}")

    remote_log = remote.iter_output("journalctl -t postfix/lmtp")
    user1.smtp.sendmail(from_addr=user1.addr, to_addrs=[user3.addr], msg=msg)

    for line in remote_log:
        # print(line)
        if "500 invalid from" in line and user3.addr in line:
            break
    else:
        pytest.fail("remote postfix/filtermail failed to reject message")

    # also check that the forging-user got a non-delivery notice
    for message in user1.imap.fetch_all_messages():
        if "Invalid FROM" in message and user_to_forge in message:
            return
    pytest.fail(f"forged From={user_to_forge} did not cause non-delivery notice")