Пример #1
0
 /**
  * @return void
  */
 public function testRun()
 {
     /** @var $worker PHPUnit_Framework_MockObject_MockObject|jp_worker_Interface */
     $worker = $this->getMock('jp_worker_Interface');
     $worker->expects($this->exactly(5))->method('processItem')->with($this->isInstanceOf('ExampleData'))->will($this->returnValue(true));
     /** @var $client PHPUnit_Framework_MockObject_MockObject|JobPoolIf */
     $client = $this->getMock('JobPoolIf');
     $client->expects($this->exactly(6))->method('acquire')->will($this->returnCallback(array($this, 'mockAcquire')));
     $protocolFactory = new TBinaryProtocolFactory();
     $transport = new TMemoryBuffer();
     $protocol = $protocolFactory->getProtocol($transport);
     // Fill a mock queue backend
     for ($i = 0; $i < 5; $i++) {
         /** @var $job PHPUnit_Framework_MockObject_MockObject|Job */
         $job = $this->getMock('Job');
         $doc = new ExampleData();
         $doc->language = 'php';
         $doc->api = 'simple' . $i;
         $doc->format = 'thrift';
         $doc->write($protocol);
         $message = $transport->read($transport->available());
         $job->message = $message;
         $job->id = $i;
         $this->addToQueueBackend($job);
     }
     $consumer = new jp_consumer_Thrift('foo', array('client' => $client, 'poll_interval' => 0), $worker, 'ExampleData');
     $this->assertEquals(5, $consumer->run());
 }
Пример #2
0
 /**
  * @return void
  */
 public function testAdd()
 {
     $doc = new ExampleData();
     $doc->language = 'php';
     $doc->api = 'simple';
     $doc->format = 'thrift';
     $protocolFactory = new TBinaryProtocolFactory();
     $transport = new TMemoryBuffer();
     $protocol = $protocolFactory->getProtocol($transport);
     $doc->write($protocol);
     $message = $transport->read($transport->available());
     /** @var $client PHPUnit_Framework_MockObject_MockObject|JobPoolIf */
     $client = $this->getMock('JobPoolIf');
     $client->expects($this->once())->method('add')->with($this->equalTo('foo'), $this->equalTo($message));
     $consumer = new jp_producer_Thrift('foo', array('client' => $client));
     $consumer->add($doc);
 }