def test_create_multipart_string_from_files(self):
files = []
values = []
for i in range(3):
_, filename = tempfile.mkstemp()
files.append(('plain', filename))
with open(filename, 'wb') as f:
value = "value-%d" % i
f.write(value)
values.append(value)
result = Util.create_multipart_string_from_files(files)
for mime, f in files:
os.remove(f)
with closing(StringIO(result)) as buf:
parser = Parser()
msg = parser.parse(buf)
for part in msg.walk():
if msg.is_multipart():
i = 0
for msg in part.get_payload():
self.assertEqual(values[i], msg.get_payload())
i += 1
def test_create_multipart_string_from_files_with_none(self):
files = []
values = []
for i in range(3):
_, filename = tempfile.mkstemp()
files.append(('none', filename))
with open(filename, 'wb') as f:
value = "value-%d" % i
f.write(value)
values.append(value)
result = Util.create_multipart_string_from_files(files)
self.assertEqual(values[0], result)
