from yaml import load, SafeLoader from os import environ import pytest from imaplib2 import IMAP4_SSL from smtplib import SMTP, SMTPNotSupportedError, SMTPAuthenticationError conf_file = f'{environ["MACHINE_TEST_DIR"]}/imap.yml' with open(conf_file) as yaml: data = load(yaml, Loader=SafeLoader) parameters = (('user', data['username'], data['password']), ('family', data['username_family'], data['password_family']), ) def get_msg(username, msg='MESSAGE'): return f'From: {username}\r\nTo: {username}\r\n\r\nSubject: TEST\r\n{msg}\r\n' @pytest.mark.parametrize('typ, username, password', parameters) def test_imap_wrong_password(typ, username, password): imap = IMAP4_SSL(data['address']) try: imap.LOGIN(username, 'b') except: pass else: raise Exception('wrong login !') @pytest.mark.parametrize('typ, username, password', parameters) def test_imap_migration(typ, username, password): msg = get_msg(username, 'MIGRATION') if 'FIRST_RUN' in environ: smtp = SMTP(data['address'], '587') smtp.starttls() smtp.login(username, password) smtp.sendmail(username, username, msg) smtp.quit() imap = IMAP4_SSL(data['address']) imap.LOGIN(username, password) imap.SELECT(readonly=True) typ, req = imap.SEARCH(None, 'ALL') assert typ == 'OK' assert len(req) == 1 assert req[0] == b'1' field = imap.FETCH('1', '(RFC822)') assert field[0] == 'OK' assert field[1][-2][-1].decode().endswith(msg) imap.CLOSE() imap.LOGOUT() @pytest.mark.parametrize('typ, username, password', parameters) def test_smtp_no_tls(typ, username, password): smtp = SMTP(data['address'], '587') try: smtp.login(username, password) raise Exception('no tls!') except SMTPNotSupportedError: pass @pytest.mark.parametrize('typ, username, password', parameters) def test_smtp_wrong_passwd(typ, username, password): smtp = SMTP(data['address'], '587') smtp.starttls() try: smtp.login(username, 'a') raise Exception('wrong password!') except SMTPAuthenticationError: pass smtp.quit() @pytest.mark.parametrize('typ, username, password', parameters) def test_smtp_login(typ, username, password): smtp = SMTP(data['address'], '587') smtp.starttls() smtp.login(username, password) smtp.quit() @pytest.mark.parametrize('typ, username, password', parameters) def test_smtp_sendmail(typ, username, password): smtp = SMTP(data['address'], '587') smtp.starttls() smtp.login(username, password) smtp.sendmail(username, username, get_msg(username)) smtp.quit() @pytest.mark.parametrize('typ, username, password', parameters) def test_imap_read_mail(typ, username, password): imap = IMAP4_SSL(data['address']) imap.LOGIN(username, password) imap.SELECT(readonly=True) typ, req = imap.SEARCH(None, 'ALL') assert typ == 'OK' assert len(req) == 1 msg = get_msg(username) msg_no = req[0].split() assert len(msg_no) == 2 for num in msg_no[1:]: field = imap.FETCH(num, '(RFC822)') assert field[0] == 'OK' assert field[1][-2][-1].decode().endswith(msg) imap.CLOSE() imap.LOGOUT() @pytest.mark.parametrize('typ, username, password', parameters) def test_imap_delete_mail(typ, username, password): imap = IMAP4_SSL(data['address']) imap.LOGIN(username, password) imap.SELECT() typ, req = imap.SEARCH(None, 'ALL') msg_no = req[0].split() for num in msg_no[1:]: ret = imap.store(num, '+FLAGS', '\\Deleted') assert ret[0] == 'OK', f'error when deleting mail: {ret}' imap.expunge() imap.CLOSE() imap.LOGOUT()