gemfile entries can be added to the app builder

1 files changed, 35 insertions, 0 deletions

diff --git a/railties/test/generators/generator_test.rb b/railties/test/generators/generator_test.rb
index 7600a40ae9..dd4868585b 100644
--- a/railties/test/generators/generator_test.rb
+++ b/railties/test/generators/generator_test.rb
@@ -14,6 +14,7 @@ module Rails
             def gemfile_entries; super; end
             def invoke_all; super; self; end
             def add_gem_entry_filter; super; end
+            def gemfile_entry(*args); super; end
           })
         end
       end
@@ -23,6 +24,40 @@ module Rails
         assert klass.start(['new', 'blah'])
       end
 
+      def test_add_gem
+        klass = make_builder_class
+        generator = klass.start(['new', 'blah'])
+        generator.gemfile_entry 'tenderlove'
+        assert_includes generator.gemfile_entries.map(&:name), 'tenderlove'
+      end
+
+      def test_add_gem_with_version
+        klass = make_builder_class
+        generator = klass.start(['new', 'blah'])
+        generator.gemfile_entry 'tenderlove', '2.0.0'
+        assert generator.gemfile_entries.find { |gfe|
+          gfe.name == 'tenderlove' && gfe.version == '2.0.0'
+        }
+      end
+
+      def test_add_github_gem
+        klass = make_builder_class
+        generator = klass.start(['new', 'blah'])
+        generator.gemfile_entry 'tenderlove', github: 'hello world'
+        assert generator.gemfile_entries.find { |gfe|
+          gfe.name == 'tenderlove' && gfe.options[:github] == 'hello world'
+        }
+      end
+
+      def test_add_path_gem
+        klass = make_builder_class
+        generator = klass.start(['new', 'blah'])
+        generator.gemfile_entry 'tenderlove', path: 'hello world'
+        assert generator.gemfile_entries.find { |gfe|
+          gfe.name == 'tenderlove' && gfe.options[:path] == 'hello world'
+        }
+      end
+
       def test_filter
         klass     = make_builder_class
         generator = klass.start(['new', 'blah'])