mmucklo /
DtcQueueBundle
| @@ 10-38 (lines=29) @@ | ||
| 7 | ||
| 8 | class RunManagerTest extends TestCase |
|
| 9 | { |
|
| 10 | public function testPruneStaleRuns() |
|
| 11 | { |
|
| 12 | JobManagerTest::setUpBeforeClass(); |
|
| 13 | $jobManager = JobManagerTest::$jobManager; |
|
| 14 | $runClass = \Dtc\QueueBundle\Entity\Run::class; |
|
| 15 | $runArchiveClass = \Dtc\QueueBundle\Entity\RunArchive::class; |
|
| 16 | $runManager = new \Dtc\QueueBundle\ORM\RunManager($jobManager->getObjectManager(), $runClass, RunArchive::class); |
|
| 17 | $objectManager = $runManager->getObjectManager(); |
|
| 18 | $runRepository = $objectManager->getRepository($runClass); |
|
| 19 | self::assertEmpty($runRepository->findAll()); |
|
| 20 | $runArchiveRepository = $objectManager->getRepository($runArchiveClass); |
|
| 21 | self::assertEmpty($runArchiveRepository->findAll()); |
|
| 22 | ||
| 23 | $run = new $runClass(); |
|
| 24 | $time = time() - 96400; |
|
| 25 | $date = new \DateTime("@$time"); |
|
| 26 | ||
| 27 | $run->setStartedAt($date); |
|
| 28 | $run->setLastHeartbeatAt($date); |
|
| 29 | $objectManager->persist($run); |
|
| 30 | $objectManager->flush($run); |
|
| 31 | self::assertCount(1, $runRepository->findAll()); |
|
| 32 | ||
| 33 | $count = $runManager->pruneStalledRuns(); |
|
| 34 | self::assertEquals(1, $count); |
|
| 35 | self::assertEmpty($runRepository->findAll()); |
|
| 36 | $count = $runManager->pruneStalledRuns(); |
|
| 37 | self::assertEquals(0, $count); |
|
| 38 | } |
|
| 39 | } |
|
| 40 | ||
| @@ 43-71 (lines=29) @@ | ||
| 40 | self::assertEmpty($runRepository->findAll()); |
|
| 41 | } |
|
| 42 | ||
| 43 | public function testPruneStaleRuns() |
|
| 44 | { |
|
| 45 | JobManagerTest::setUpBeforeClass(); |
|
| 46 | $jobManager = JobManagerTest::$jobManager; |
|
| 47 | $runClass = \Dtc\QueueBundle\Document\Run::class; |
|
| 48 | $runArchiveClass = \Dtc\QueueBundle\Document\RunArchive::class; |
|
| 49 | $runManager = new \Dtc\QueueBundle\ODM\RunManager($jobManager->getObjectManager(), $runClass, RunArchive::class); |
|
| 50 | $objectManager = $runManager->getObjectManager(); |
|
| 51 | $runRepository = $objectManager->getRepository($runClass); |
|
| 52 | self::assertEmpty($runRepository->findAll()); |
|
| 53 | $runArchiveRepository = $objectManager->getRepository($runArchiveClass); |
|
| 54 | self::assertEmpty($runArchiveRepository->findAll()); |
|
| 55 | ||
| 56 | $run = new $runClass(); |
|
| 57 | $time = time() - 96400; |
|
| 58 | $date = new \DateTime("@$time"); |
|
| 59 | ||
| 60 | $run->setStartedAt($date); |
|
| 61 | $run->setLastHeartbeatAt($date); |
|
| 62 | $objectManager->persist($run); |
|
| 63 | $objectManager->flush($run); |
|
| 64 | self::assertCount(1, $runRepository->findAll()); |
|
| 65 | ||
| 66 | $count = $runManager->pruneStalledRuns(); |
|
| 67 | self::assertEquals(1, $count); |
|
| 68 | self::assertEmpty($runRepository->findAll()); |
|
| 69 | $count = $runManager->pruneStalledRuns(); |
|
| 70 | self::assertEquals(0, $count); |
|
| 71 | } |
|
| 72 | } |
|
| 73 | ||
