mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-31 18:22:22 +01:00
67 lines
2.9 KiB
Python
67 lines
2.9 KiB
Python
"""The bounded arrival batch is displayed by Date, rather than arrival ID."""
|
|
from contextlib import contextmanager
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
from tests.cli.test_mail_cli_read_empty_fetch import _load_mail_cli
|
|
|
|
|
|
@pytest.mark.parametrize('dates,expected', [
|
|
(['Wed, 03 Jun 2026 12:00:00 +0000', 'Mon, 01 Jun 2026 09:00:00 +0000'], ['1', '2']),
|
|
(['Wed, 03 Jun 2026 12:00:00 +0000', 'Wed, 03 Jun 2026 13:00:00 +0200'], ['1', '2']),
|
|
(['Wed, 03 Jun 2026 12:00:00', 'Wed, 03 Jun 2026 11:00:00 +0000'], ['1', '2']),
|
|
(['Wed, 03 Jun 2026 12:00:00 +0000', None, 'not a date'], ['1', '3', '2']),
|
|
(['Wed, 03 Jun 2026 12:00:00 +0000', 'Wed, 03 Jun 2026 14:00:00 +0200'], ['2', '1']),
|
|
(['Wed, 03 Jun 2026 12:00:00 +0000', 'Thu, 04 Jun 2026 12:00:00 +0000'], ['2', '1']),
|
|
(['Wed, 31 Dec 1969 12:00:00 +0000', None], ['1', '2']),
|
|
])
|
|
def test_list_sorts_parsed_dates_without_changing_row_schema(monkeypatch, dates, expected):
|
|
cli = _load_mail_cli(monkeypatch)
|
|
fetched = []
|
|
|
|
class Connection:
|
|
def select(self, folder, readonly=True):
|
|
assert readonly is True
|
|
return 'OK', [str(len(dates)).encode()]
|
|
|
|
def search(self, *_):
|
|
return 'OK', [' '.join(str(i+1) for i in range(len(dates))).encode()]
|
|
|
|
def fetch(self, uid, _):
|
|
fetched.append(uid.decode())
|
|
date = dates[int(uid)-1]
|
|
raw = 'From: Sender <sender@example.com>\r\nSubject: Item '+uid.decode()+'\r\n'
|
|
if date is not None:
|
|
raw += 'Date: '+date+'\r\n'
|
|
return 'OK', [(b'1 (FLAGS (\\Seen) RFC822.HEADER)', (raw+'\r\n').encode())]
|
|
|
|
@contextmanager
|
|
def imap(_):
|
|
yield Connection()
|
|
|
|
rows = []
|
|
monkeypatch.setattr(cli, '_imap', imap)
|
|
monkeypatch.setattr(cli, 'emit', lambda value, args: rows.extend(value))
|
|
cli.cmd_list(SimpleNamespace(account='fixture', folder='INBOX', limit=20))
|
|
assert [row['uid'] for row in rows] == expected
|
|
assert fetched == [str(i) for i in range(len(dates), 0, -1)]
|
|
assert all(set(row) == {'uid', 'date', 'from_addr', 'from_name', 'subject', 'is_read', 'is_answered'} for row in rows)
|
|
assert all(row['is_read'] and not row['is_answered'] for row in rows)
|
|
|
|
|
|
def test_list_still_limits_the_arrival_batch(monkeypatch):
|
|
cli = _load_mail_cli(monkeypatch)
|
|
fetched = []
|
|
class Connection:
|
|
def select(self, *_args, **_kwargs):return 'OK', []
|
|
def search(self, *_):return 'OK', [b'1 2 3']
|
|
def fetch(self, uid, _):
|
|
fetched.append(uid)
|
|
return 'OK', [(b'3 (FLAGS () RFC822.HEADER)', b'From: x@example.com\r\nSubject: bounded\r\n\r\n')]
|
|
@contextmanager
|
|
def imap(_):yield Connection()
|
|
monkeypatch.setattr(cli, '_imap', imap)
|
|
monkeypatch.setattr(cli, 'emit', lambda *_:None)
|
|
cli.cmd_list(SimpleNamespace(account='fixture', folder='INBOX', limit=1))
|
|
assert fetched == [b'3']
|