Add spec for ActiveRecord shared connection.

Rakefile

@@ -1,4 +1,10 @@
 require "bundler/gem_tasks"
 
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec
+task :release => :spec
+
 require "yard"
 YARD::Rake::YardocTask.new

lib/watir/rspec/active_record_shared_connection.rb

@@ -5,14 +5,14 @@
   #
   # Tip from http://blog.plataformatec.com.br/2011/12/three-tips-to-improve-the-performance-of-your-test-suite/
   class ::ActiveRecord::Base
-    @@shared_connection_semaphore = Mutex.new
-    @@shared_connection = nil
+    @shared_connection_semaphore = Mutex.new
 
-    def self.connection
-      @@shared_connection_semaphore.synchronize do
-        @@shared_connection ||= retrieve_connection
+    class << self
+      def connection
+        @shared_connection_semaphore.synchronize do
+          @shared_connection ||= retrieve_connection
+        end
       end
     end
   end
-
 end

spec/spec_helper.rb

@@ -0,0 +1,9 @@
+RSpec.configure do |c|
+  c.color = true
+end
+
+# Make sure constants are defined
+module ActiveRecord
+  class Base
+  end
+end

spec/watir/rspec/active_record_shared_connection_spec.rb

@@ -0,0 +1,25 @@
+require "spec_helper"
+require "watir/rspec/active_record_shared_connection"
+
+describe ::ActiveRecord::Base do
+  before { described_class.instance_variable_set :@shared_connection, nil }
+
+  context ".connection" do
+    it "reuses the connection" do
+      described_class.should_receive(:retrieve_connection).once.and_return(:established_connection)
+
+      2.times { described_class.connection.should == :established_connection }
+    end
+
+    it "uses mutex" do
+      described_class.should_receive(:retrieve_connection).once { sleep 0.1; :established_connection }
+
+      thr = Thread.new { described_class.connection }
+
+      t = Time.now
+      described_class.connection
+      (Time.now - t).should be >= 0.1
+      thr.join
+    end
+  end
+end