- $extension1 = $this->createExtensionMock(true);
- $extension1->expects($this->once())->method('process');
- $extension2 = $this->createExtensionMock(false);
- $extension3 = $this->createExtensionMock(false);
- $extension4 = $this->createExtensionMock(true);
- $extension4->expects($this->once())->method('process');
-
- $this->container->expects($this->any())
- ->method('getExtensions')
- ->will($this->returnValue(array($extension1, $extension2, $extension3, $extension4)))
- ;
+ $extension1 = new CompilerPassExtension('extension1');
+ $extension2 = new DummyExtension('extension2');
+ $extension3 = new DummyExtension('extension3');
+ $extension4 = new CompilerPassExtension('extension4');
+
+ $this->container->registerExtension($extension1);
+ $this->container->registerExtension($extension2);
+ $this->container->registerExtension($extension3);
+ $this->container->registerExtension($extension4);