Merge pull request #8 from chadrosen/add-support-for-multiple-groups

add suport for multi-groups and tests

lib/ext/bundler_fake_dsl.rb

@@ -1,6 +1,6 @@
 module Bundler
   class FakeDsl
-    
+
     def initialize(file)
       @groups = Hash.new{|h, k| h[k] = []}
       @gems = []
@@ -19,18 +19,20 @@ def __gem_names
       __gems.map(&:first)
     end
 
-    def group(name, &blk)
-      @current_group = name.to_sym
-      instance_eval(&blk)
-      @current_group = nil
+    def group(*group_names, &blk)
+      group_names.each do |name|
+        @current_group = name.to_sym
+        instance_eval(&blk)
+        @current_group = nil
+      end
     end
     alias_method :groups, :group
 
     def gem(*args)
       @gems << args
       @groups[@current_group] << args.first
     end
-    
+
     def method_missing(*args)
     end
   end

spec/spitball_spec.rb

@@ -305,3 +305,55 @@
     end
   end
 end
+
+describe Bundler::FakeDsl do
+  it "should support a single group" do
+    gemfile = <<-end_gemfile
+        source :rubygems
+        gem "json_pure"
+
+        group :development do
+          gem 'rails'
+          gem 'json_pure'
+        end
+      end_gemfile
+
+    dsl = Bundler::FakeDsl.new(gemfile)
+    dsl.__groups[:development].should == ["rails", "json_pure"]
+  end
+
+  it "should support multiple groups in one line" do
+    gemfile = <<-end_gemfile
+        source :rubygems
+        gem "json_pure"
+
+        group :development, :test do
+          gem 'rails'
+          gem 'json_pure'
+        end
+      end_gemfile
+
+    dsl = Bundler::FakeDsl.new(gemfile)
+    dsl.__groups[:development].should == ["rails", "json_pure"]
+    dsl.__groups[:test].should == ["rails", "json_pure"]
+  end
+
+  it "should support multiple groups on several lines" do
+    gemfile = <<-end_gemfile
+        source :rubygems
+        gem "json_pure"
+
+        group :development do
+          gem 'rails'
+        end
+
+        group :test do
+          gem 'json_pure'
+        end
+      end_gemfile
+
+    dsl = Bundler::FakeDsl.new(gemfile)
+    dsl.__groups[:development].should == ["rails"]
+    dsl.__groups[:test].should == ["json_pure"]
+  end
+end