php-flasher/Tests/Storage/StorageManagerTest.php

<?php

namespace Flasher\Prime\Tests\Storage;

use Flasher\Prime\Envelope;
use Flasher\Prime\EventDispatcher\EventDispatcher;
use Flasher\Prime\EventDispatcher\EventListener\PostFlushListener;
use Flasher\Prime\Notification\Notification;
use Flasher\Prime\Stamp\HopsStamp;
use Flasher\Prime\Stamp\UuidStamp;
use Flasher\Prime\Storage\ArrayStorage;
use Flasher\Prime\Storage\StorageManager;
use PHPUnit\Framework\TestCase;

class StorageManagerTest extends TestCase
{
    public function testAll()
    {
        $storageManager = new StorageManager(new ArrayStorage(), new EventDispatcher());
        $envelopes      = array();

        foreach (range(0, 4) as $index) {
            $envelopes[$index] = new Envelope(new Notification('success', 'success message'));
            $storageManager->add($envelopes[$index]);
        }

        $this->assertCount(5, $storageManager->all());
    }

    public function testClear()
    {
        $storageManager = new StorageManager(new ArrayStorage(), new EventDispatcher());
        $envelopes      = array();

        foreach (range(0, 4) as $index) {
            $envelopes[$index] = new Envelope(new Notification('success', 'success message'));
            $storageManager->add($envelopes[$index]);
        }

        $storageManager->clear();

        $this->assertSame(array(), $storageManager->all());
    }

    public function testAdd()
    {
        $storageManager = new StorageManager(new ArrayStorage(),new EventDispatcher());
        $storageManager->add(new Envelope(new Notification('success')));

        $envelopes = $storageManager->all();

        $this->assertCount(1, $envelopes);

        $uuid = $envelopes[0]->get('Flasher\Prime\Stamp\UuidStamp');
        $this->assertNotNull($uuid);
    }

    public function testFlush()
    {
        $storage = new ArrayStorage();

        $eventDispatcher = new EventDispatcher();
        $eventDispatcher->addSubscriber(new PostFlushListener($storage));

        $storageManager = new StorageManager($storage, $eventDispatcher);

        $envelope = new Envelope(
            new Notification('error message', 'error'),
            new HopsStamp(2),
            new UuidStamp('fake-uuid')
        );
        $storageManager->add($envelope);

        $envelopes = array();
        foreach (range(0, 2) as $index) {
            $envelopes[$index] = new Envelope(new Notification('success message', 'success'));
            $storageManager->add($envelopes[$index]);
        }
        $envelopes[] = $envelope;

        $storageManager->flush($envelopes);

        $this->assertSame(array('fake-uuid'), array_keys(UuidStamp::indexWithUuid($storageManager->all())));
    }
}