Adding tests for other object types.

cakephp · May 25, 2009 · 66b36ba · 66b36ba
1 parent 975aab8
commit 66b36ba
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/cake/console/libs/tasks/test.php b/cake/console/libs/tasks/test.php
@@ -390,12 +390,12 @@ function generateMockClass($type, $class) {
 	function generateConstructor($type, $fullClassName) {
 		$type = strtolower($type);
 		if ($type == 'model') {
-			return "ClassRegistry::init('$fullClassName');";
+			return "ClassRegistry::init('$fullClassName');\n";
 		}
 		if ($type == 'controller') {
-			return "new Test$fullClassName();\n\t\t\$this->{$fullClassName}->constructClasses();";
+			return "new Test$fullClassName();\n\t\t\$this->{$fullClassName}->constructClasses();\n";
 		}
-		return "new $fullClassName";
+		return "new $fullClassName()\n";
 	}
 
 /**

diff --git a/cake/tests/cases/console/libs/tasks/test.test.php b/cake/tests/cases/console/libs/tasks/test.test.php
@@ -312,7 +312,15 @@ function testBake() {
  **/
 	function testGenerateContsructor() {
 		$result = $this->Task->generateConstructor('controller', 'PostsController');
-		$expected = "new TestPostsController();\n\t\t\$this->PostsController->constructClasses();";
+		$expected = "new TestPostsController();\n\t\t\$this->PostsController->constructClasses();\n";
+		$this->assertEqual($result, $expected);
+
+		$result = $this->Task->generateConstructor('model', 'Post');
+		$expected = "ClassRegistry::init('Post');\n";
+		$this->assertEqual($result, $expected);
+
+		$result = $this->Task->generateConstructor('helper', 'FormHelper');
+		$expected = "new FormHelper()\n";
 		$this->assertEqual($result, $expected);
 	}
 }