lib/mongo_doc/connection.rb in mongo_doc-0.6.22 vs lib/mongo_doc/connection.rb in mongo_doc-0.6.23

@@ -4,11 +4,11 @@
 
   module Connection
 
     extend self
 
-    attr_writer :config_path, :default_name, :env, :host, :name, :options, :port, :strict
+    attr_writer :config_path, :connection, :default_name, :env, :host, :name, :options, :port, :strict
 
     def config_path
       @config_path || './mongodb.yml'
     end
 
@@ -53,16 +53,15 @@
     end
 
     private
 
     def connect
-      connection = Mongo::Connection.new(host, port, options)
-      raise NoConnectionError unless connection
-      verify_server_version(connection)
+      self.connection = Mongo::Connection.new(host, port, options) || (raise NoConnectionError.new)
+      verify_server_version
       connection
     end
 
-    def verify_server_version(connection)
+    def verify_server_version
       raise UnsupportedServerVersionError.new('MongoDoc requires at least mongoDB version 1.4.0') unless connection.server_version >= "1.4.0"
     end
   end
 end