Added extensions to evergreen

qrush · Jan 14, 2011 · c10f789 · c10f789
1 parent b11394b
commit c10f789
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/lib/evergreen.rb b/lib/evergreen.rb
@@ -21,6 +21,11 @@ def configure
       yield self
     end
 
+    def extensions(&block)
+      @extensions = block if block
+      @extensions
+    end
+
     def use_defaults!
       configure do |config|
         config.driver = :selenium

diff --git a/lib/evergreen/application.rb b/lib/evergreen/application.rb
@@ -1,6 +1,8 @@
 module Evergreen
   def self.application(suite)
     Rack::Builder.new do
+      instance_eval(&Evergreen.extensions) if Evergreen.extensions
+
       map "/jasmine" do
         use Rack::Static, :urls => ["/"], :root => File.expand_path('../jasmine/lib', File.dirname(__FILE__))
         run lambda { |env| [404, {}, "No such file"]}

diff --git a/spec/evergreen_spec.rb b/spec/evergreen_spec.rb
@@ -21,4 +21,9 @@
     page.should have_content("2 specs, 1 failure")
     page.should have_content("Expected 'bar' to equal 'noooooo'.")
   end
+
+  it "should add extensions to Evergreen" do
+    visit('/awesome')
+    page.should have_content('Totally awesome')
+  end
 end
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -9,6 +9,12 @@
 
 TEST_DRIVER = :envjs
 
+Evergreen.extensions do
+  map "/awesome" do
+    run lambda { |env| [200, {}, "<html><body>Totally awesome</body></html>"]}
+  end
+end
+
 Capybara.app = Evergreen::Suite.new(File.expand_path('suite1', File.dirname(__FILE__))).application
 Capybara.default_driver = TEST_DRIVER