Test FIFO with delay jobs

Author: mariocamaraneto

tests/app/HeavyJob.php

@@ -0,0 +1,34 @@
+<?php
+/**
+ * @link http://www.yiiframework.com/
+ * @copyright Copyright (c) 2008 Yii Software LLC
+ * @license http://www.yiiframework.com/license/
+ */
+
+namespace tests\app;
+
+use Yii;
+use yii\base\BaseObject;
+use yii\queue\JobInterface;
+
+/**
+ * Simple Job.
+ *
+ * @author Roman Zhuravlev <[email protected]>
+ */
+class HeavyJob extends BaseObject implements JobInterface
+{
+    public $uid;
+    public $load;
+
+    public function execute($queue)
+    {
+        file_put_contents($this->getFileName(), '');
+        sleep($this->load);
+    }
+
+    public function getFileName()
+    {
+        return Yii::getAlias("@runtime/job-{$this->uid}.lock");
+    }
+}

tests/drivers/TestCase.php

@@ -7,6 +7,7 @@
 
 namespace tests\drivers;
 
+use tests\app\HeavyJob;
 use Yii;
 use tests\app\SimpleJob;
 use yii\queue\Queue;
@@ -34,6 +35,18 @@ protected function createSimpleJob($data='')
         ]);
     }
 
+    /**
+     * @param int $load
+     * @return HeavyJob
+     */
+    protected function createHeavyJob($load=0)
+    {
+        return new HeavyJob([
+            'uid'   => uniqid(),
+            'load'  =>$load
+        ]);
+    }
+
     /**
      * @param SimpleJob $job
      */
@@ -67,15 +80,33 @@ protected function assertSimpleJobLaterDone(SimpleJob $job, $delay)
     }
 
     /**
-     * @param SimpleJob $job
+     * @param HeavyJob $job
      * @param int $delay
+     * @param $initialTime
      */
-    protected function assertSimpleJobDelayedFifoDone(SimpleJob $job1, SimpleJob $job2, $msg)
+    protected function assertHeavyJobLaterDone(HeavyJob $job, $delay, $initialTime)
+    {
+        $time = $initialTime + $delay;
+        sleep($delay);
+        $timeout = 5000000; // 5 sec
+        $step = 50000;
+        while (!file_exists($job->getFileName()) && $timeout > 0) {
+            usleep($step);
+            $timeout -= $step;
+        }
+        $this->assertFileExists($job->getFileName());
+        $this->assertGreaterThanOrEqual($time, filemtime($job->getFileName()));
+    }
+
+    /**
+     * @param HeavyJob $job
+     * @param int $delay
+     */
+    protected function assertHeavyJobDelayedFifoDone(HeavyJob $job1, HeavyJob $job2, $msg)
     {
         $this->assertFileExists($job1->getFileName());
         $this->assertFileExists($job2->getFileName());
-
-        $this->assertGreaterThanOrEqual(file_get_contents($job1->getFileName()), file_get_contents($job2->getFileName()), $msg);
+        $this->assertGreaterThanOrEqual(filemtime($job1->getFileName()), filemtime($job2->getFileName()), $msg);
     }
 
     /**

tests/drivers/redis/QueueTest.php

@@ -58,25 +58,33 @@ public function testLater()
         $this->assertSimpleJobLaterDone($job, 2);
     }
 
-    public function testLaterFifo()
+    public function testHeavyJobDelayedFifo()
     {
         $this->startProcess(['php', 'yii', 'queue/listen', '1']);
 
-        $job1 = $this->createSimpleJob('1');
-        $this->getQueue()->delay(2)->push($job1);
+        $delay = 1;
+        $load = $delay+1;
+        $initialTimeFirstBatch = time();
 
-        $job2 = $this->createSimpleJob('2');
-        $this->getQueue()->delay( 2)->push($job2);
+        $job1 = $this->createHeavyJob($load);
+        $this->getQueue()->delay($delay)->push($job1);
 
-        $job3 = $this->createSimpleJob('3');
-        $this->getQueue()->delay(2)->push($job3);
+        $job2 = $this->createHeavyJob($load);
+        $this->getQueue()->delay($delay)->push($job2);
 
-        $this->assertSimpleJobLaterDone($job1, 2);
-        $this->assertSimpleJobLaterDone($job2, 2);
-        $this->assertSimpleJobLaterDone($job3, 2);
+        sleep($delay);
 
-        $this->assertSimpleJobDelayedFifoDone($job1, $job2, 'Job1 < Job2');
-        $this->assertSimpleJobDelayedFifoDone($job2, $job3, 'Job2 < Job3');
+        $initialTimeSecondBatch = time();
+        $job3 = $this->createHeavyJob($load);
+        $this->getQueue()->delay($delay)->push($job3);
+
+
+        $this->assertHeavyJobLaterDone($job1, $delay, $initialTimeFirstBatch);
+        $this->assertHeavyJobLaterDone($job2, $delay, $initialTimeFirstBatch);
+        $this->assertHeavyJobLaterDone($job3, $delay, $initialTimeSecondBatch);
+
+        $this->assertHeavyJobDelayedFifoDone($job1, $job2, 'Job1 < Job2');
+        $this->assertHeavyJobDelayedFifoDone($job2, $job3, 'Job2 < Job3');
     }
 
     public function testRetry()