def test_message_writer_and_abort(self):
from zope.app.mail.maildir import MaildirMessageWriter
filename1 = '/path/to/maildir/tmp/1234500002.4242.myhostname'
filename2 = '/path/to/maildir/new/1234500002.4242.myhostname'
fd = FakeFile(filename1, 'w')
writer = MaildirMessageWriter(fd, filename1, filename2)
self.assertEquals(writer._fd._filename, filename1)
self.assertEquals(writer._fd._mode, 'w') # TODO or 'wb'?
print >> writer, 'fee',
writer.write(' fie')
writer.writelines([' foe', ' foo'])
self.assertEquals(writer._fd._written, 'fee fie foe foo')
writer.abort()
self.assertEquals(writer._fd._closed, True)
self.assert_(filename1 in self.fake_os_module._removed_files)
# Once aborted, abort does nothing
self.fake_os_module._removed_files = ()
writer.abort()
writer.abort()
self.assertEquals(self.fake_os_module._removed_files, ())
# Once aborted, commit fails
self.assertRaises(RuntimeError, writer.commit)
