QueueDatabaseQueueUnitTest.php 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156
  1. <?php
  2. namespace Illuminate\Tests\Queue;
  3. use Illuminate\Container\Container;
  4. use Illuminate\Database\Connection;
  5. use Illuminate\Queue\DatabaseQueue;
  6. use Illuminate\Queue\Queue;
  7. use Illuminate\Support\Str;
  8. use Mockery as m;
  9. use PHPUnit\Framework\TestCase;
  10. use ReflectionClass;
  11. use stdClass;
  12. class QueueDatabaseQueueUnitTest extends TestCase
  13. {
  14. protected function tearDown(): void
  15. {
  16. m::close();
  17. }
  18. public function testPushProperlyPushesJobOntoDatabase()
  19. {
  20. $uuid = Str::uuid();
  21. Str::createUuidsUsing(function () use ($uuid) {
  22. return $uuid;
  23. });
  24. $queue = $this->getMockBuilder(DatabaseQueue::class)->onlyMethods(['currentTime'])->setConstructorArgs([$database = m::mock(Connection::class), 'table', 'default'])->getMock();
  25. $queue->expects($this->any())->method('currentTime')->willReturn('time');
  26. $queue->setContainer($container = m::spy(Container::class));
  27. $database->shouldReceive('table')->with('table')->andReturn($query = m::mock(stdClass::class));
  28. $query->shouldReceive('insertGetId')->once()->andReturnUsing(function ($array) use ($uuid) {
  29. $this->assertSame('default', $array['queue']);
  30. $this->assertSame(json_encode(['uuid' => $uuid, 'displayName' => 'foo', 'job' => 'foo', 'maxTries' => null, 'maxExceptions' => null, 'failOnTimeout' => false, 'backoff' => null, 'timeout' => null, 'data' => ['data']]), $array['payload']);
  31. $this->assertEquals(0, $array['attempts']);
  32. $this->assertNull($array['reserved_at']);
  33. $this->assertIsInt($array['available_at']);
  34. });
  35. $queue->push('foo', ['data']);
  36. $container->shouldHaveReceived('bound')->with('events')->once();
  37. Str::createUuidsNormally();
  38. }
  39. public function testDelayedPushProperlyPushesJobOntoDatabase()
  40. {
  41. $uuid = Str::uuid();
  42. Str::createUuidsUsing(function () use ($uuid) {
  43. return $uuid;
  44. });
  45. $queue = $this->getMockBuilder(
  46. DatabaseQueue::class)->onlyMethods(
  47. ['currentTime'])->setConstructorArgs(
  48. [$database = m::mock(Connection::class), 'table', 'default']
  49. )->getMock();
  50. $queue->expects($this->any())->method('currentTime')->willReturn('time');
  51. $queue->setContainer($container = m::spy(Container::class));
  52. $database->shouldReceive('table')->with('table')->andReturn($query = m::mock(stdClass::class));
  53. $query->shouldReceive('insertGetId')->once()->andReturnUsing(function ($array) use ($uuid) {
  54. $this->assertSame('default', $array['queue']);
  55. $this->assertSame(json_encode(['uuid' => $uuid, 'displayName' => 'foo', 'job' => 'foo', 'maxTries' => null, 'maxExceptions' => null, 'failOnTimeout' => false, 'backoff' => null, 'timeout' => null, 'data' => ['data']]), $array['payload']);
  56. $this->assertEquals(0, $array['attempts']);
  57. $this->assertNull($array['reserved_at']);
  58. $this->assertIsInt($array['available_at']);
  59. });
  60. $queue->later(10, 'foo', ['data']);
  61. $container->shouldHaveReceived('bound')->with('events')->once();
  62. Str::createUuidsNormally();
  63. }
  64. public function testFailureToCreatePayloadFromObject()
  65. {
  66. $this->expectException('InvalidArgumentException');
  67. $job = new stdClass;
  68. $job->invalid = "\xc3\x28";
  69. $queue = $this->getMockForAbstractClass(Queue::class);
  70. $class = new ReflectionClass(Queue::class);
  71. $createPayload = $class->getMethod('createPayload');
  72. $createPayload->setAccessible(true);
  73. $createPayload->invokeArgs($queue, [
  74. $job,
  75. 'queue-name',
  76. ]);
  77. }
  78. public function testFailureToCreatePayloadFromArray()
  79. {
  80. $this->expectException('InvalidArgumentException');
  81. $queue = $this->getMockForAbstractClass(Queue::class);
  82. $class = new ReflectionClass(Queue::class);
  83. $createPayload = $class->getMethod('createPayload');
  84. $createPayload->setAccessible(true);
  85. $createPayload->invokeArgs($queue, [
  86. ["\xc3\x28"],
  87. 'queue-name',
  88. ]);
  89. }
  90. public function testBulkBatchPushesOntoDatabase()
  91. {
  92. $uuid = Str::uuid();
  93. Str::createUuidsUsing(function () use ($uuid) {
  94. return $uuid;
  95. });
  96. $database = m::mock(Connection::class);
  97. $queue = $this->getMockBuilder(DatabaseQueue::class)->onlyMethods(['currentTime', 'availableAt'])->setConstructorArgs([$database, 'table', 'default'])->getMock();
  98. $queue->expects($this->any())->method('currentTime')->willReturn('created');
  99. $queue->expects($this->any())->method('availableAt')->willReturn('available');
  100. $database->shouldReceive('table')->with('table')->andReturn($query = m::mock(stdClass::class));
  101. $query->shouldReceive('insert')->once()->andReturnUsing(function ($records) use ($uuid) {
  102. $this->assertEquals([[
  103. 'queue' => 'queue',
  104. 'payload' => json_encode(['uuid' => $uuid, 'displayName' => 'foo', 'job' => 'foo', 'maxTries' => null, 'maxExceptions' => null, 'failOnTimeout' => false, 'backoff' => null, 'timeout' => null, 'data' => ['data']]),
  105. 'attempts' => 0,
  106. 'reserved_at' => null,
  107. 'available_at' => 'available',
  108. 'created_at' => 'created',
  109. ], [
  110. 'queue' => 'queue',
  111. 'payload' => json_encode(['uuid' => $uuid, 'displayName' => 'bar', 'job' => 'bar', 'maxTries' => null, 'maxExceptions' => null, 'failOnTimeout' => false, 'backoff' => null, 'timeout' => null, 'data' => ['data']]),
  112. 'attempts' => 0,
  113. 'reserved_at' => null,
  114. 'available_at' => 'available',
  115. 'created_at' => 'created',
  116. ]], $records);
  117. });
  118. $queue->bulk(['foo', 'bar'], ['data'], 'queue');
  119. Str::createUuidsNormally();
  120. }
  121. public function testBuildDatabaseRecordWithPayloadAtTheEnd()
  122. {
  123. $queue = m::mock(DatabaseQueue::class);
  124. $record = $queue->buildDatabaseRecord('queue', 'any_payload', 0);
  125. $this->assertArrayHasKey('payload', $record);
  126. $this->assertArrayHasKey('payload', array_slice($record, -1, 1, true));
  127. }
  128. }