Viewing file:  Bug51Test.php (3.66 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
<?php
class Swift_Bug51Test extends \SwiftMailerTestCase
{
private $_attachmentFile;
private $_outputFile;
public function setUp()
{
if (!defined('SWIFT_TMP_DIR') || !is_writable(SWIFT_TMP_DIR)) {
$this->markTestSkipped(
'Cannot run test without a writable directory to use ('.
'define SWIFT_TMP_DIR in tests/config.php if you wish to run this test)'
);
}
$this->_attachmentFile = SWIFT_TMP_DIR.'/attach.rand.bin';
file_put_contents($this->_attachmentFile, '');
$this->_outputFile = SWIFT_TMP_DIR.'/attach.out.bin';
file_put_contents($this->_outputFile, '');
}
public function tearDown()
{
unlink($this->_attachmentFile);
unlink($this->_outputFile);
}
public function testAttachmentsDoNotGetTruncatedUsingToByteStream()
{
//Run 100 times with 10KB attachments
for ($i = 0; $i < 10; ++$i) {
$message = $this->_createMessageWithRandomAttachment(
10000, $this->_attachmentFile
);
file_put_contents($this->_outputFile, '');
$message->toByteStream(
new Swift_ByteStream_FileByteStream($this->_outputFile, true)
);
$emailSource = file_get_contents($this->_outputFile);
$this->assertAttachmentFromSourceMatches(
file_get_contents($this->_attachmentFile),
$emailSource
);
}
}
public function testAttachmentsDoNotGetTruncatedUsingToString()
{
//Run 100 times with 10KB attachments
for ($i = 0; $i < 10; ++$i) {
$message = $this->_createMessageWithRandomAttachment(
10000, $this->_attachmentFile
);
$emailSource = $message->toString();
$this->assertAttachmentFromSourceMatches(
file_get_contents($this->_attachmentFile),
$emailSource
);
}
}
// -- Custom Assertions
public function assertAttachmentFromSourceMatches($attachmentData, $source)
{
$encHeader = 'Content-Transfer-Encoding: base64';
$base64declaration = strpos($source, $encHeader);
$attachmentDataStart = strpos($source, "\r\n\r\n", $base64declaration);
$attachmentDataEnd = strpos($source, "\r\n--", $attachmentDataStart);
if (false === $attachmentDataEnd) {
$attachmentBase64 = trim(substr($source, $attachmentDataStart));
} else {
$attachmentBase64 = trim(substr(
$source, $attachmentDataStart,
$attachmentDataEnd - $attachmentDataStart
));
}
$this->assertIdenticalBinary($attachmentData, base64_decode($attachmentBase64));
}
// -- Creation Methods
private function _fillFileWithRandomBytes($byteCount, $file)
{
// I was going to use dd with if=/dev/random but this way seems more
// cross platform even if a hella expensive!!
file_put_contents($file, '');
$fp = fopen($file, 'wb');
for ($i = 0; $i < $byteCount; ++$i) {
$byteVal = rand(0, 255);
fwrite($fp, pack('i', $byteVal));
}
fclose($fp);
}
private function _createMessageWithRandomAttachment($size, $attachmentPath)
{
$this->_fillFileWithRandomBytes($size, $attachmentPath);
$message = Swift_Message::newInstance()
->setSubject('test')
->setBody('test')
->setFrom('a@b.c')
->setTo('d@e.f')
->attach(Swift_Attachment::fromPath($attachmentPath))
;
return $message;
}
}
|