Conditionally do inflections used in TestTask.

This hack is required due to cakephp/bake#438.
HavokInspiration · Jun 2, 2018 · 46decaa · 46decaa
1 parent 5142505
commit 46decaa
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/src/Shell/Task/MaintenanceModeTask.php b/src/Shell/Task/MaintenanceModeTask.php
@@ -12,6 +12,7 @@
 namespace Wrench\Shell\Task;
 
 use Bake\Shell\Task\SimpleBakeTask;
+use Cake\Utility\Inflector;
 
 /**
  * Bake task responsible of generating Maintenance Mode skeleton
@@ -56,13 +57,20 @@ public function template()
      */
     public function bakeTest($className)
     {
-        if (!isset($this->Test->classSuffixes[$this->name()])) {
-            $this->Test->classSuffixes[$this->name()] = '';
+        $suffixName = $typeName = $this->name();
+
+        if (isset($this->Test->classSuffixes['entity'])) {
+            $typeName = ucfirst($typeName);
+        } else {
+            $suffixName = $typeName = Inflector::camelize($typeName);
+        }
+
+        if (!isset($this->Test->classSuffixes[$suffixName])) {
+            $this->Test->classSuffixes[$suffixName] = '';
         }
 
-        $name = ucfirst($this->name());
-        if (!isset($this->Test->classTypes[$name])) {
-            $this->Test->classTypes[$name] = 'Maintenance\Mode';
+        if (!isset($this->Test->classTypes[$typeName])) {
+            $this->Test->classTypes[$typeName] = 'Maintenance\Mode';
         }
 
         return parent::bakeTest($className);