n_plus_one_control 0.3.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fa826fde3f47f08d3054ca3cf80a4c4d2a4ef2f70aa8c6f5d56a94dadb8ef656
-  data.tar.gz: d8eb72c89592e99d3864b89fcdaa857fae1566ee4e2e5733fea50faa2463ba5c
+  metadata.gz: 227fb8257ab67628ffd449aff0358d729dd3ff3eafc291f9fca427aa355e0650
+  data.tar.gz: 2e65e5f09da3ad989fbb1e2418a657c8142ea8d25540a0daf729adeb9f0b0d5b
 SHA512:
-  metadata.gz: 87a27c36da8bfb778027b3cbd2cad307b0ed1426b4b8273989fbe8171aace9efe7e90d8919d5d4e275d48d081c1bd0924cafbbfe87d750b28e04eb4cf3a57c4e
-  data.tar.gz: dd8c8be25a92742551d955d258b26a8abcb2fba15f1040afa2b39f4fca096bdf17eff8b3f54e2975dae4dc63952c79197932d819a61edd7faed8d4a5f11cbf1d
+  metadata.gz: e79cf5bd9ec1afd8b10ca184a483edd9e8d8fbbeaca62e147d277102715ead2fa909d720eb0085fc6d2c2441abb5e61aa08f8a081a4b37079b3f552694d3e220
+  data.tar.gz: 59c76b177dc8b6ad4d58cfbad1f3c3e89204eef6e92d2d39989b5829f714939c77d0d38806e935d3624e20469447607f7b403af2b2bd633a02486e82c6cc4f6c

data/CHANGELOG.md

@@ -0,0 +1,8 @@
+## 0.4.0 (2020-07-20)
+
+- Make scale factor available in tests via `#current_scale` method. ([@Earendil95][])
+
+- Start keeping a changelog. ([@palkan][])
+
+[@Earendil95]: https://github.com/Earendil95
+[@palkan]: https://github.com/palkan

data/LICENSE.txt

@@ -1,6 +1,6 @@
 The MIT License (MIT)
 
-Copyright (c) 2017 palkan
+Copyright (c) 2017-2020 Vladimir Dementyev
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

data/README.md

@@ -92,6 +92,26 @@ expect { ... }.to perform_constant_number_of_queries.matching(/INSERT/)
 expect { ... }.to perform_constant_number_of_queries.with_scale_factors(10, 100)
 ```
 
+#### Using scale factor in spec
+
+Let's suppose your action accepts parameter, which can make impact on the number of returned records:
+
+```ruby
+get :index, params: { per_page: 10 }
+```
+
+Then it is enough to just change `per_page` parameter between executions and do not recreate records in DB. For this purpose, you can use `current_scale` method in your example:
+
+```ruby
+context "N+1", :n_plus_one do
+  before { create_list :post, 3 }
+
+  specify do
+    expect { get :index, params: { per_page: current_scale } }.to perform_constant_number_of_queries
+  end
+end
+```
+
 ### Minitest
 
 First, add NPlusOneControl to your `test_helper.rb`:
@@ -147,6 +167,16 @@ def test_no_n_plus_one_error
 end
 ```
 
+As in RSpec, you can use `current_scale` factor instead of `populate` block:
+
+```ruby
+def test_no_n_plus_one_error
+  assert_perform_constant_number_of_queries do
+    get :index, params: { per_page: current_scale }
+  end
+end
+```
+
 ### With caching
 
 If you use caching you can face the problem when first request performs more DB queries than others. The solution is:

data/lib/n_plus_one_control/executor.rb

@@ -3,7 +3,7 @@
 module NPlusOneControl
   # Runs code for every scale factor
   # and returns collected queries.
-  module Executor
+  class Executor
     # Subscribes to ActiveSupport notifications and collect matching queries.
     class Collector
       def initialize(pattern)
@@ -29,37 +29,59 @@ module NPlusOneControl
     class << self
       attr_accessor :transaction_begin
       attr_accessor :transaction_rollback
+    end
+
+    attr_reader :current_scale
+
+    self.transaction_begin = -> do
+      ActiveRecord::Base.connection.begin_transaction(joinable: false)
+    end
+
+    self.transaction_rollback = -> do
+      ActiveRecord::Base.connection.rollback_transaction
+    end
 
-      def call(population:, scale_factors: nil, matching: nil)
-        raise ArgumentError, "Block is required!" unless block_given?
+    def initialize(population: nil, scale_factors: nil, matching: nil)
+      @population = population
+      @scale_factors = scale_factors
+      @matching = matching
+    end
+
+    # rubocop:disable Metrics/MethodLength
+    def call
+      raise ArgumentError, "Block is required!" unless block_given?
 
-        results = []
-        collector = Collector.new(matching)
+      results = []
+      collector = Collector.new(matching)
 
-        (scale_factors || NPlusOneControl.default_scale_factors).each do |scale|
-          with_transaction do
-            population.call(scale)
-            results << [scale, collector.call { yield }]
-          end
+      (scale_factors || NPlusOneControl.default_scale_factors).each do |scale|
+        @current_scale = scale
+        with_transaction do
+          population&.call(scale)
+          results << [scale, collector.call { yield }]
         end
-        results
       end
+      results
+    end
+    # rubocop:enable Metrics/MethodLength
 
-      private
+    private
 
-      def with_transaction
-        transaction_begin.call
-        yield
-      ensure
-        transaction_rollback.call
-      end
+    def with_transaction
+      transaction_begin.call
+      yield
+    ensure
+      transaction_rollback.call
     end
 
-    self.transaction_begin = -> do
-      ActiveRecord::Base.connection.begin_transaction(joinable: false)
+    def transaction_begin
+      self.class.transaction_begin
     end
-    self.transaction_rollback = -> do
-      ActiveRecord::Base.connection.rollback_transaction
+
+    def transaction_rollback
+      self.class.transaction_rollback
     end
+
+    attr_reader :population, :scale_factors, :matching
   end
 end

data/lib/n_plus_one_control/minitest.rb

@@ -5,10 +5,6 @@ require "n_plus_one_control"
 module NPlusOneControl
   # Minitest assertions
   module MinitestHelper
-    def warming_up(warmup)
-      (warmup || methods.include?(:warmup) ? method(:warmup) : nil)&.call
-    end
-
     def assert_perform_constant_number_of_queries(
       populate: nil,
       matching: nil,
@@ -20,16 +16,32 @@ module NPlusOneControl
 
       warming_up warmup
 
-      queries = NPlusOneControl::Executor.call(
-        population: populate || method(:populate),
+      @executor = NPlusOneControl::Executor.new(
+        population: population_proc(populate),
         matching: matching || /^SELECT/i,
         scale_factors: scale_factors || NPlusOneControl.default_scale_factors
-      ) { yield }
+      )
+
+      queries = @executor.call { yield }
 
       counts = queries.map(&:last).map(&:size)
 
       assert counts.max == counts.min, NPlusOneControl.failure_message(queries)
     end
+
+    def current_scale
+      @executor&.current_scale
+    end
+
+    private
+
+    def warming_up(warmup)
+      (warmup || methods.include?(:warmup) ? method(:warmup) : nil)&.call
+    end
+
+    def population_proc(populate)
+      (populate || methods.include?(:populate) ? method(:populate) : nil)
+    end
   end
 end
 

data/lib/n_plus_one_control/rspec.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+gem "rspec-core", ">= 3.5"
+
 require "n_plus_one_control"
 require "n_plus_one_control/rspec/dsl"
 require "n_plus_one_control/rspec/matcher"
@@ -11,5 +13,6 @@ module NPlusOneControl
 end
 
 ::RSpec.configure do |config|
-  config.extend NPlusOneControl::RSpec::DSL, n_plus_one: true
+  config.extend NPlusOneControl::RSpec::DSL::ClassMethods, n_plus_one: true
+  config.include NPlusOneControl::RSpec::DSL, n_plus_one: true
 end

data/lib/n_plus_one_control/rspec/context.rb

@@ -3,17 +3,8 @@
 RSpec.shared_context "n_plus_one_control" do
   # Helper to access populate block from within example/matcher
   let(:n_plus_one_populate) do |ex|
-    if ex.example_group.populate.nil?
-      raise(
-        <<-MSG
-          Populate block is missing!
+    return if ex.example_group.populate.nil?
 
-          Please provide populate callback, e.g.:
-
-            populate { |n| n.times { create_some_stuff } }
-        MSG
-      )
-    end
     ->(n) { ex.instance_exec(n, &ex.example_group.populate) }
   end
 

data/lib/n_plus_one_control/rspec/dsl.rb

@@ -2,23 +2,32 @@
 
 module NPlusOneControl
   module RSpec
-    # Extends RSpec ExampleGroup with populate & warmup methods
+    # Includes scale method into RSpec Example
     module DSL
-      # Setup warmup block, wich will run before matching
-      # for example, if using cache, then later queries
-      # will perform less DB queries than first
-      def warmup
-        return @warmup unless block_given?
+      # Extends RSpec ExampleGroup with populate & warmup methods
+      module ClassMethods
+        # Setup warmup block, wich will run before matching
+        # for example, if using cache, then later queries
+        # will perform less DB queries than first
+        def warmup
+          return @warmup unless block_given?
 
-        @warmup = Proc.new
+          @warmup = Proc.new
+        end
+
+        # Setup populate callback, which is used
+        # to prepare data for each run.
+        def populate
+          return @populate unless block_given?
+
+          @populate = Proc.new
+        end
       end
 
-      # Setup populate callback, which is used
-      # to prepare data for each run.
-      def populate
-        return @populate unless block_given?
+      attr_accessor :executor
 
-        @populate = Proc.new
+      def current_scale
+        executor&.current_scale
       end
     end
   end

data/lib/n_plus_one_control/rspec/matcher.rb

@@ -30,13 +30,14 @@
     # by default we're looking for select queries
     pattern = @pattern || /^SELECT/i
 
-    @queries = NPlusOneControl::Executor.call(
+    @matcher_execution_context.executor = NPlusOneControl::Executor.new(
       population: populate,
       matching: pattern,
-      scale_factors: @factors,
-      &actual
+      scale_factors: @factors
     )
 
+    @queries = @matcher_execution_context.executor.call(&actual)
+
     counts = @queries.map(&:last).map(&:size)
 
     counts.max == counts.min

data/lib/n_plus_one_control/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module NPlusOneControl
-  VERSION = "0.3.1"
+  VERSION = "0.4.0"
 end

metadata

@@ -1,27 +1,27 @@
 --- !ruby/object:Gem::Specification
 name: n_plus_one_control
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0
 platform: ruby
 authors:
 - palkan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-02-18 00:00:00.000000000 Z
+date: 2020-07-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '1.10'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '1.10'
 - !ruby/object:Gem::Dependency
@@ -146,16 +146,9 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".gitignore"
-- ".rspec"
-- ".rubocop.yml"
-- ".travis.yml"
-- Gemfile
+- CHANGELOG.md
 - LICENSE.txt
 - README.md
-- Rakefile
-- bin/console
-- bin/setup
 - lib/n_plus_one_control.rb
 - lib/n_plus_one_control/executor.rb
 - lib/n_plus_one_control/minitest.rb
@@ -164,19 +157,15 @@ files:
 - lib/n_plus_one_control/rspec/dsl.rb
 - lib/n_plus_one_control/rspec/matcher.rb
 - lib/n_plus_one_control/version.rb
-- n_plus_one_control.gemspec
-- spec/n_plus_one_control/executor_spec.rb
-- spec/n_plus_one_control/rspec_spec.rb
-- spec/n_plus_one_control_spec.rb
-- spec/spec_helper.rb
-- spec/support/post.rb
-- spec/support/user.rb
-- tests/minitest_test.rb
-- tests/test_helper.rb
 homepage: http://github.com/palkan/n_plus_one_control
 licenses:
 - MIT
-metadata: {}
+metadata:
+  bug_tracker_uri: http://github.com/palkan/n_plus_one_control/issues
+  changelog_uri: https://github.com/palkan/n_plus_one_control/blob/master/CHANGELOG.md
+  documentation_uri: http://github.com/palkan/n_plus_one_control
+  homepage_uri: http://github.com/palkan/n_plus_one_control
+  source_code_uri: http://github.com/palkan/n_plus_one_control
 post_install_message: 
 rdoc_options: []
 require_paths:
@@ -192,8 +181,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.6
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: RSpec and Minitest matchers to prevent N+1 queries problem

data/.gitignore

@@ -1,11 +0,0 @@
-/.bundle/
-/.yardoc
-/Gemfile.lock
-/_yardoc/
-/coverage/
-/doc/
-/pkg/
-/spec/reports/
-/tmp/
-/log/
-*.gem

data/.rspec

@@ -1,2 +0,0 @@
---format documentation
---color

data/.rubocop.yml

@@ -1,78 +0,0 @@
-AllCops:
-  Include:
-    - 'lib/**/*.rb'
-    - 'lib/**/*.rake'
-    - 'spec/**/*.rb'
-  Exclude:
-    - 'bin/**/*'
-    - 'spec/dummy/**/*'
-    - 'tmp/**/*'
-    - 'vendor/**/*'
-    - 'Rakefile'
-    - 'Gemfile'
-    - '*.gemspec'
-  DisplayCopNames: true
-  StyleGuideCopsOnly: false
-  TargetRubyVersion: 2.4
-
-Rails:
-  Enabled: false
-
-Naming/UncommunicativeMethodParamName:
-  Enabled: false
-
-Naming/AccessorMethodName:
-  Enabled: false
-
-Style/TrivialAccessors:
-  Enabled: false
-
-Style/Documentation:
-  Exclude:
-    - 'spec/**/*.rb'
-    - 'tests/**/*.rb'
-
-Style/StringLiterals: 
-  Enabled: false
-
-Style/RegexpLiteral:
-  Enabled: false
-
-Style/Lambda:
-  Enabled: false
-
-Layout/SpaceInsideStringInterpolation:
-  EnforcedStyle: no_space
-
-Style/ClassAndModuleChildren:
-  Enabled: false
-
-Style/BlockDelimiters:
-  Exclude:
-    - 'spec/**/*.rb'
-
-Lint/AmbiguousRegexpLiteral:
-  Enabled: false
-
-
-Metrics/MethodLength:
-  Exclude:
-    - 'spec/**/*.rb'
-
-Metrics/AbcSize:
-  Max: 20
-
-Metrics/LineLength:
-  Max: 100
-  Exclude:
-    - 'spec/**/*.rb'
-
-Metrics/BlockLength:
-  Exclude:
-    - 'spec/**/*.rb'
-
-Rails/Date:
-  Enabled: false
-
-Rails/TimeZone:
-  Enabled: false

data/.travis.yml

@@ -1,9 +0,0 @@
-sudo: false
-language: ruby
-cache: bundler
-rvm:
-  - 2.5.3
-
-before_install:
-  - gem uninstall -v '>= 2' -i $(rvm gemdir)@global -ax bundler || true
-  - gem install bundler -v '< 2'

data/Gemfile

@@ -1,4 +0,0 @@
-source 'https://rubygems.org'
-
-# Specify your gem's dependencies in n_plus_one_control.gemspec
-gemspec

data/Rakefile

@@ -1,13 +0,0 @@
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-require "rubocop/rake_task"
-require "rake/testtask"
-
-Rake::TestTask.new do |t|
-  t.test_files = FileList['tests/**/*_test.rb']
-end
-
-RuboCop::RakeTask.new
-RSpec::Core::RakeTask.new(:spec)
-
-task :default => [:spec, :test, :rubocop]

data/bin/console

@@ -1,14 +0,0 @@
-#!/usr/bin/env ruby
-
-require "bundler/setup"
-require "n_plus_one_control"
-
-# You can add fixtures and/or initialization code here to make experimenting
-# with your gem easier. You can also use a different console, if you like.
-
-# (If you use this, don't forget to add pry to your Gemfile!)
-# require "pry"
-# Pry.start
-
-require "irb"
-IRB.start

data/bin/setup

@@ -1,8 +0,0 @@
-#!/usr/bin/env bash
-set -euo pipefail
-IFS=$'\n\t'
-set -vx
-
-bundle install
-
-# Do any other automated setup that you need to do here

data/n_plus_one_control.gemspec

@@ -1,35 +0,0 @@
-# coding: utf-8
-lib = File.expand_path('../lib', __FILE__)
-$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
-require 'n_plus_one_control/version'
-
-Gem::Specification.new do |spec|
-  spec.name          = "n_plus_one_control"
-  spec.version       = NPlusOneControl::VERSION
-  spec.authors       = ["palkan"]
-  spec.email         = ["dementiev.vm@gmail.com"]
-
-  spec.summary       = "RSpec and Minitest matchers to prevent N+1 queries problem"
-  spec.required_ruby_version = '>= 2.0.0'
-  spec.description   = %{
-    RSpec and Minitest matchers to prevent N+1 queries problem.
-
-    Evaluates code under consideration several times with different scale factors
-    to make sure that the number of DB queries behaves as expected (i.e. O(1) instead of O(N)).
-  }
-  spec.homepage      = "http://github.com/palkan/n_plus_one_control"
-  spec.license       = "MIT"
-
-  spec.files         = `git ls-files`.split($/)
-  spec.require_paths = ["lib"]
-
-  spec.add_development_dependency "bundler", "~> 1.10"
-  spec.add_development_dependency "rake", "~> 10.0"
-  spec.add_development_dependency "rspec", "~> 3.5"
-  spec.add_development_dependency "minitest", "~> 5.9"
-  spec.add_development_dependency "factory_girl", "~> 4.8.0"
-  spec.add_development_dependency "rubocop", "~> 0.61.0"
-  spec.add_development_dependency "activerecord", "~> 5.1"
-  spec.add_development_dependency "sqlite3", "~> 1.3.6"
-  spec.add_development_dependency "pry-byebug"
-end

data/spec/n_plus_one_control/executor_spec.rb

@@ -1,65 +0,0 @@
-# frozen_string_literal: true
-
-require "spec_helper"
-
-describe NPlusOneControl::Executor do
-  let(:populate) do
-    ->(n) { create_list(:post, n) }
-  end
-
-  let(:observable) do
-    -> { Post.find_each(&:user) }
-  end
-
-  it "raises when block is missing" do
-    expect { described_class.call(population: populate) }
-      .to raise_error(ArgumentError, "Block is required!")
-  end
-
-  it "raises when populate is missing" do
-    expect { described_class.call(&observable) }
-      .to raise_error(ArgumentError, /population/)
-  end
-
-  it "returns correct counts for default scales" do
-    result = described_class.call(
-      population: populate,
-      &observable
-    )
-
-    expect(result.size).to eq 2
-    expect(result.first[0]).to eq 2
-    expect(result.first[1].size).to eq 3
-    expect(result.last[0]).to eq 3
-    expect(result.last[1].size).to eq 4
-  end
-
-  it "returns correct counts for custom scales" do
-    result = described_class.call(
-      population: populate,
-      scale_factors: [5, 10, 100],
-      &observable
-    )
-
-    expect(result.size).to eq 3
-    expect(result.first[0]).to eq 5
-    expect(result.first[1].size).to eq 6
-    expect(result.second[0]).to eq 10
-    expect(result.second[1].size).to eq 11
-    expect(result.last[0]).to eq 100
-    expect(result.last[1].size).to eq 101
-  end
-
-  it "returns correct counts with custom match" do
-    result = described_class.call(
-      population: populate,
-      matching: /users/,
-      &observable
-    )
-
-    expect(result.first[0]).to eq 2
-    expect(result.first[1].size).to eq 2
-    expect(result.last[0]).to eq 3
-    expect(result.last[1].size).to eq 3
-  end
-end

data/spec/n_plus_one_control/rspec_spec.rb

@@ -1,113 +0,0 @@
-# frozen_string_literal: true
-
-require "spec_helper"
-
-describe NPlusOneControl::RSpec do
-  context "when no N+1", :n_plus_one do
-    populate { |n| create_list(:post, n) }
-
-    specify do
-      expect { Post.preload(:user).find_each { |p| p.user.name } }
-        .to perform_constant_number_of_queries
-    end
-  end
-
-  context "when has N+1", :n_plus_one do
-    populate { |n| create_list(:post, n) }
-
-    specify do
-      expect do
-        expect { Post.find_each { |p| p.user.name } }
-          .to perform_constant_number_of_queries
-      end.to raise_error(RSpec::Expectations::ExpectationNotMetError)
-    end
-  end
-
-  context "when context is missing" do
-    specify do
-      expect do
-        expect { subject }.to perform_constant_number_of_queries
-      end.to raise_error(/missing tag/i)
-    end
-  end
-
-  context "when populate is missing", :n_plus_one do
-    specify do
-      expect do
-        expect { subject }.to perform_constant_number_of_queries
-      end.to raise_error(/please provide populate/i)
-    end
-  end
-
-  context "when negated" do
-    specify do
-      expect do
-        expect { subject }.not_to perform_constant_number_of_queries
-      end.to raise_error(/support negation/i)
-    end
-  end
-
-  context "when verbose", :n_plus_one do
-    populate { |n| create_list(:post, n) }
-
-    around(:each) do |ex|
-      NPlusOneControl.verbose = true
-      ex.run
-      NPlusOneControl.verbose = false
-    end
-
-    specify do
-      expect do
-        expect { Post.find_each { |p| p.user.name } }
-          .to perform_constant_number_of_queries
-      end.to raise_error(RSpec::Expectations::ExpectationNotMetError, /select .+ from/i)
-    end
-  end
-
-  context "with scale_factors", :n_plus_one do
-    populate { |n| create_list(:post, n) }
-
-    specify do
-      expect { Post.find_each { |p| p.user.name } }
-        .to perform_constant_number_of_queries.with_scale_factors(1, 1)
-    end
-  end
-
-  context "with matching", :n_plus_one do
-    populate { |n| create_list(:post, n) }
-
-    specify do
-      expect { Post.find_each { |p| p.user.name } }
-        .to perform_constant_number_of_queries.matching(/posts/)
-    end
-  end
-
-  context 'with warming up', :n_plus_one do
-    let(:cache) { double "cache" }
-
-    before do
-      allow(cache).to receive(:setup).and_return(:result)
-      allow(NPlusOneControl::Executor).to receive(:call) { raise StandardError }
-    end
-
-    populate { |n| create_list(:post, n) }
-
-    warmup { cache.setup }
-
-    it "runs warmup before calling Executor" do
-      expect(cache).to receive(:setup)
-      expect do
-        expect { Post.find_each(&:id) }.to perform_constant_number_of_queries
-      end.to raise_error StandardError
-    end
-  end
-
-  context 'with_warming_up', :n_plus_one do
-    populate { |n| create_list(:post, n) }
-
-    it "runs actual one more time" do
-      expect(Post).to receive(:all).exactly(3).times
-      expect { Post.all }.to perform_constant_number_of_queries.with_warming_up
-    end
-  end
-end

data/spec/n_plus_one_control_spec.rb

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-require "spec_helper"
-
-describe NPlusOneControl do
-  it "has a version number" do
-    expect(NPlusOneControl::VERSION).not_to be nil
-  end
-end

data/spec/spec_helper.rb

@@ -1,30 +0,0 @@
-# frozen_string_literal: true
-
-$LOAD_PATH.unshift File.expand_path('../lib', __dir__)
-require "n_plus_one_control/rspec"
-require "benchmark"
-require "active_record"
-require "factory_girl"
-require "pry-byebug"
-
-ActiveRecord::Base.establish_connection(adapter: 'sqlite3', database: ':memory:')
-
-Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each { |f| require f }
-
-RSpec.configure do |config|
-  config.mock_with :rspec
-
-  config.order = :random
-  config.filter_run focus: true
-  config.run_all_when_everything_filtered = true
-
-  config.include FactoryGirl::Syntax::Methods
-
-  config.before(:each) do
-    ActiveRecord::Base.connection.begin_transaction(joinable: false)
-  end
-
-  config.after(:each) do
-    ActiveRecord::Base.connection.rollback_transaction
-  end
-end

data/spec/support/post.rb

@@ -1,19 +0,0 @@
-# frozen_string_literal: true
-
-ActiveRecord::Schema.define do
-  create_table :posts do |t|
-    t.string :title
-    t.integer :user_id
-  end
-end
-
-class Post < ActiveRecord::Base
-  belongs_to :user
-end
-
-FactoryGirl.define do
-  factory :post do
-    title "Title"
-    user
-  end
-end

data/spec/support/user.rb

@@ -1,17 +0,0 @@
-# frozen_string_literal: true
-
-ActiveRecord::Schema.define do
-  create_table :users do |t|
-    t.string :name
-  end
-end
-
-class User < ActiveRecord::Base
-  has_many :posts
-end
-
-FactoryGirl.define do
-  factory :user do
-    name "John"
-  end
-end

data/tests/minitest_test.rb

@@ -1,63 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "test_helper"
-
-class TestMinitest < Minitest::Test
-  def test_no_n_plus_one_error
-    populate = ->(n) { create_list(:post, n) }
-
-    assert_perform_constant_number_of_queries(populate: populate) do
-      Post.preload(:user).find_each { |p| p.user.name }
-    end
-  end
-
-  def test_with_n_plus_one_error
-    populate = ->(n) { create_list(:post, n) }
-
-    e = assert_raises Minitest::Assertion do
-      assert_perform_constant_number_of_queries(populate: populate) do
-        Post.find_each { |p| p.user.name }
-      end
-    end
-
-    assert_match "Expected to make the same number of queries", e.message
-    assert_match "3 for N=2", e.message
-    assert_match "4 for N=3", e.message
-  end
-
-  def test_no_n_plus_one_error_with_scale_factors
-    populate = ->(n) { create_list(:post, n) }
-
-    assert_perform_constant_number_of_queries(
-      populate: populate,
-      scale_factors: [1, 1]
-    ) do
-      Post.find_each { |p| p.user.name }
-    end
-  end
-
-  def test_no_n_plus_one_error_with_matching
-    populate = ->(n) { create_list(:post, n) }
-
-    assert_perform_constant_number_of_queries(
-      populate: populate,
-      matching: /posts/
-    ) do
-      Post.find_each { |p| p.user.name }
-    end
-  end
-
-  def populate(n)
-    create_list(:post, n)
-  end
-
-  def test_fallback_to_populate_method
-    e = assert_raises Minitest::Assertion do
-      assert_perform_constant_number_of_queries do
-        Post.find_each { |p| p.user.name }
-      end
-    end
-
-    assert_match "Expected to make the same number of queries", e.message
-  end
-end

data/tests/test_helper.rb

@@ -1,32 +0,0 @@
-# frozen_string_literal: true
-
-require "minitest/autorun"
-require "minitest/pride"
-
-$LOAD_PATH << File.expand_path('../lib', __dir__)
-Thread.abort_on_exception = true
-
-require "n_plus_one_control/minitest"
-require "benchmark"
-require "active_record"
-require "factory_girl"
-require "pry-byebug"
-
-ActiveRecord::Base.establish_connection(adapter: 'sqlite3', database: ':memory:')
-
-Dir["#{File.dirname(__FILE__)}/../spec/support/**/*.rb"].each { |f| require f }
-
-module TransactionalTests
-  def setup
-    ActiveRecord::Base.connection.begin_transaction(joinable: false)
-    super
-  end
-
-  def teardown
-    super
-    ActiveRecord::Base.connection.rollback_transaction
-  end
-end
-
-Minitest::Test.prepend TransactionalTests
-Minitest::Test.include FactoryGirl::Syntax::Methods