Add a test for PR #153

cakephp · Nov 26, 2015 · cc48233 · cc48233
1 parent 7da0df7
commit cc48233
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/tests/TestCase/Shell/Task/MigrationSnapshotTaskTest.php b/tests/TestCase/Shell/Task/MigrationSnapshotTaskTest.php
@@ -49,7 +49,7 @@ public function setUp()
 
         $this->Task = $this->getMock(
             'Migrations\Shell\Task\MigrationSnapshotTask',
-            ['in', 'err', 'dispatchShell', '_stop'],
+            ['in', 'err', 'dispatchShell', '_stop', 'findTables', 'fetchTableName'],
             [$inputOutput]
         );
         $this->Task->name = 'Migration';
@@ -59,6 +59,21 @@ public function setUp()
         $this->Task->BakeTemplate->interactive = false;
     }
 
+    public function testGetTableNames()
+    {
+        $this->Task->expects($this->any())
+            ->method('findTables')
+            ->with('Blog')
+            ->will($this->returnValue(['ArticlesTable.php', 'TagsTable.php']));
+
+        $this->Task->method('fetchTableName')
+            ->will($this->onConsecutiveCalls(['articles_tags', 'articles'], ['articles_tags', 'tags']));
+
+        $results = $this->Task->getTableNames('Blog');
+        $expected = ['articles_tags', 'articles', 'tags'];
+        $this->assertEquals(array_values($expected), array_values($results));
+    }
+
     public function testNotEmptySnapshot()
     {
         $this->Task->params['require-table'] = false;

diff --git a/tests/bootstrap.php b/tests/bootstrap.php
@@ -71,3 +71,4 @@
 Plugin::load('Migrations', [
     'path' => dirname(dirname(__FILE__)) . DS,
 ]);
+Plugin::load('Blog');
diff --git a/tests/test_app/Plugin/Blog/empty b/tests/test_app/Plugin/Blog/empty