test/test_helper.rb in identity_cache-0.0.1 vs test/test_helper.rb in identity_cache-0.0.2

@@ -4,10 +4,18 @@
 require 'helpers/cache'
 require 'helpers/database_connection'
 
 require File.dirname(__FILE__) + '/../lib/identity_cache'
 
+if ENV['BOXEN_HOME'].present?
+  $memcached_port = 21211
+  $mysql_port = 13306
+else
+  $memcached_port = 11211
+  $mysql_port = 3306
+end
+
 DatabaseConnection.setup
 
 class IdentityCache::TestCase < MiniTest::Unit::TestCase
   def setup
     DatabaseConnection.drop_tables
@@ -33,10 +41,25 @@
 
   def assert_not_nil(*args)
     assert *args
   end
 
+  def assert_queries(num = 1)
+    counter = SQLCounter.new
+    subscriber = ActiveSupport::Notifications.subscribe('sql.active_record', counter)
+      yield
+    ensure
+      ActiveSupport::Notifications.unsubscribe(subscriber)
+      assert_equal num, counter.log.size, "#{counter.log.size} instead of #{num} queries were executed.#{counter.log.size == 0 ? '' : "\nQueries:\n#{counter.log.join("\n")}"}"
+  end
+
+  def assert_no_queries
+    assert_queries(0) do
+      yield
+    end
+  end
+
   def cache_hash(key)
     CityHash.hash64(key)
   end
 
   private
@@ -67,7 +90,33 @@
       klass.has_many :not_cached_records, :order => "id DESC"
       klass.has_many :polymorphic_records, :as => 'owner'
       klass.has_one :polymorphic_record, :as => 'owner'
       klass.has_one :associated, :class_name => 'AssociatedRecord', :order => "id ASC"
     }
+  end
+end
+
+class SQLCounter
+  cattr_accessor :ignored_sql
+  self.ignored_sql = [/^PRAGMA (?!(table_info))/, /^SELECT currval/, /^SELECT CAST/, /^SELECT @@IDENTITY/, /^SELECT @@ROWCOUNT/, /^SAVEPOINT/, /^ROLLBACK TO SAVEPOINT/, /^RELEASE SAVEPOINT/, /^SHOW max_identifier_length/, /^BEGIN/, /^COMMIT/]
+
+  # FIXME: this needs to be refactored so specific database can add their own
+  # ignored SQL.  This ignored SQL is for Oracle.
+  ignored_sql.concat [/^select .*nextval/i, /^SAVEPOINT/, /^ROLLBACK TO/, /^\s*select .* from all_triggers/im]
+
+  attr_reader :ignore
+  attr_accessor :log
+
+  def initialize(ignore = self.class.ignored_sql)
+    @ignore   = ignore
+    @log = []
+  end
+
+  def call(name, start, finish, message_id, values)
+    sql = values[:sql]
+
+    # FIXME: this seems bad. we should probably have a better way to indicate
+    # the query was cached
+    return if 'CACHE' == values[:name] || ignore.any? { |x| x =~ sql }
+    self.log << sql
   end
 end