2 files changed, 138 insertions, 0 deletions
diff --git a/test/generators/controller_generator_test.rb b/test/generators/controller_generator_test.rb
new file mode 100644
index 000000000..46aeac88e
--- /dev/null
+++ b/test/generators/controller_generator_test.rb
@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+# Redmine - project management software
+# Copyright (C) 2006-  Jean-Philippe Lang
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+
+require_relative '../test_helper'
+require_relative '../../lib/generators/redmine_plugin_controller/redmine_plugin_controller_generator'
+
+class ControllerGeneratorTest < Rails::Generators::TestCase
+  TMP_DIR = Rails.root / 'tmp/test/generators'
+
+  tests RedminePluginControllerGenerator
+  destination TMP_DIR
+  setup :prepare_destination
+
+  setup do
+    @plugin_directory = Redmine::Plugin.directory
+    Redmine::Plugin.directory = TMP_DIR
+  end
+
+  teardown do
+    Redmine::Plugin.directory = @plugin_directory
+  end
+
+  def test_generates_files_from_templates
+    g = generator ['ControllerDemo', 'Todo']
+
+    assert_name g, 'Todo', :controller
+
+    capture(:stdout) do
+      g.copy_templates
+    end
+
+    controller_path_names = (Redmine::Plugin.directory / 'controller_demo/app/controllers')
+      .glob('*.rb')
+    assert_equal 1, controller_path_names.count
+    assert_equal 'todo_controller.rb', controller_path_names.first.basename.to_s
+
+    helper_path_names = (Redmine::Plugin.directory / 'controller_demo/app/helpers')
+      .glob('*.rb')
+    assert_equal 1, helper_path_names.count
+    assert_equal 'todo_helper.rb', helper_path_names.first.basename.to_s
+
+    test_path_names = (Redmine::Plugin.directory / 'controller_demo/test/functional')
+      .glob('*.rb')
+    assert_equal 1, test_path_names.count
+    assert_equal 'todo_controller_test.rb', test_path_names.first.basename.to_s
+  end
+
+  private
+
+  def assert_name(generator, value, method)
+    assert_equal value, generator.send(method)
+  end
+end
diff --git a/test/generators/model_generator_test.rb b/test/generators/model_generator_test.rb
new file mode 100644
index 000000000..f030de6f7
--- /dev/null
+++ b/test/generators/model_generator_test.rb
@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+# Redmine - project management software
+# Copyright (C) 2006-  Jean-Philippe Lang
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+
+require_relative '../test_helper'
+require_relative '../../lib/generators/redmine_plugin_model/redmine_plugin_model_generator'
+
+class ModelGeneratorTest < Rails::Generators::TestCase
+  TMP_DIR = Rails.root / 'tmp/test/generators'
+
+  tests RedminePluginModelGenerator
+  destination TMP_DIR
+  setup :prepare_destination
+
+  setup do
+    @plugin_directory = Redmine::Plugin.directory
+    Redmine::Plugin.directory = TMP_DIR
+  end
+
+  teardown do
+    Redmine::Plugin.directory = @plugin_directory
+  end
+
+  def test_generates_files_from_templates
+    g = generator ['ModelDemo', 'TodoModel']
+
+    assert_name g, 'TodoModel', :model
+
+    capture(:stdout) do
+      g.copy_templates
+    end
+
+    model_path_names = (Redmine::Plugin.directory / 'model_demo/app/models')
+      .glob('*.rb')
+    assert_equal 1, model_path_names.count
+    assert_equal 'todo_model.rb', model_path_names.first.basename.to_s
+
+    test_path_names = (Redmine::Plugin.directory / 'model_demo/test/unit')
+      .glob('*.rb')
+    assert_equal 1, test_path_names.count
+    assert_equal 'todo_model_test.rb', test_path_names.first.basename.to_s
+
+    migration_path_names = (Redmine::Plugin.directory / 'model_demo/db/migrate')
+      .glob('*.rb')
+    assert_equal 1, migration_path_names.count
+    assert_match(/\d+_create_todo_models\.rb/, migration_path_names.first.basename.to_s)
+  end
+
+  private
+
+  def assert_name(generator, value, method)
+    assert_equal value, generator.send(method)
+  end
+end