97 lines
4.7 KiB
PHP
97 lines
4.7 KiB
PHP
<?php
|
|
|
|
use PrivateBin\data\data;
|
|
|
|
class privatebin_dataTest extends PHPUnit_Framework_TestCase
|
|
{
|
|
private $_model;
|
|
|
|
private $_path;
|
|
|
|
public function setUp()
|
|
{
|
|
/* Setup Routine */
|
|
$this->_path = sys_get_temp_dir() . DIRECTORY_SEPARATOR . 'privatebin_data';
|
|
$this->_model = data::getInstance(array('dir' => $this->_path));
|
|
}
|
|
|
|
public function tearDown()
|
|
{
|
|
/* Tear Down Routine */
|
|
helper::rmdir($this->_path);
|
|
}
|
|
|
|
public function testFileBasedDataStoreWorks()
|
|
{
|
|
$this->_model->delete(helper::getPasteId());
|
|
|
|
// storing pastes
|
|
$paste = helper::getPaste(array('expire_date' => 1344803344));
|
|
$this->assertFalse($this->_model->exists(helper::getPasteId()), 'paste does not yet exist');
|
|
$this->assertTrue($this->_model->create(helper::getPasteId(), $paste), 'store new paste');
|
|
$this->assertTrue($this->_model->exists(helper::getPasteId()), 'paste exists after storing it');
|
|
$this->assertFalse($this->_model->create(helper::getPasteId(), $paste), 'unable to store the same paste twice');
|
|
$this->assertEquals(json_decode(json_encode($paste)), $this->_model->read(helper::getPasteId()));
|
|
|
|
// storing comments
|
|
$this->assertFalse($this->_model->existsComment(helper::getPasteId(), helper::getPasteId(), helper::getCommentId()), 'comment does not yet exist');
|
|
$this->assertTrue($this->_model->createComment(helper::getPasteId(), helper::getPasteId(), helper::getCommentId(), helper::getComment()) !== false, 'store comment');
|
|
$this->assertTrue($this->_model->existsComment(helper::getPasteId(), helper::getPasteId(), helper::getCommentId()), 'comment exists after storing it');
|
|
$comment = json_decode(json_encode(helper::getComment()));
|
|
$comment->id = helper::getCommentId();
|
|
$comment->parentid = helper::getPasteId();
|
|
$this->assertEquals(
|
|
array($comment->meta->postdate => $comment),
|
|
$this->_model->readComments(helper::getPasteId())
|
|
);
|
|
|
|
// deleting pastes
|
|
$this->_model->delete(helper::getPasteId());
|
|
$this->assertFalse($this->_model->exists(helper::getPasteId()), 'paste successfully deleted');
|
|
$this->assertFalse($this->_model->existsComment(helper::getPasteId(), helper::getPasteId(), helper::getCommentId()), 'comment was deleted with paste');
|
|
$this->assertFalse($this->_model->read(helper::getPasteId()), 'paste can no longer be found');
|
|
}
|
|
|
|
public function testFileBasedAttachmentStoreWorks()
|
|
{
|
|
$this->_model->delete(helper::getPasteId());
|
|
$original = $paste = helper::getPasteWithAttachment(array('expire_date' => 1344803344));
|
|
$paste['meta']['attachment'] = $paste['attachment'];
|
|
$paste['meta']['attachmentname'] = $paste['attachmentname'];
|
|
unset($paste['attachment'], $paste['attachmentname']);
|
|
$this->assertFalse($this->_model->exists(helper::getPasteId()), 'paste does not yet exist');
|
|
$this->assertTrue($this->_model->create(helper::getPasteId(), $paste), 'store new paste');
|
|
$this->assertTrue($this->_model->exists(helper::getPasteId()), 'paste exists after storing it');
|
|
$this->assertFalse($this->_model->create(helper::getPasteId(), $paste), 'unable to store the same paste twice');
|
|
$this->assertEquals(json_decode(json_encode($original)), $this->_model->read(helper::getPasteId()));
|
|
}
|
|
|
|
public function testPurge()
|
|
{
|
|
mkdir($this->_path . DIRECTORY_SEPARATOR . '00', 0777, true);
|
|
$expired = helper::getPaste(array('expire_date' => 1344803344));
|
|
$paste = helper::getPaste(array('expire_date' => time() + 3600));
|
|
$keys = array('a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'x', 'y', 'z');
|
|
$ids = array();
|
|
foreach ($keys as $key) {
|
|
$ids[$key] = substr(md5($key), 0, 16);
|
|
$this->assertFalse($this->_model->exists($ids[$key]), "paste $key does not yet exist");
|
|
if (in_array($key, array('x', 'y', 'z'))) {
|
|
$this->assertTrue($this->_model->create($ids[$key], $paste), "store $key paste");
|
|
} else {
|
|
$this->assertTrue($this->_model->create($ids[$key], $expired), "store $key paste");
|
|
}
|
|
$this->assertTrue($this->_model->exists($ids[$key]), "paste $key exists after storing it");
|
|
}
|
|
$this->_model->purge(10);
|
|
foreach ($ids as $key => $id) {
|
|
if (in_array($key, array('x', 'y', 'z'))) {
|
|
$this->assertTrue($this->_model->exists($id), "paste $key exists after purge");
|
|
$this->_model->delete($id);
|
|
} else {
|
|
$this->assertFalse($this->_model->exists($id), "paste $key was purged");
|
|
}
|
|
}
|
|
}
|
|
}
|