/**
  * @param Context $context
  */
 public function initializeContext(Context $context)
 {
     if ($context instanceof PageObjectAware) {
         $context->setPageObjectFactory($this->pageObjectFactory);
     }
 }
 function it_should_inject_the_page_factory_into_the_context(Context $context, PageObjectFactory $pageObjectFactory)
 {
     $context->implement('SensioLabs\\Behat\\PageObjectExtension\\Context\\PageObjectAware');
     $context->setPageObjectFactory($pageObjectFactory)->shouldBeCalled();
     $this->initializeContext($context)->shouldReturn(null);
 }