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("forgeaddr", ["internal", "someone@example.org"])
def test_reject_forged_from(cmsetup, mailgen, lp, remote, forgeaddr):
    user1, user3 = cmsetup.gen_users(2)

    lp.sec("send encrypted message with forged from")
    print("envelope_from", user1.addr)
    if forgeaddr == "internal":
        addr_to_forge = cmsetup.gen_users(1)[0].addr
    else:
        addr_to_forge = "someone@example.org"

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

    lp.sec("Send forged mail and check remote postfix lmtp processing result")
    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")

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