/** * Test ongr:repository-crawler:crawl progress helper behavior. */ public function testCommandProgress() { $document1 = $this->getMockForAbstractClass('ONGR\\ElasticsearchBundle\\Document\\DocumentInterface'); $source = $this->getMockForAbstractClass('ONGR\\ConnectionsBundle\\EventListener\\AbstractCrawlerSource'); $modifier = $this->getMockForAbstractClass('ONGR\\ConnectionsBundle\\EventListener\\AbstractCrawlerModifier'); $container = new ContainerBuilder(); $command = new RepositoryCrawlerCommand(); $command->setContainer($container); $input = $this->getMock('Symfony\\Component\\Console\\Input\\InputInterface'); $input->expects($this->any())->method('getOption')->with('action-name')->will($this->returnValue('default')); $writes = 0; $callback = function () use(&$writes) { $writes++; }; $output = $this->getMock('Symfony\\Component\\Console\\Output\\OutputInterface'); $outputFormatter = $this->getMock('Symfony\\Component\\Console\\Formatter\\OutputFormatterInterface'); $output->expects($this->any())->method('write')->will($this->returnCallback($callback)); $output->expects($this->any())->method('isDecorated')->will($this->returnValue(true)); $output->expects($this->any())->method('getFormatter')->will($this->returnValue($outputFormatter)); $itemEvent = new ItemPipelineEvent($document1); $dispatcher = $this->getMock('Symfony\\Component\\EventDispatcher\\EventDispatcherInterface'); $dispatcher->expects($this->exactly(3))->method('dispatch')->withConsecutive(['ongr.pipeline.repository_crawler.default.source', $this->anything()], ['ongr.pipeline.repository_crawler.default.start', $this->anything()], ['ongr.pipeline.repository_crawler.default.finish', $this->anything()], ['ongr.pipeline.repository_crawler.default.modify', $this->anything()])->willReturnOnConsecutiveCalls($this->returnValue($source->onSource(new SourcePipelineEvent())), $this->returnValue(null), $this->returnValue(null), $this->returnValue($modifier->onModify($itemEvent))); $pipelineFactory = new PipelineFactory(); $pipelineFactory->setClassName('\\ONGR\\ConnectionsBundle\\Pipeline\\Pipeline'); $pipelineFactory->setDispatcher($dispatcher); $crawler = new PipelineStarter(); $crawler->setPipelineFactory($pipelineFactory); $container->set('ongr_connections.repository_crawler_service', $crawler); $command->run($input, $output); if ($writes < 2) { $this->fail("Console output was expected to be written at least 2 times, actually written {$writes} times."); } }
/** * Tests pipeline with progressbar. * * @param array $data * @param int $expectedConsumedItems * @param int $expectedSkippedItems * * @dataProvider PipelineData */ public function testPipelineProgress($data, $expectedConsumedItems, $expectedSkippedItems) { $pipelineFactory = new PipelineFactory(); $pipelineFactory->setDispatcher(new EventDispatcher()); $pipelineFactory->setClassName('ONGR\\ConnectionsBundle\\Pipeline\\Pipeline'); $expectedContext = 'This is a test of context'; $progressBar = $this->getMockBuilder('Symfony\\Component\\Console\\Helper\\ProgressBar')->disableOriginalConstructor()->getMock(); $progressBar->expects($this->once())->method('start'); $progressBar->expects($this->exactly(count($data)))->method('advance'); $progressBar->expects($this->once())->method('finish'); $consumer = new PipelineTestConsumer(); $source = function (SourcePipelineEvent $event) use($data, $expectedContext) { $event->addSource($data); $event->setContext($expectedContext); }; $pipeline = $pipelineFactory->create('test', ['sources' => [$source], 'modifiers' => [[$this, 'onModify']], 'consumers' => [[$consumer, 'onConsume']]]); $pipeline->setProgressBar($progressBar); $pipeline->start(); $this->assertEquals($expectedConsumedItems, $consumer->getConsumeCalled()); $this->assertEquals($expectedSkippedItems, $consumer->getSkipCalled()); $this->assertEquals($expectedContext, $pipeline->getContext()); }
/** * Test pipeline factory exception. * * @expectedException \InvalidArgumentException */ public function testPipelineFactoryException() { $pipelineFactory = new PipelineFactory(); $pipelineFactory->setClassName('ONGR\\ConnectionsBundle\\Pipeline\\Event\\SourcePipelineEvent'); $pipelineFactory->create(null); }