From d2988420fc6dd91ca751d96ed648fd1ed52ce342 Mon Sep 17 00:00:00 2001
From: Bryan Helmkamp <bryan@brynary.com>
Date: Sun, 17 May 2009 17:53:40 -0400
Subject: Added PostgreSQL to build

---
 spec/spec_helper.rb | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'spec/spec_helper.rb')

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index 6a9a2ef23c..beb634fbd3 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -13,15 +13,27 @@ end
 
 module AdapterGuards
   def adapter_is(name)
+    verify_adapter_name(name)
     yield if name.to_s == adapter_name
   end
 
   def adapter_is_not(name)
+    verify_adapter_name(name)
     yield if name.to_s != adapter_name
   end
 
   def adapter_name
-    Arel::Table.engine.connection.class.name.underscore.split("/").last.gsub(/_adapter/, '')
+    name = ActiveRecord::Base.configurations["unit"][:adapter]
+    verify_adapter_name(name)
+    name
+  end
+
+  def verify_adapter_name(name)
+    raise "Invalid adapter name: #{name}" unless valid_adapters.include?(name.to_s)
+  end
+
+  def valid_adapters
+    %w[mysql postgresql sqlite3]
   end
 end
 
-- 
cgit v1.2.3