Fix preloadable assoc test, fix test for scrooge_installed? for assoc…

…iations
methodmissing · Mar 22, 2009 · 8930328 · 8930328
1 parent 1ca441a
commit 8930328
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/optimizations/associations/macro.rb b/lib/optimizations/associations/macro.rb
@@ -17,7 +17,7 @@ def install!
           protected
 
             def scrooge_installed?
-              ActiveRecord::Base.included_modules.include?( InstanceMethods )
+              ActiveRecord::Associations::AssociationProxy.included_modules.include?( InstanceMethods )
             end
 
         end

diff --git a/test/optimizations/associations/macro_test.rb b/test/optimizations/associations/macro_test.rb
@@ -17,7 +17,7 @@ class OptimizationsAssociationsMacroTest < ActiveSupport::TestCase
   test "should only flag preloadable associations" do
     Scrooge::Callsite.any_instance.expects(:association!).once
     @user = MysqlUser.find(:first)
-    @user.table_privileges
+    @user.host
     assert_equal MysqlUser.scrooge_callsite( @user.callsite_signature ).associations, Set.new
   end