spec/integration/logging_spec.rb in do_mysql-0.9.2 vs spec/integration/logging_spec.rb in do_mysql-0.9.3

@@ -3,31 +3,36 @@
 
 describe DataObjects::Mysql::Command do
 
   before(:each) do
     @connection = if JRUBY
-      DataObjects::Mysql::Connection.new(DO_MYSQL_SPEC_JDBC_URI)
+      DataObjects::Connection.new(DO_MYSQL_SPEC_JDBC_URI)
     else
-      DataObjects::Mysql::Connection.new(DO_MYSQL_SPEC_URI)
+      DataObjects::Connection.new(DO_MYSQL_SPEC_URI)
     end
   end
 
+  after(:each) do
+    @connection.close
+  end
+
   describe "Executing a Reader" do
 
     it "should log reader queries when the level is Debug (0)" do
       command = @connection.create_command("SELECT * FROM widgets WHERE name = ?")
       @mock_logger = mock('MockLogger', :level => 0)
       DataObjects::Mysql.should_receive(:logger).and_return(@mock_logger)
       @mock_logger.should_receive(:debug).with("SELECT * FROM widgets WHERE name = 'Scott'")
-      command.execute_reader('Scott')
+
+      command.execute_reader('Scott').close # Readers must be closed!
     end
 
     it "shouldn't log reader queries when the level isn't Debug (0)" do
       command = @connection.create_command("SELECT * FROM widgets WHERE name = ?")
       @mock_logger = mock('MockLogger', :level => 1)
       DataObjects::Mysql.should_receive(:logger).and_return(@mock_logger)
       @mock_logger.should_not_receive(:debug)
-      command.execute_reader('Scott')
+      command.execute_reader('Scott').close # Readers must be closed!
     end
   end
 
   describe "Executing a Non-Query" do
     it "should log non-query statements when the level is Debug (0)" do