paste.chapril.org-privatebin/tst/Persistence/ServerSaltTest.php

<?php

use PrivateBin\Data\Filesystem;
use PrivateBin\Persistence\ServerSalt;

class ServerSaltTest extends PHPUnit_Framework_TestCase
{
    private $_path;

    private $_invalidPath;

    private $_otherPath;

    private $_invalidFile;

    public function setUp()
    {
        /* Setup Routine */
        $this->_path = sys_get_temp_dir() . DIRECTORY_SEPARATOR . 'privatebin_data';
        if (!is_dir($this->_path)) {
            mkdir($this->_path);
        }
        ServerSalt::setStore(
            Filesystem::getInstance(array('dir' => $this->_path))
        );

        $this->_otherPath = $this->_path . DIRECTORY_SEPARATOR . 'foo';

        $this->_invalidPath = $this->_path . DIRECTORY_SEPARATOR . 'bar';
        if (!is_dir($this->_invalidPath)) {
            mkdir($this->_invalidPath);
        }
        $this->_invalidFile = $this->_invalidPath . DIRECTORY_SEPARATOR . 'salt.php';
    }

    public function tearDown()
    {
        /* Tear Down Routine */
        chmod($this->_invalidPath, 0700);
        Helper::rmDir($this->_path);
    }

    public function testGeneration()
    {
        // generating new salt
        ServerSalt::setStore(
            Filesystem::getInstance(array('dir' => $this->_path))
        );
        $salt = ServerSalt::get();

        // try setting a different path and resetting it
        ServerSalt::setStore(
            Filesystem::getInstance(array('dir' => $this->_otherPath))
        );
        $this->assertNotEquals($salt, ServerSalt::get());
        ServerSalt::setStore(
            Filesystem::getInstance(array('dir' => $this->_path))
        );
        $this->assertEquals($salt, ServerSalt::get());
    }

    public function testPathShenanigans()
    {
        // try setting an invalid path
        chmod($this->_invalidPath, 0000);
        $store = Filesystem::getInstance(array('dir' => $this->_invalidPath));
        ServerSalt::setStore($store);
        $salt = ServerSalt::get();
        ServerSalt::setStore($store);
        $this->assertNotEquals($salt, ServerSalt::get());
    }

    public function testFileRead()
    {
        // try setting an invalid file
        chmod($this->_invalidPath, 0700);
        file_put_contents($this->_invalidFile, '');
        chmod($this->_invalidFile, 0000);
        $store = Filesystem::getInstance(array('dir' => $this->_invalidPath));
        ServerSalt::setStore($store);
        $salt = ServerSalt::get();
        ServerSalt::setStore($store);
        $this->assertNotEquals($salt, ServerSalt::get());
    }

    public function testFileWrite()
    {
        // try setting an invalid file
        chmod($this->_invalidPath, 0700);
        if (is_file($this->_invalidFile)) {
            chmod($this->_invalidFile, 0600);
            unlink($this->_invalidFile);
        }
        file_put_contents($this->_invalidPath . DIRECTORY_SEPARATOR . '.htaccess', '');
        chmod($this->_invalidPath, 0500);
        $store = Filesystem::getInstance(array('dir' => $this->_invalidPath));
        ServerSalt::setStore($store);
        $salt = ServerSalt::get();
        ServerSalt::setStore($store);
        $this->assertNotEquals($salt, ServerSalt::get());
    }

    public function testPermissionShenanigans()
    {
        // try creating an invalid path
        chmod($this->_invalidPath, 0000);
        ServerSalt::setStore(
            Filesystem::getInstance(array('dir' => $this->_invalidPath . DIRECTORY_SEPARATOR . 'baz'))
        );
        $store = Filesystem::getInstance(array('dir' => $this->_invalidPath));
        ServerSalt::setStore($store);
        $salt = ServerSalt::get();
        ServerSalt::setStore($store);
        $this->assertNotEquals($salt, ServerSalt::get());
    }
}