Replace _test too as seen in some test files

luqman · Jun 18, 2012 · 56feb65 · 56feb65
1 parent 9aaba3b
commit 56feb65
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/Rails.py b/Rails.py
@@ -116,7 +116,7 @@ def for_models(app_folder, working_directory, file_name_base_no_ext):
   def for_tests(app_folder, working_directory, base_file_name):
 
     if '_controller' in base_file_name:
-      controller = base_file_name.replace('_controller', '').replace('_spec', '').replace('test_', '')
+      controller = base_file_name.replace('_controller', '').replace('_spec', '').replace('_test', '').replace('test_', '')
       model = Inflector(English).singularize(controller).lower()
     else:
       model = base_file_name.replace('_spec', '').replace('test_', '')

diff --git a/Rails.pyc b/Rails.pyc