PrivateBin/tst/Vizhash16x16Test.php

<?php

use PrivateBin\Persistence\ServerSalt;
use PrivateBin\Vizhash16x16;

class Vizhash16x16Test extends PHPUnit_Framework_TestCase
{
    private $_file;

    private $_path;

    public function setUp()
    {
        /* Setup Routine */
        $this->_path = sys_get_temp_dir() . DIRECTORY_SEPARATOR . 'privatebin_data';
        if (!is_dir($this->_path)) {
            mkdir($this->_path);
        }
        $this->_file = $this->_path . DIRECTORY_SEPARATOR . 'vizhash.png';
        ServerSalt::setPath($this->_path);
    }

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

    public function testVizhashGeneratesUniquePngsPerIp()
    {
        $vz = new Vizhash16x16();
        $pngdata = $vz->generate(hash('sha512', '127.0.0.1'));
        file_put_contents($this->_file, $pngdata);
        $finfo = new finfo(FILEINFO_MIME_TYPE);
        $this->assertEquals('image/png', $finfo->file($this->_file));
        $this->assertNotEquals($pngdata, $vz->generate(hash('sha512', '2001:1620:2057:dead:beef::cafe:babe')));
        $this->assertEquals($pngdata, $vz->generate(hash('sha512', '127.0.0.1')));
    }
}