database_flusher 0.2.3 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 75d8472e33710dc5a47fa06ae32e618e929d6b20
-  data.tar.gz: 5eaf9adc3245e5f0ab400a592db7d884382cc66a
+  metadata.gz: a5757e018763a6a24a1ab1d4458342652e37c6dd
+  data.tar.gz: eef0f72db7a2688c90cb43829dce03f5cbc04484
 SHA512:
-  metadata.gz: d95f2e35fa0107d9011119bf590cb655f06850f1436a3aafae43c3101e8c5c356eace416d2138b5fc9d2287d51a469b0b4126ff0d35b0f7593fb48c312f677e9
-  data.tar.gz: 23e7639d7fb80545489f65820a48f7d48465060699a8959a8a2cd567000192f38ba68460a8372d576f50eb2a42aaeebe2e8aeb2ba8c3a387add426f65fd1ba05
+  metadata.gz: 8202dd0665c71467cd1d9ad20740453e94e7b674f432c6dd4c6238a2ae02cb1f061a017ea84faea86e0c73d57268ddd179edfd1f9aef1adeaf63a76a6729dc51
+  data.tar.gz: 19f3166b15a7d8933d478a00b01fe875acf0d15fd00fd4fccf06317ec956c518e88416936b5a88a75b822e3ac8e3e713aaa0d5b57fc753b135a36e6b630a609c

data/README.md

@@ -96,6 +96,26 @@ After do
 end
 ```
 
+In case of unclean test shutdown, you can clean the whole database, using:
+
+```ruby
+RSpec.configure do |config|
+  config.before :suite do
+    if File.exists?('.~lock')
+      puts "Unclean shutdown, cleaning the whole database..."
+      DatabaseFlusher[:active_record].clean_with(:deletion)
+      DatabaseFlusher[:mongoid].clean_with(:deletion)
+    else
+      File.open('.~lock', 'a') {}
+    end
+  end
+
+  config.after :suite do
+    File.unlink('.~lock')
+  end
+end
+```
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/ebeigarts/database_flusher. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.

data/lib/database_flusher/active_record/deletion_strategy.rb

@@ -7,7 +7,7 @@ require 'database_flusher/active_record/adapters/sqlite_adapter'
 module DatabaseFlusher
   module ActiveRecord
     class DeletionStrategy
-      attr_reader :tables, :adapter
+      attr_reader :tables
 
       class Subscriber
         def initialize(strategy)
@@ -34,10 +34,6 @@ module DatabaseFlusher
           'sql.active_record',
           Subscriber.new(self)
         )
-        connection = ::ActiveRecord::Base.connection
-        @adapter = DatabaseFlusher::ActiveRecord.
-          const_get("#{connection.adapter_name}Adapter").
-          new(connection)
       end
 
       def stop
@@ -55,6 +51,31 @@ module DatabaseFlusher
 
         tables.clear
       end
+
+      def clean_all
+        adapter.delete(*all_tables)
+      end
+
+      private
+
+      def connection
+        @connection ||= ::ActiveRecord::Base.connection
+      end
+
+      def adapter
+        @adapter ||= DatabaseFlusher::ActiveRecord.
+          const_get("#{connection.adapter_name}Adapter").
+          new(connection)
+      end
+
+      def all_tables
+        tables = connection.tables
+        tables.reject do |t|
+          (t == ::ActiveRecord::Migrator.schema_migrations_table_name) ||
+          (::ActiveRecord::Base.respond_to?(:internal_metadata_table_name) &&
+            (t == ::ActiveRecord::Base.internal_metadata_table_name))
+        end
+      end
     end
   end
 end

data/lib/database_flusher/cleaner.rb

@@ -22,7 +22,7 @@ module DatabaseFlusher
 
     def clean_with(name)
       self.strategy = name
-      strategy.clean
+      strategy.clean_all
     end
 
     def start

data/lib/database_flusher/mongoid/deletion_strategy.rb

@@ -10,11 +10,9 @@ module DatabaseFlusher
         end
 
         def started(event)
-          if event.command_name == :insert || event.command_name == 'insert'.freeze
-            collection = event.command['insert'.freeze]
-            if collection
-              @strategy.collections << collection
-            end
+          collection = event.command['insert'.freeze]
+          if collection
+            @strategy.collections << collection
           end
         end
 
@@ -29,7 +27,7 @@ module DatabaseFlusher
       end
 
       def start
-        @subscriber ||= Mongo::Monitoring::Global.subscribe(
+        @subscriber ||= client.subscribe(
           Mongo::Monitoring::COMMAND,
           Subscriber.new(self)
         )
@@ -42,12 +40,27 @@ module DatabaseFlusher
       def clean
         return if collections.empty?
         # puts "Cleaning #{collections.inspect}"
-        client = ::Mongoid::Clients.default
         collections.each do |name|
           client[name].delete_many
         end
         collections.clear
       end
+
+      def clean_all
+        all_collections.each do |name|
+          client[name].delete_many
+        end
+      end
+
+      private
+
+      def client
+        @client ||= ::Mongoid::Clients.default
+      end
+
+      def all_collections
+        client.database.collections.collect { |c| c.namespace.split('.',2)[1] }
+      end
     end
   end
 end

data/lib/database_flusher/version.rb

@@ -1,4 +1,4 @@
 # frozen_string_literal: true
 module DatabaseFlusher
-  VERSION = '0.2.3'.freeze
+  VERSION = '0.3.0'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: database_flusher
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.3.0
 platform: ruby
 authors:
 - Edgars Beigarts
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-12-12 00:00:00.000000000 Z
+date: 2016-12-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler