test/test_helper.rb in contextr-0.1.0 vs test/test_helper.rb in contextr-0.1.1

@@ -1,2 +1,57 @@
 require 'test/unit'
 require File.dirname(__FILE__) + '/../lib/contextr'
+
+unless Object.const_defined?("ExampleTest")
+  module ExampleTest
+    module ObjectExtension
+      def test_class(name)
+        $latest_test_class = Class.new(Test::Unit::TestCase)
+        $latest_test_case  = 0
+        Object.const_set(name, $latest_test_class) 
+      end
+
+      def example(&block)
+        $latest_test_class.class_eval do
+          define_method("test_%03d" % ($latest_test_case += 1), &block)
+        end
+      end
+    end
+    
+    module TestExtension
+      def assert_to_s(expected, actual)
+        assert_equal(expected, actual.to_s)
+      end
+
+      def result_of(object)
+        Result.new(object, self)
+      end
+
+      def output_of(object)
+        Output.new(object, self)
+      end
+
+      class Result 
+        attr_accessor :object, :test_class
+        def initialize(object, test_class)
+          self.object = object
+          self.test_class = test_class 
+        end
+        def ==(string)
+          test_class.assert_equal(string, object)
+        end
+      end
+      class Output < Result
+        def ==(string)
+          test_class.assert_equal(string, object.to_s)
+        end
+      end
+    end
+  end
+
+  class Test::Unit::TestCase
+    include ExampleTest::TestExtension
+  end
+  class Object 
+    include ExampleTest::ObjectExtension
+  end
+end