SchedulerTest.php 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381
  1. <?php namespace GO\Job\Tests;
  2. use GO\Job;
  3. use DateTime;
  4. use GO\FailedJob;
  5. use GO\Scheduler;
  6. use PHPUnit\Framework\TestCase;
  7. class SchedulerTest extends TestCase
  8. {
  9. public function testShouldQueueJobs()
  10. {
  11. $scheduler = new Scheduler();
  12. $this->assertEquals(count($scheduler->getQueuedJobs()), 0);
  13. $scheduler->raw('ls');
  14. $this->assertEquals(count($scheduler->getQueuedJobs()), 1);
  15. }
  16. public function testShouldQueueAPhpScript()
  17. {
  18. $scheduler = new Scheduler();
  19. $script = __DIR__ . '/../test_job.php';
  20. $this->assertEquals(count($scheduler->getQueuedJobs()), 0);
  21. $scheduler->php($script);
  22. $this->assertEquals(count($scheduler->getQueuedJobs()), 1);
  23. }
  24. public function testShouldAllowCustomPhpBin()
  25. {
  26. $scheduler = new Scheduler();
  27. $script = __DIR__ . '/../test_job.php';
  28. // Create fake bin
  29. $bin = __DIR__ . '/../custom_bin';
  30. touch($bin);
  31. $job = $scheduler->php($script, $bin)->inForeground();
  32. unlink($bin);
  33. $this->assertEquals($bin . ' ' . $script, $job->compile());
  34. }
  35. public function testShouldUseSystemPhpBinIfCustomBinDoesNotExist()
  36. {
  37. $scheduler = new Scheduler();
  38. $script = __DIR__ . '/../test_job.php';
  39. // Create fake bin
  40. $bin = '/my/custom/php/bin';
  41. $job = $scheduler->php($script, $bin)->inForeground();
  42. $this->assertNotEquals($bin . ' ' . $script, $job->compile());
  43. $this->assertEquals(PHP_BINARY . ' ' . $script, $job->compile());
  44. }
  45. public function testShouldThrowExceptionIfScriptIsNotAString()
  46. {
  47. $this->expectException(\InvalidArgumentException::class);
  48. $scheduler = new Scheduler();
  49. $scheduler->php(function () {
  50. return false;
  51. });
  52. $scheduler->run();
  53. }
  54. public function testShouldMarkJobAsFailedIfScriptPathIsInvalid()
  55. {
  56. $scheduler = new Scheduler();
  57. $scheduler->php('someInvalidPathToAScript');
  58. $scheduler->run();
  59. $fail = $scheduler->getFailedJobs();
  60. $this->assertCount(1, $fail);
  61. $this->assertContainsOnlyInstancesOf(FailedJob::class, $fail);
  62. }
  63. public function testShouldQueueAShellCommand()
  64. {
  65. $scheduler = new Scheduler();
  66. $this->assertEquals(count($scheduler->getQueuedJobs()), 0);
  67. $scheduler->raw('ls');
  68. $this->assertEquals(count($scheduler->getQueuedJobs()), 1);
  69. }
  70. public function testShouldQueueAFunction()
  71. {
  72. $scheduler = new Scheduler();
  73. $this->assertEquals(count($scheduler->getQueuedJobs()), 0);
  74. $scheduler->call(function () {
  75. return true;
  76. });
  77. $this->assertEquals(count($scheduler->getQueuedJobs()), 1);
  78. }
  79. public function testShouldKeepTrackOfExecutedJobs()
  80. {
  81. $scheduler = new Scheduler();
  82. $scheduler->call(function () {
  83. return true;
  84. });
  85. $this->assertEquals(count($scheduler->getQueuedJobs()), 1);
  86. $this->assertEquals(count($scheduler->getExecutedJobs()), 0);
  87. $scheduler->run();
  88. $this->assertEquals(count($scheduler->getExecutedJobs()), 1);
  89. }
  90. public function testShouldPassParametersToAFunction()
  91. {
  92. $scheduler = new Scheduler();
  93. $outputFile = __DIR__ . '/../tmp/output.txt';
  94. $scheduler->call(function ($phrase) {
  95. return $phrase;
  96. }, [
  97. 'Hello World!',
  98. ])->output($outputFile);
  99. @unlink($outputFile);
  100. $this->assertFalse(file_exists($outputFile));
  101. $scheduler->run();
  102. $this->assertNotEquals('Hello', file_get_contents($outputFile));
  103. $this->assertEquals('Hello World!', file_get_contents($outputFile));
  104. @unlink($outputFile);
  105. }
  106. public function testShouldKeepTrackOfFailedJobs()
  107. {
  108. $scheduler = new Scheduler();
  109. $exception = new \Exception('Something failed');
  110. $scheduler->call(function () use ($exception) {
  111. throw $exception;
  112. });
  113. $this->assertEquals(count($scheduler->getFailedJobs()), 0);
  114. $scheduler->run();
  115. $this->assertEquals(count($scheduler->getExecutedJobs()), 0);
  116. $this->assertEquals(count($scheduler->getFailedJobs()), 1);
  117. $failedJob = $scheduler->getFailedJobs()[0];
  118. $this->assertInstanceOf(FailedJob::class, $failedJob);
  119. $this->assertSame($exception, $failedJob->getException());
  120. $this->assertInstanceOf(Job::class, $failedJob->getJob());
  121. }
  122. public function testShouldKeepExecutingJobsIfOneFails()
  123. {
  124. $scheduler = new Scheduler();
  125. $scheduler->call(function () {
  126. throw new \Exception('Something failed');
  127. });
  128. $scheduler->call(function () {
  129. return true;
  130. });
  131. $scheduler->run();
  132. $this->assertEquals(count($scheduler->getExecutedJobs()), 1);
  133. $this->assertEquals(count($scheduler->getFailedJobs()), 1);
  134. }
  135. public function testShouldInjectConfigToTheJobs()
  136. {
  137. $schedulerConfig = [
  138. 'email' => [
  139. 'subject' => 'My custom subject',
  140. ],
  141. ];
  142. $scheduler = new Scheduler($schedulerConfig);
  143. $job = $scheduler->raw('ls');
  144. $this->assertEquals($job->getEmailConfig()['subject'], $schedulerConfig['email']['subject']);
  145. }
  146. public function testShouldPrioritizeJobConfigOverSchedulerConfig()
  147. {
  148. $schedulerConfig = [
  149. 'email' => [
  150. 'subject' => 'My custom subject',
  151. ],
  152. ];
  153. $scheduler = new Scheduler($schedulerConfig);
  154. $jobConfig = [
  155. 'email' => [
  156. 'subject' => 'My job subject',
  157. ],
  158. ];
  159. $job = $scheduler->raw('ls')->configure($jobConfig);
  160. $this->assertNotEquals($job->getEmailConfig()['subject'], $schedulerConfig['email']['subject']);
  161. $this->assertEquals($job->getEmailConfig()['subject'], $jobConfig['email']['subject']);
  162. }
  163. public function testShouldShowClosuresVerboseOutputAsText()
  164. {
  165. $scheduler = new Scheduler();
  166. $scheduler->call(function ($phrase) {
  167. return $phrase;
  168. }, [
  169. 'Hello World!',
  170. ]);
  171. $scheduler->run();
  172. $this->assertMatchesRegularExpression('/ Executing Closure$/', $scheduler->getVerboseOutput());
  173. $this->assertMatchesRegularExpression('/ Executing Closure$/', $scheduler->getVerboseOutput('text'));
  174. }
  175. public function testShouldShowClosuresVerboseOutputAsHtml()
  176. {
  177. $scheduler = new Scheduler();
  178. $scheduler->call(function ($phrase) {
  179. return $phrase;
  180. }, [
  181. 'Hello World!',
  182. ]);
  183. $scheduler->call(function () {
  184. return true;
  185. });
  186. $scheduler->run();
  187. $this->assertMatchesRegularExpression('/<br>/', $scheduler->getVerboseOutput('html'));
  188. }
  189. public function testShouldShowClosuresVerboseOutputAsArray()
  190. {
  191. $scheduler = new Scheduler();
  192. $scheduler->call(function ($phrase) {
  193. return $phrase;
  194. }, [
  195. 'Hello World!',
  196. ]);
  197. $scheduler->call(function () {
  198. return true;
  199. });
  200. $scheduler->run();
  201. $this->assertTrue(is_array($scheduler->getVerboseOutput('array')));
  202. $this->assertEquals(count($scheduler->getVerboseOutput('array')), 2);
  203. }
  204. public function testShouldThrowExceptionWithInvalidOutputType()
  205. {
  206. $this->expectException(\InvalidArgumentException::class);
  207. $scheduler = new Scheduler();
  208. $scheduler->call(function ($phrase) {
  209. return $phrase;
  210. }, [
  211. 'Hello World!',
  212. ]);
  213. $scheduler->call(function () {
  214. return true;
  215. });
  216. $scheduler->run();
  217. $scheduler->getVerboseOutput('multiline');
  218. }
  219. public function testShouldPrioritizeJobsInBackround()
  220. {
  221. $scheduler = new Scheduler();
  222. $scheduler->php(__DIR__ . '/../async_job.php', null, null, 'async_foreground')->then(function () {
  223. return true;
  224. });
  225. $scheduler->php(__DIR__ . '/../async_job.php', null, null, 'async_background');
  226. $jobs = $scheduler->getQueuedJobs();
  227. $this->assertEquals('async_background', $jobs[0]->getId());
  228. $this->assertEquals('async_foreground', $jobs[1]->getId());
  229. }
  230. public function testCouldRunTwice()
  231. {
  232. $scheduler = new Scheduler();
  233. $scheduler->call(function () {
  234. return true;
  235. });
  236. $scheduler->run();
  237. $this->assertCount(1, $scheduler->getExecutedJobs(), 'Number of executed jobs');
  238. $scheduler->resetRun();
  239. $scheduler->run();
  240. $this->assertCount(1, $scheduler->getExecutedJobs(), 'Number of executed jobs');
  241. }
  242. public function testClearJobs()
  243. {
  244. $scheduler = new Scheduler();
  245. $scheduler->call(function () {
  246. return true;
  247. });
  248. $this->assertCount(1, $scheduler->getQueuedJobs(), 'Number of queued jobs');
  249. $scheduler->clearJobs();
  250. $this->assertCount(0, $scheduler->getQueuedJobs(), 'Number of queued jobs');
  251. }
  252. public function testShouldRunDelayedJobsIfDueWhenCreated()
  253. {
  254. $scheduler = new Scheduler();
  255. $currentTime = date('H:i');
  256. $scheduler->call(function () {
  257. $s = (int) date('s');
  258. sleep(60 - $s + 1);
  259. })->daily($currentTime);
  260. $scheduler->call(function () {
  261. // do nothing
  262. })->daily($currentTime);
  263. $executed = $scheduler->run();
  264. $this->assertEquals(2, count($executed));
  265. }
  266. public function testShouldRunAtSpecificTime()
  267. {
  268. $scheduler = new Scheduler();
  269. $runTime = new DateTime('2017-09-13 00:00:00');
  270. $scheduler->call(function () {
  271. // do nothing
  272. })->daily('00:00');
  273. $executed = $scheduler->run($runTime);
  274. $this->assertEquals(1, count($executed));
  275. }
  276. }