tickets controller tests passing

1 files changed, 9 insertions, 4 deletions

diff --git a/users/test/support/stub_record_helper.rb b/users/test/support/stub_record_helper.rb
index 18db5ae..168a827 100644
--- a/users/test/support/stub_record_helper.rb
+++ b/users/test/support/stub_record_helper.rb
@@ -4,10 +4,11 @@ module StubRecordHelper
   # return the record given.
   # If no record is given but a hash or nil will create a stub based on
   # that instead and returns the stub.
-  def find_record(factory)
-    record = stub_record factory
+  def find_record(factory, attribs_hash = {})
+    attribs_hash.reverse_merge!(:id => Random.rand(10000).to_s)
+    record = stub_record factory, attribs_hash
     klass = record.class
-    finder = klass.respond_to?(:find_by_param) ? :find_by_param : :find_by_id
+    finder = klass.respond_to?(:find_by_param) ? :find_by_param : :find
     klass.expects(finder).with(record.to_param.to_s).returns(record)
     return record
   end
@@ -18,11 +19,15 @@ module StubRecordHelper
   # If the second parameter is a record we return the record itself.
   # This way you can build functions that either take a record or a
   # method hash to stub from. See find_record for an example.
-  def stub_record(factory, record_or_method_hash = {})
+  def stub_record(factory, record_or_method_hash = {}, persisted=false)
     if record_or_method_hash && !record_or_method_hash.is_a?(Hash)
       return record_or_method_hash
     end
     FactoryGirl.build_stubbed(factory).tap do |record|
+      if persisted or record.persisted?
+        record_or_method_hash.reverse_merge! :created_at => Time.now,
+          :updated_at => Time.now, :id => Random.rand(100000).to_s
+      end
       record.stubs(record_or_method_hash) if record_or_method_hash.present?
     end
   end