From 1e17a9d367c54c680368be72f44247ae28b98904 Mon Sep 17 00:00:00 2001
From: Steve Jorgensen <stevej@renewfund.com>
Date: Wed, 8 Aug 2012 14:43:53 -0700
Subject: [PATCH] Fix only-once stub logic.

Didn't fail the test because adapter#query happens to
not call raw connection's #query, but don't want to count
on that and have a fragile test.
---
 activerecord/test/cases/adapters/postgresql/connection_test.rb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/activerecord/test/cases/adapters/postgresql/connection_test.rb b/activerecord/test/cases/adapters/postgresql/connection_test.rb
index 47089d5d0d..1ff307c735 100644
--- a/activerecord/test/cases/adapters/postgresql/connection_test.rb
+++ b/activerecord/test/cases/adapters/postgresql/connection_test.rb
@@ -92,9 +92,9 @@ module ActiveRecord
         def query_fake(*args)
           if !( @called ||= false )
             self.stubs(:status).returns(PGconn::CONNECTION_BAD)
+            @called = true
             raise PGError
           else
-            @called = true
             self.unstub(:status)
             query_unfake(*args)
           end