DG.enhance! should include Fluent and Bothback

samstokes · Jan 5, 2011 · 91c650c · 91c650c
1 parent 946442f
commit 91c650c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/lib/deferrable_gratification.rb b/lib/deferrable_gratification.rb
@@ -17,6 +17,8 @@ module DeferrableGratification
   # included that module.
   def self.enhance!(module_or_class)
     module_or_class.send :include, Combinators
+    module_or_class.send :include, Fluent
+    module_or_class.send :include, Bothback
   end
 
   # Enhance EventMachine::Deferrable itself so that any class including it

diff --git a/spec/deferrable_gratification/default_deferrable_spec.rb b/spec/deferrable_gratification/default_deferrable_spec.rb
@@ -3,5 +3,7 @@
 describe DeferrableGratification::DefaultDeferrable do
   it_should_behave_like 'a Deferrable'
 
+  it_should_include DeferrableGratification::Fluent
+  it_should_include DeferrableGratification::Bothback
   it_should_include DeferrableGratification::Combinators
 end
diff --git a/spec/deferrable_gratification/deferrable_gratification_spec.rb b/spec/deferrable_gratification/deferrable_gratification_spec.rb
@@ -9,6 +9,8 @@ class EnhancedDeferrable < EventMachine::DefaultDeferrable
   end
 
   describe EnhancedDeferrable do
+    it_should_include DeferrableGratification::Fluent
+    it_should_include DeferrableGratification::Bothback
     it_should_include DeferrableGratification::Combinators
   end
 end