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