| 3016 | self.assertEqual(text, s.getvalue()) |
| 3017 | |
| 3018 | def test_message_from_file(self): |
| 3019 | with openfile('msg_01.txt', encoding="utf-8") as fp: |
| 3020 | text = fp.read() |
| 3021 | fp.seek(0) |
| 3022 | msg = email.message_from_file(fp) |
| 3023 | s = StringIO() |
| 3024 | # Don't wrap/continue long headers since we're trying to test |
| 3025 | # idempotency. |
| 3026 | g = Generator(s, maxheaderlen=0) |
| 3027 | g.flatten(msg) |
| 3028 | self.assertEqual(text, s.getvalue()) |
| 3029 | |
| 3030 | def test_message_from_string_with_class(self): |
| 3031 | with openfile('msg_01.txt', encoding="utf-8") as fp: |