def test_attach_file_without_content_id(self):
msg = Message(to='me', text='hi')
with patch(open_label) as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = MagicMock(spec=file)
mock_open.return_value.read.return_value = b'x'
msg.attach_file('dummy.png')
self.assertEqual('ContentID' in msg.attachments[0], False)
def test_attach_file_without_content_id(self):
msg = Message(to='me', text='hi')
with patch(open_label) as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = MagicMock(spec=file)
mock_open.return_value.read.return_value = b'x'
msg.attach_file('dummy.png')
self.assertEqual('ContentID' in msg.attachments[0], False)
def test_attach_file_with_content_id(self):
msg = Message(to='me', text='hi')
with patch(open_label) as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = MagicMock(spec=file)
mock_open.return_value.read.return_value = b'x'
content_type = 'image/png'
content_id = 'cid:valid_cid'
filename = 'dummy.png'
msg.attach_file('dummy.png', content_type=content_type,
content_id=content_id)
attachment = {
'Content': b64encode(b'x').decode('utf-8'),
'ContentType': content_type,
'Name': filename,
'ContentID': content_id
}
self.assertEqual(msg.attachments, [attachment])
def test_attach_file_with_content_id(self):
msg = Message(to='me', text='hi')
with patch(open_label) as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = MagicMock(spec=file)
mock_open.return_value.read.return_value = b'x'
content_type = 'image/png'
content_id = 'cid:valid_cid'
filename = 'dummy.png'
msg.attach_file('dummy.png', content_type=content_type,
content_id=content_id)
attachment = {
'Content': b64encode(b'x').decode('utf-8'),
'ContentType': content_type,
'Name': filename,
'ContentID': content_id
}
self.assertEqual(msg.attachments, [attachment])
