Use an in memory db while testing

cristianrasch · May 31, 2012 · 7ea4714 · 7ea4714
1 parent a914e05
commit 7ea4714
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/db/dao.rb b/db/dao.rb
@@ -10,12 +10,17 @@ def db
   private
 
     def connect
-      dir = File.dirname(__FILE__)
       env = ENV["RACK_ENV"] || "development"
-      db_file = File.join(dir, "#{env}.db")
-      @db = Sequel.sqlite(db_file)
-      @db.loggers << Logger.new($stdout) if env == "development"
-      @db
+      if env == "test"
+        @db = Sequel.sqlite
+      else
+        dir = File.dirname(__FILE__)
+        env = ENV["RACK_ENV"] || "development"
+        db_file = File.join(dir, "#{env}.db")
+        @db = Sequel.sqlite(db_file)
+        @db.loggers << Logger.new($stdout) if env == "development"
+        @db
+      end
     end
   end
 end