From 6220ae7b4331313edff722f5b766c0ef16915f77 Mon Sep 17 00:00:00 2001
From: Adam Meehan <adam.meehan@gmail.com>
Date: Thu, 23 Sep 2010 08:13:37 +1000
Subject: [PATCH] add validation method check to each ORM

---
 .../validates_timeliness/helper_methods_spec.rb | 17 +++++++++--------
 .../orm/active_record_spec.rb                   | 12 ++++++++++++
 spec/validates_timeliness/orm/mongoid_spec.rb   | 12 ++++++++++++
 3 files changed, 33 insertions(+), 8 deletions(-)

diff --git a/spec/validates_timeliness/helper_methods_spec.rb b/spec/validates_timeliness/helper_methods_spec.rb
index a6f089a..4a5a96b 100644
--- a/spec/validates_timeliness/helper_methods_spec.rb
+++ b/spec/validates_timeliness/helper_methods_spec.rb
@@ -1,16 +1,16 @@
 require 'spec_helper'
 
 describe ValidatesTimeliness::HelperMethods do
-  it 'should define class validation methods on extended classes' do
-    ActiveRecord::Base.should respond_to(:validates_date)
-    ActiveRecord::Base.should respond_to(:validates_time)
-    ActiveRecord::Base.should respond_to(:validates_datetime)
+  it 'should define class validation methods' do
+    Person.should respond_to(:validates_date)
+    Person.should respond_to(:validates_time)
+    Person.should respond_to(:validates_datetime)
   end
 
-  it 'should define instance validation methods on extended classes' do
-    ActiveRecord::Base.instance_methods.should include('validates_date')
-    ActiveRecord::Base.instance_methods.should include('validates_time')
-    ActiveRecord::Base.instance_methods.should include('validates_datetime')
+  it 'should define instance validation methods' do
+    Person.instance_methods.should include('validates_date')
+    Person.instance_methods.should include('validates_time')
+    Person.instance_methods.should include('validates_datetime')
   end
 
   it 'should validate instance when validation method called' do
@@ -33,4 +33,5 @@ describe ValidatesTimeliness::HelperMethods do
       }
     end
   end
+
 end
diff --git a/spec/validates_timeliness/orm/active_record_spec.rb b/spec/validates_timeliness/orm/active_record_spec.rb
index 041d4af..9367e6a 100644
--- a/spec/validates_timeliness/orm/active_record_spec.rb
+++ b/spec/validates_timeliness/orm/active_record_spec.rb
@@ -1,6 +1,18 @@
 require 'spec_helper'
 
 describe ValidatesTimeliness, 'ActiveRecord' do
+  it 'should define class validation methods' do
+    ActiveRecord::Base.should respond_to(:validates_date)
+    ActiveRecord::Base.should respond_to(:validates_time)
+    ActiveRecord::Base.should respond_to(:validates_datetime)
+  end
+
+  it 'should define instance validation methods' do
+    ActiveRecord::Base.instance_methods.should include('validates_date')
+    ActiveRecord::Base.instance_methods.should include('validates_time')
+    ActiveRecord::Base.instance_methods.should include('validates_datetime')
+  end
+
   it 'should define _timeliness_raw_value_for instance method' do
     Employee.instance_methods.should include('_timeliness_raw_value_for')
   end
diff --git a/spec/validates_timeliness/orm/mongoid_spec.rb b/spec/validates_timeliness/orm/mongoid_spec.rb
index 9ad96f3..4f8928b 100644
--- a/spec/validates_timeliness/orm/mongoid_spec.rb
+++ b/spec/validates_timeliness/orm/mongoid_spec.rb
@@ -24,6 +24,18 @@ describe ValidatesTimeliness, 'Mongoid' do
     ::ValidatesTimeliness.use_plugin_parser = false
   end
 
+  it 'should define class validation methods' do
+    Article.should respond_to(:validates_date)
+    Article.should respond_to(:validates_time)
+    Article.should respond_to(:validates_datetime)
+  end
+
+  it 'should define instance validation methods' do
+    Article.instance_methods.should include('validates_date')
+    Article.instance_methods.should include('validates_time')
+    Article.instance_methods.should include('validates_datetime')
+  end
+
   it 'should define _timeliness_raw_value_for instance method' do
     Article.instance_methods.should include('_timeliness_raw_value_for')
   end