Merge pull request #343 from andreale/support_for_mongodb_3_wired_tiger

Fix truncation on MongoDB 3
DatabaseCleaner · Jun 28, 2015 · f86e4f0 · f86e4f0
2 parents 19773bb + ecc2951
commit f86e4f0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/lib/database_cleaner/moped/base.rb b/lib/database_cleaner/moped/base.rb
@@ -25,6 +25,10 @@ def host
         @host ||= '127.0.0.1:27017'
       end
 
+      def db_version
+        @db_version ||= session.command('buildinfo' => 1)['version']
+      end
+
       private
 
       def session

diff --git a/lib/database_cleaner/moped/truncation_base.rb b/lib/database_cleaner/moped/truncation_base.rb
@@ -23,9 +23,15 @@ def collections
           session.use(db)
         end
 
-        session['system.namespaces'].find(:name => { '$not' => /\.system\.|\$/ }).to_a.map do |collection|
-          _, name = collection['name'].split('.', 2)
-          name
+        if db_version.split('.').first.to_i >= 3
+          session.command(listCollections: 1, filter: { 'name' => { '$not' => /.?system\.|\$/ } })['cursor']['firstBatch'].map do |collection|
+            collection['name']
+          end
+        else
+          session['system.namespaces'].find(name: { '$not' => /\.system\.|\$/ }).to_a.map do |collection|
+            _, name = collection['name'].split('.', 2)
+            name
+          end
         end
       end