Skip to content

Commit ab6d3b3

Browse files
committed
Add a test for a BabylMessage with no visible headers
1 parent 718d11f commit ab6d3b3

1 file changed

Lines changed: 13 additions & 1 deletion

File tree

‎Lib/test/test_mailbox.py‎

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1521,7 +1521,7 @@ def test_labels(self):
15211521
self._box.remove(key1)
15221522
self.assertEqual(set(self._box.get_labels()), set(['blah']))
15231523

1524-
def test_visible(self):
1524+
def test_visible_headers(self):
15251525
# gh-157880: visible headers are written to the file and read back
15261526
msg = mailbox.BabylMessage(self._template % 0)
15271527
msg.update_visible()
@@ -1534,6 +1534,18 @@ def test_visible(self):
15341534
self.assertEqual(self._box.get_bytes(key),
15351535
(self._template % 0).encode('ascii'))
15361536

1537+
def test_no_visible_headers(self):
1538+
# gh-157880: empty visible headers are written to the file
1539+
# and read back
1540+
msg = mailbox.BabylMessage(self._template % 0)
1541+
key = self._box.add(msg)
1542+
for retrieved in (self._box.get_message(key), self._box[key]):
1543+
self.assertEqual(retrieved['from'], 'foo')
1544+
self.assertEqual(retrieved.get_payload(), '0\n')
1545+
self.assertEqual(retrieved.get_visible().keys(), [])
1546+
self.assertEqual(self._box.get_bytes(key),
1547+
(self._template % 0).encode('ascii'))
1548+
15371549

15381550
class FakeFileLikeObject:
15391551

0 commit comments

Comments
 (0)