mongo_test_server 0.0.3 → 0.0.4

data/lib/mongo_test_server/engine.rb

@@ -0,0 +1,4 @@
+module MongoTestServer #:nodoc:
+  class Engine < ::Rails::Engine #:nodoc:
+  end
+end

data/lib/mongo_test_server/railtie.rb

@@ -0,0 +1,21 @@
+require 'rails'
+
+module MongoTestServer
+  class MongoTestServerRailtie < ::Rails::Railtie #:nodoc:
+    config.before_initialize do
+      if Rails.env.test?
+        $stderr.puts "MongoTestServer: starting..."
+        app_name = ENV['APP_NAME'] || Rails.application.class.to_s.split("::").first.downcase
+        config_file = File.expand_path(File.join(Rails.root, 'config', 'mongoid.yml'))
+        mongoid_options = YAML.load(ERB.new(File.read(config_file)).result)[ENV['RAILS_ENV']]
+        port = mongoid_options['sessions']['default']['hosts'].first.split(':').last.to_i
+        MongoTestServer::Mongod.configure do |server|
+          server.name = app_name
+          server.port = port
+          server.path = `which mongod`.chomp
+        end
+        MongoTestServer::Mongod.server.start
+      end
+    end
+  end
+end

data/lib/mongo_test_server/version.rb

@@ -1,3 +1,3 @@
 module MongoTestServer
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/lib/mongo_test_server.rb

@@ -10,6 +10,11 @@ require 'yaml'
 require 'tempfile'
 require "mongo_test_server/version"
 
+if defined?(Rails)
+  require 'mongo_test_server/railtie'
+  require 'mongo_test_server/engine'
+end
+
 module MongoTestServer
 
   class Mongod
@@ -47,6 +52,7 @@ module MongoTestServer
     attr_writer :path
     attr_writer :name
     attr_reader :mongo_instance_id
+    attr_reader :use_ram_disk
 
     def initialize(port=nil, name=nil, path=nil)
       self.port = port
@@ -59,6 +65,38 @@ module MongoTestServer
       self.started = false
     end
 
+    def use_ram_disk=(bool)
+      if bool && (`which hdiutil`!='')
+        @use_ram_disk = true
+      else
+        $stderr.puts "MongoTestServer: can't use a ram disk on this system"
+        @use_ram_disk = false
+      end
+    end
+
+    def use_ram_disk?
+      @use_ram_disk
+    end
+
+    def ram_disk_name
+      @ram_disk_name ||= "mongodb-ram-#{self.name}"
+    end
+
+    def setup_ram_disk
+      @ram_disk_device = `hdiutil attach -nomount ram://1000000`.chomp
+      `diskutil erasevolume HFS+ #{ram_disk_name} #{@ram_disk_device}`
+      ram_disk_mount
+    end
+
+    def ram_disk_mount
+      "/Volumes/#{ram_disk_name}"
+    end
+
+    def teardown_ram_disk
+      `umount #{ram_disk_mount} 2> /dev/null`
+      `hdiutil detach #{@ram_disk_device} 2> /dev/null`
+    end
+
     def mongo_log
       "#{mongo_dir}/mongo_log"
     end
@@ -76,7 +114,22 @@ module MongoTestServer
     end
 
     def mongo_dir
-      @mongo_dir ||= "/tmp/#{self.name}_mongo_testserver_#{@mongo_instance_id}"
+      @mongo_dir ||= lambda {
+        if self.use_ram_disk?
+          $stderr.puts "MongoTestServer: using ramdisk"
+          setup_ram_disk
+        else
+          "/tmp/#{self.name}_mongo_testserver_#{@mongo_instance_id}"
+        end
+        }.call
+    end
+
+    def remove_mongo_dir
+      if self.use_ram_disk?
+        teardown_ram_disk
+      else
+        FileUtils.rm_rf self.mongo_dir
+      end
     end
 
     def mongo_cmd_line
@@ -84,7 +137,7 @@ module MongoTestServer
     end
 
     def prepare
-      FileUtils.rm_rf self.mongo_dir
+      remove_mongo_dir
       FileUtils.mkdir_p self.mongo_dir
     end
 
@@ -205,7 +258,7 @@ module MongoTestServer
       self.killed = true
       self.started = false
       mongo_pids.each { |ppid| `kill -9 #{ppid} 2> /dev/null` }
-      FileUtils.rm_rf self.mongo_dir
+      remove_mongo_dir
       self
     end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongo_test_server
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-11-27 00:00:00.000000000 Z
+date: 2012-11-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mongo
@@ -89,6 +89,8 @@ files:
 - README.md
 - Rakefile
 - lib/mongo_test_server.rb
+- lib/mongo_test_server/engine.rb
+- lib/mongo_test_server/railtie.rb
 - lib/mongo_test_server/version.rb
 - mongo_test_server.gemspec
 - spec/mongo_test_server/mongo_test_server_spec.rb