parallel/test/Sync/PosixSemaphoreTest.php

<?php

namespace Amp\Tests\Concurrent\Sync;

use Amp\Concurrent\Forking\Fork;
use Amp\Concurrent\Sync\{PosixSemaphore, Semaphore};
use Amp\Coroutine;
use Amp\Loop;

/**
 * @group posix
 * @requires extension sysvmsg
 */
class PosixSemaphoreTest extends AbstractSemaphoreTest
{
    public function createSemaphore($locks)
    {
        return new PosixSemaphore($locks);
    }

    public function tearDown()
    {
        if ($this->semaphore && !$this->semaphore->isFreed()) {
            $this->semaphore->free();
        }
    }

    public function testCloneIsNewSemaphore()
    {
        Coroutine\create(function () {
            $this->semaphore = $this->createSemaphore(1);
            $clone = clone $this->semaphore;

            $lock = yield from $clone->acquire();

            $this->assertCount(1, $this->semaphore);
            $this->assertCount(0, $clone);

            $lock->release();

            $clone->free();
        })->done();

        Loop\run();
    }

    public function testFree()
    {
        $this->semaphore = $this->createSemaphore(1);

        $this->assertFalse($this->semaphore->isFreed());

        $this->semaphore->free();

        $this->assertTrue($this->semaphore->isFreed());
    }

    /**
     * @requires extension pcntl
     */
    public function testAcquireInMultipleForks()
    {
        Coroutine\create(function () {
            $this->semaphore = $this->createSemaphore(1);

            $fork1 = new Fork(function (Semaphore $semaphore) {
                $lock = yield from $semaphore->acquire();

                usleep(1e5);

                $lock->release();

                yield 0;
            }, $this->semaphore);

            $fork2 = new Fork(function (Semaphore $semaphore) {
                $lock = yield from $semaphore->acquire();

                usleep(1e5);

                $lock->release();

                yield 1;
            }, $this->semaphore);

            $start = microtime(true);

            $fork1->start();
            $fork2->start();

            yield from $fork1->join();
            yield from $fork2->join();

            $this->assertGreaterThan(1, microtime(true) - $start);
        });

        Loop\run();
    }
}