test/test_helper.rb in espresso-0.1.12 vs test/test_helper.rb in espresso-0.2.1

@@ -1,10 +1,51 @@
 require 'rubygems'
 require 'test/unit'
 require 'shoulda'
+require 'redgreen'
 
 $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 $LOAD_PATH.unshift(File.dirname(__FILE__))
 require 'espresso'
 
-class Test::Unit::TestCase
+require 'active_support/core_ext/string'
+module Test
+  module Unit
+    class TestCase
+      def self.should_have_class_methods(*methods)
+        get_options!(methods)
+        klass = described_type
+        methods.each do |method|
+          should "respond to class method ##{method}" do
+            assert_respond_to(klass, method, "#{klass.name} does not have class method #{method}")
+          end
+        end
+      end
+
+      def self.should_have_instance_methods(*methods)
+        get_options!(methods)
+        klass = described_type
+        instance = if block_given?
+                     # Class initializer requires some attributes,
+                     # or another custom generator used
+                     yield
+                   elsif klass.is_a?(Class)
+                     klass.new
+                   else
+                     # klass is a module
+                     metaclass = Class.new
+                     metaclass.send(:include, klass)
+                     metaclass.new
+                   end
+        methods.each do |method|
+          should "respond to instance method ##{method}" do
+            assert_respond_to(instance, method, "#{klass.name} does not have instance method #{method}")
+          end
+        end
+      end
+    end
+
+    class TestCaseTest < TestCase
+      should_have_class_methods :should_have_class_methods, :should_have_instance_methods
+    end
+  end
 end