Container will flush an restore each fixture loader

app/code/community/EcomDev/PHPUnit/Model/Fixture/Loader/Container.php

@@ -61,4 +61,32 @@ class EcomDev_PHPUnit_Model_Fixture_Loader_Container
     {
         return isset($this->_loaders[$code]);
     }
-}
\ No newline at end of file
+
+
+    /**
+     * Flush all loader in this container.
+     *
+     * @return void
+     */
+    public function flush()
+    {
+        foreach ($this->_loaders as $loader)
+        {
+            $loader->flush();
+        }
+    }
+
+
+    /**
+     * Restore all loader in this container.
+     *
+     * @return void
+     */
+    public function restore()
+    {
+        foreach ($this->_loaders as $loader)
+        {
+            $loader->restore();
+        }
+    }
+}

app/code/community/EcomDev/PHPUnitTest/Test/Model/Fixture/Loader/Container.php

@@ -9,7 +9,7 @@ class EcomDev_PHPUnitTest_Test_Model_Fixture_Loader_Container
      * @var EcomDev_PHPUnit_Model_Fixture_Loader_Container
      */
     protected $_factory;
-    
+
     protected function setUp()
     {
         $this->_factory = new EcomDev_PHPUnit_Model_Fixture_Loader_Container();
@@ -79,6 +79,45 @@ class EcomDev_PHPUnitTest_Test_Model_Fixture_Loader_Container
         );
     }
 
+    /**
+     * Check if the methods are forwarded by the collection to it's elements.
+     *
+     * @dataProvider getMethods
+     *
+     * @param string $methodName
+     * @param array  $args []
+     *
+     * @return void
+     */
+    public function testFlushAndRestoreWillBeForwardedToEachLoader($methodName, $args)
+    {
+        $loaderSet = $this->_stubLoaders(2);
+
+        foreach ($loaderSet as $loader)
+        {
+            /** @var PHPUnit_Framework_MockObject_MockObject $loader */
+            $loader->expects($this->exactly(1))->method($methodName);
+        }
+
+        $this->_factory->$methodName();
+    }
+
+
+    /**
+     * Data provider for methods to check forwarding with.
+     *
+     * @see testFlushAndRestoreWillBeForwardedToEachLoader
+     *
+     * @return array
+     */
+    public function getMethods()
+    {
+        return array(
+            array('flush', array()),
+            array('restore', array()),
+        );
+    }
+
     /**
      * Generates loaders for tests
      *
@@ -112,4 +151,4 @@ class EcomDev_PHPUnitTest_Test_Model_Fixture_Loader_Container
         return $loaders;
     }
 
-}
\ No newline at end of file
+}