database_plumber 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 56ed59f538667d495b0d65ff3f41123960c24ad3
-  data.tar.gz: 176d79d1c384c06e07c1545e80c2d48c328fb55a
+  metadata.gz: daf1a5f4a3baccca239f4278220bb38c6bf23e03
+  data.tar.gz: 411fe8bed18fb3edf20ea16a7be0d20e8ef71cac
 SHA512:
-  metadata.gz: ccdc5f2947c9ae74775e53f096d8c1576331101e54e5f62dfd90f2b4e7798b9a1374ac585da556dac6eb5f0764af911ad6deffe9bb196cce44b55e0e718123d9
-  data.tar.gz: fb8dab7896d3c5da4fc595fdc6d0736cba4f48e871b82f6c0eeb488f0a078803b7bf94bd519608da69c90acf1a34b7690a3097cf4854104cd7795ac711472a34
+  metadata.gz: 78f57584b621b51e8f8d52e2c554c1a5d1216f7a2f8e84ed57bced09bd6c770144db8ac546851c5e0c832f8a788c51450473a82a2e7b77cf6afd02b46067125d
+  data.tar.gz: ae1e8ffec52530b5bc86534cb6fce7dbd98a283bd9b3f745639fbc80bcaa902b21e56205970a384874ae0f4ad66c1f010075ecb8da204e42c122757bd8f0bf21

data/.gitignore

@@ -12,3 +12,5 @@
 *.o
 *.a
 mkmf.log
+
+*.gem

data/README.md

@@ -133,8 +133,22 @@ config.after(:all) do
 end
 ```
 
-This is also useful if your test suite is problem-free and you would like ensure it
-stays that way.
+### Setting thresholds for Models
+
+You may have some models you would like to report on, but which should also have
+entries in the database, for example a table that is seeded or loaded with fixtures.
+In order to allow this you can provide a threshold for a Model, which is the
+maximum number of entries allowed in the database for the Model before it is
+regarded as leaky.
+
+To provide a threshold for a Model, you can can the following:
+
+```ruby
+config.after(:all) do
+  # Perform the report after each example group
+  DatabasePlumber.inspect model_thresholds: { Bar => 3 }
+end
+
 
 ## Contributing
 

data/lib/database_plumber/leak_finder.rb

@@ -12,12 +12,13 @@ module DatabasePlumber
     def initialize(options)
       @ignored_models = (options[:ignored_models] || []) + IGNORED_AR_INTERNALS
       @ignored_adapters = options[:ignored_adapters] || []
+      @model_thresholds = options[:model_thresholds] || {}
     end
 
     def inspect
       filtered_models.each_with_object({}) do |model, results|
         records = count_for(model)
-        if records > 0
+        if records > threshold_for(model)
           results[model.to_s] = records
           mop_up(model)
         end
@@ -34,6 +35,10 @@ module DatabasePlumber
       raise InvalidModelError, "#{model} does not have a valid table definition"
     end
 
+    def threshold_for(model)
+      @model_thresholds.key?(model) ? @model_thresholds[model].to_i : 0
+    end
+
     def mop_up(model)
       model.destroy_all
     end

data/lib/database_plumber/version.rb

@@ -1,3 +1,3 @@
 module DatabasePlumber
-  VERSION = '1.0.0'
+  VERSION = '1.1.0'
 end

data/spec/lib/database_plumber/leak_finder_spec.rb

@@ -3,7 +3,7 @@ RSpec.describe DatabasePlumber::LeakFinder do
   let(:ignored_connection)  { double(:connection, adapter_name: 'SQLite') }
 
   let(:happy_model)           { double(:happy,    name: 'Happy',        abstract_class?: nil,  connection: normal_connection,  count: 0) }
-  let(:leaky_model)           { double(:leaky,    name: 'Leaky',        abstract_class?: nil,  connection: normal_connection,  count: 1) }
+  let(:leaky_model)           { double(:leaky,    name: 'Leaky',        abstract_class?: nil,  connection: normal_connection,  count: 2) }
   let(:abstract_model)        { double(:abstract, name: 'Abstract',     abstract_class?: true, connection: normal_connection,  count: 2) }
   let(:ignored_model)         { double(:ignored,  name: 'Ignored',      abstract_class?: nil,  connection: normal_connection,  count: 3) }
   let(:ignored_adapter_model) { double(:anon,     name: 'Anon',         abstract_class?: nil,  connection: ignored_connection, count: 4) }
@@ -24,7 +24,7 @@ RSpec.describe DatabasePlumber::LeakFinder do
     context 'with no params' do
       let(:expected_leaks) do
         {
-          leaky_model.to_s => 1,
+          leaky_model.to_s => 2,
           ignored_model.to_s => 3,
           ignored_adapter_model.to_s => 4
         }
@@ -54,7 +54,7 @@ RSpec.describe DatabasePlumber::LeakFinder do
 
       let(:expected_leaks) do
         {
-          leaky_model.to_s => 1,
+          leaky_model.to_s => 2,
           ignored_adapter_model.to_s => 4
         }
       end
@@ -83,7 +83,7 @@ RSpec.describe DatabasePlumber::LeakFinder do
 
       let(:expected_leaks) do
         {
-          leaky_model.to_s => 1,
+          leaky_model.to_s => 2,
           ignored_model.to_s => 3
         }
       end
@@ -103,6 +103,86 @@ RSpec.describe DatabasePlumber::LeakFinder do
       it { expect(join_model_stub).not_to have_received(:destroy_all) }
     end
 
+    context 'with a threshold' do
+      context 'with a leaky model at the threshold' do
+        let(:options_params) do
+          {
+            ignored_models: [ignored_model],
+            ignored_adapters: [:sqlite],
+            model_thresholds: { leaky_model => 2 }
+          }
+        end
+
+        before(:each) { @leaks = described_class.inspect(options_params) }
+
+        it { expect(@leaks).to be_empty }
+
+        it { expect(ActiveRecord::SchemaMigration).not_to have_received(:destroy_all) }
+
+        it { expect(happy_model).not_to have_received(:destroy_all) }
+        it { expect(leaky_model).not_to have_received(:destroy_all) }
+
+        it { expect(ignored_model).not_to have_received(:destroy_all) }
+        it { expect(ignored_adapter_model).not_to have_received(:destroy_all) }
+
+        it { expect(join_model_stub).not_to have_received(:destroy_all) }
+      end
+
+      context 'with a leaky model below the threshold' do
+        let(:options_params) do
+          {
+            ignored_models: [ignored_model],
+            ignored_adapters: [:sqlite],
+            model_thresholds: { leaky_model => 5 }
+          }
+        end
+
+        before(:each) { @leaks = described_class.inspect(options_params) }
+
+        it { expect(@leaks).to be_empty }
+
+        it { expect(ActiveRecord::SchemaMigration).not_to have_received(:destroy_all) }
+
+        it { expect(happy_model).not_to have_received(:destroy_all) }
+        it { expect(leaky_model).not_to have_received(:destroy_all) }
+
+        it { expect(ignored_model).not_to have_received(:destroy_all) }
+        it { expect(ignored_adapter_model).not_to have_received(:destroy_all) }
+
+        it { expect(join_model_stub).not_to have_received(:destroy_all) }
+      end
+
+      context 'with a leaky model above the threshold' do
+        let(:options_params) do
+          {
+            ignored_models: [ignored_model],
+            ignored_adapters: [:sqlite],
+            model_thresholds: { leaky_model => 1 }
+          }
+        end
+
+        let(:expected_leaks) do
+          {
+            leaky_model.to_s => 2
+          }
+        end
+
+        before(:each) { @leaks = described_class.inspect(options_params) }
+
+        it { expect(@leaks).to eql(expected_leaks) }
+
+        it { expect(ActiveRecord::SchemaMigration).not_to have_received(:destroy_all) }
+
+        it { expect(happy_model).not_to have_received(:destroy_all) }
+        it { expect(leaky_model).to have_received(:destroy_all) }
+
+        it { expect(ignored_model).not_to have_received(:destroy_all) }
+        it { expect(ignored_adapter_model).not_to have_received(:destroy_all) }
+
+        it { expect(join_model_stub).not_to have_received(:destroy_all) }
+      end
+    end
+
     context 'with no leaking models in scope' do
       let(:options_params) do
         {

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: database_plumber
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Barry Gordon
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-03-04 00:00:00.000000000 Z
+date: 2016-04-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler