diff --git a/tests/unit/extensions/mongodb/console/controllers/EchoMigrateController.php b/tests/unit/extensions/mongodb/console/controllers/EchoMigrateController.php
new file mode 100644
index 0000000000..3d5cc0c818
--- /dev/null
+++ b/tests/unit/extensions/mongodb/console/controllers/EchoMigrateController.php
@@ -0,0 +1,18 @@
+<?php
+namespace yiiunit\extensions\mongodb\console\controllers;
+
+use yii\mongodb\console\controllers\MigrateController;
+
+/**
+ * MigrateController that writes output via echo instead of using output stream. Allows us to buffer it.
+ */
+class EchoMigrateController extends MigrateController
+{
+    /**
+     * @inheritdoc
+     */
+    public function stdout($string)
+    {
+        echo $string;
+    }
+}
\ No newline at end of file
diff --git a/tests/unit/extensions/mongodb/console/controllers/MigrateControllerTest.php b/tests/unit/extensions/mongodb/console/controllers/MigrateControllerTest.php
index 288f7ee989..7af784dc9a 100644
--- a/tests/unit/extensions/mongodb/console/controllers/MigrateControllerTest.php
+++ b/tests/unit/extensions/mongodb/console/controllers/MigrateControllerTest.php
@@ -8,7 +8,6 @@ use yii\mongodb\Query;
 use Yii;
 use yiiunit\extensions\mongodb\MongoDbTestCase;
 use yiiunit\framework\console\controllers\MigrateControllerTestTrait;
-use yii\mongodb\console\controllers\MigrateController;
 
 /**
  * Unit test for [[\yii\mongodb\console\controllers\MigrateController]].
@@ -23,7 +22,7 @@ class MigrateControllerTest extends MongoDbTestCase
 
     public function setUp()
     {
-        $this->migrateControllerClass = MigrateController::className();
+        $this->migrateControllerClass = EchoMigrateController::className();
         $this->migrationBaseClass = Migration::className();
 
         parent::setUp();
diff --git a/tests/unit/framework/console/controllers/CacheControllerTest.php b/tests/unit/framework/console/controllers/CacheControllerTest.php
index 8c031cd2ef..714c6f150e 100644
--- a/tests/unit/framework/console/controllers/CacheControllerTest.php
+++ b/tests/unit/framework/console/controllers/CacheControllerTest.php
@@ -16,7 +16,7 @@ class CacheControllerTest extends TestCase
 {
 
     /**
-     * @var \yiiunit\framework\console\controllers\CacheConsoledController
+     * @var SilencedCacheController
      */
     private $_cacheController;
 
@@ -27,7 +27,7 @@ class CacheControllerTest extends TestCase
         parent::setUp();
 
         $this->_cacheController = Yii::createObject([
-            'class' => 'yiiunit\framework\console\controllers\CacheConsoledController',
+            'class' => 'yiiunit\framework\console\controllers\SilencedCacheController',
             'interactive' => false,
         ],[null, null]); //id and module are null
         
@@ -122,7 +122,7 @@ class CacheControllerTest extends TestCase
     }
 
     /**
-     * @expectedException yii\console\Exception
+     * @expectedException \yii\console\Exception
      */
     public function testNothingToFlushException()
     {
@@ -141,12 +141,3 @@ class CacheControllerTest extends TestCase
     }
 
 }
-
-class CacheConsoledController extends CacheController
-{
-
-    public function stdout($string)
-    {
-    }
-
-}
diff --git a/tests/unit/framework/console/controllers/EchoMigrateController.php b/tests/unit/framework/console/controllers/EchoMigrateController.php
new file mode 100644
index 0000000000..9e362bd80b
--- /dev/null
+++ b/tests/unit/framework/console/controllers/EchoMigrateController.php
@@ -0,0 +1,18 @@
+<?php
+namespace yiiunit\framework\console\controllers;
+
+use yii\console\controllers\MigrateController;
+
+/**
+ * MigrateController that writes output via echo instead of using output stream. Allows us to buffer it.
+ */
+class EchoMigrateController extends MigrateController
+{
+    /**
+     * @inheritdoc
+     */
+    public function stdout($string)
+    {
+        echo $string;
+    }
+}
\ No newline at end of file
diff --git a/tests/unit/framework/console/controllers/MigrateControllerTest.php b/tests/unit/framework/console/controllers/MigrateControllerTest.php
index 8854614f51..02346feaaa 100644
--- a/tests/unit/framework/console/controllers/MigrateControllerTest.php
+++ b/tests/unit/framework/console/controllers/MigrateControllerTest.php
@@ -20,7 +20,7 @@ class MigrateControllerTest extends TestCase
 
     public function setUp()
     {
-        $this->migrateControllerClass = MigrateController::className();
+        $this->migrateControllerClass = EchoMigrateController::className();
         $this->migrationBaseClass = Migration::className();
 
         $this->mockApplication([
diff --git a/tests/unit/framework/console/controllers/SilencedCacheController.php b/tests/unit/framework/console/controllers/SilencedCacheController.php
new file mode 100644
index 0000000000..40eed6ee29
--- /dev/null
+++ b/tests/unit/framework/console/controllers/SilencedCacheController.php
@@ -0,0 +1,20 @@
+<?php
+namespace yiiunit\framework\console\controllers;
+
+
+use yii\console\controllers\CacheController;
+
+
+/**
+ * CacheController that discards output.
+ */
+class SilencedCacheController extends CacheController
+{
+    /**
+     * @inheritdoc
+     */
+    public function stdout($string)
+    {
+        // do nothing
+    }
+}
\ No newline at end of file