php-resque/test/Resque/Tests/JobStatusTest.php

<?php
/**
 * Resque_Job_Status tests.
 *
 * @package		Resque/Tests
 * @author		Chris Boulton <chris@bigcommerce.com>
 * @license		http://www.opensource.org/licenses/mit-license.php
 */
class Resque_Tests_JobStatusTest extends Resque_Tests_TestCase
{
    /**
     * @var \Resque_Worker
     */
    protected $worker;

	public function setUp()
	{
		parent::setUp();

		// Register a worker to test with
		$this->worker = new Resque_Worker('jobs');
	}

	public function testJobStatusCanBeTracked()
	{
		$token = Resque::enqueue('jobs', 'Test_Job', null, true);
		$status = new Resque_Job_Status($token);
		$this->assertTrue($status->isTracking());
	}

	public function testJobStatusIsReturnedViaJobInstance()
	{
		$token = Resque::enqueue('jobs', 'Test_Job', null, true);
		$job = Resque_Job::reserve('jobs');
		$this->assertEquals(Resque_Job_Status::STATUS_WAITING, $job->getStatus());
	}

	public function testQueuedJobReturnsQueuedStatus()
	{
		$token = Resque::enqueue('jobs', 'Test_Job', null, true);
		$status = new Resque_Job_Status($token);
		$this->assertEquals(Resque_Job_Status::STATUS_WAITING, $status->get());
	}

	public function testRunningJobReturnsRunningStatus()
	{
		$token = Resque::enqueue('jobs', 'Failing_Job', null, true);
		$job = $this->worker->reserve();
		$this->worker->workingOn($job);
		$status = new Resque_Job_Status($token);
		$this->assertEquals(Resque_Job_Status::STATUS_RUNNING, $status->get());
	}

	public function testFailedJobReturnsFailedStatus()
	{
		$token = Resque::enqueue('jobs', 'Failing_Job', null, true);
		$this->worker->work(0);
		$status = new Resque_Job_Status($token);
		$this->assertEquals(Resque_Job_Status::STATUS_FAILED, $status->get());
	}

	public function testCompletedJobReturnsCompletedStatus()
	{
		$token = Resque::enqueue('jobs', 'Test_Job', null, true);
		$this->worker->work(0);
		$status = new Resque_Job_Status($token);
		$this->assertEquals(Resque_Job_Status::STATUS_COMPLETE, $status->get());
	}

	public function testStatusIsNotTrackedWhenToldNotTo()
	{
		$token = Resque::enqueue('jobs', 'Test_Job', null, false);
		$status = new Resque_Job_Status($token);
		$this->assertFalse($status->isTracking());
	}

	public function testStatusTrackingCanBeStopped()
	{
		Resque_Job_Status::create('test');
		$status = new Resque_Job_Status('test');
		$this->assertEquals(Resque_Job_Status::STATUS_WAITING, $status->get());
		$status->stop();
		$this->assertFalse($status->get());
	}

	public function testRecreatedJobWithTrackingStillTracksStatus()
	{
		$originalToken = Resque::enqueue('jobs', 'Test_Job', null, true);
		$job = $this->worker->reserve();

		// Mark this job as being worked on to ensure that the new status is still
		// waiting.
		$this->worker->workingOn($job);

		// Now recreate it
		$newToken = $job->recreate();

		// Make sure we've got a new job returned
		$this->assertNotEquals($originalToken, $newToken);

		// Now check the status of the new job
		$newJob = Resque_Job::reserve('jobs');
		$this->assertEquals(Resque_Job_Status::STATUS_WAITING, $newJob->getStatus());
	}
}