test-prof 0.11.3 → 1.0.0.rc2

data/lib/test_prof/any_fixture/dump/base_adapter.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module TestProf
+  module AnyFixture
+    class Dump
+      class BaseAdapter
+        def reset_sequence!(_table_name, _start)
+        end
+
+        def compile_sql(sql, _binds)
+          sql
+        end
+
+        def setup_env
+        end
+
+        def teardown_env
+        end
+
+        def import(_path)
+          false
+        end
+
+        private
+
+        def while_disconnected
+          conn.disconnect!
+          yield
+        ensure
+          conn.reconnect!
+        end
+
+        def conn
+          ActiveRecord::Base.connection
+        end
+
+        def execute(query)
+          conn.execute(query)
+        end
+      end
+    end
+  end
+end

data/lib/test_prof/any_fixture/dump/digest.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require "digest/sha1"
+
+module TestProf
+  module AnyFixture
+    class Dump
+      module Digest
+        module_function
+
+        def call(*paths)
+          files = (AnyFixture.config.default_dump_watch_paths + paths).each_with_object([]) do |path_or_glob, acc|
+            if File.file?(path_or_glob)
+              acc << path_or_glob
+            else
+              acc = acc.concat Dir[path_or_glob]
+            end
+            acc
+          end
+
+          return if files.empty?
+
+          file_ids = files.sort.map { |f| "#{File.basename(f)}/#{::Digest::SHA1.file(f).hexdigest}" }
+          ::Digest::SHA1.hexdigest(file_ids.join("/"))
+        end
+      end
+    end
+  end
+end

data/lib/test_prof/any_fixture/dump/postgresql.rb

@@ -0,0 +1,91 @@
+# frozen_string_literal: true
+
+require "test_prof/any_fixture/dump/base_adapter"
+
+module TestProf
+  module AnyFixture
+    class Dump
+      class PostgreSQL < BaseAdapter
+        UUID_FUNCTIONS = %w[
+          gen_random_uuid
+          uuid_generate_v4
+        ]
+
+        def reset_sequence!(table_name, start)
+          _pk, sequence = conn.pk_and_sequence_for(table_name)
+          return unless sequence
+
+          sequence_name = "#{sequence.schema}.#{sequence.identifier}"
+
+          execute <<~SQL
+            ALTER SEQUENCE #{sequence_name} RESTART WITH #{start}; -- any_fixture:dump
+          SQL
+        end
+
+        def compile_sql(sql, binds)
+          sql.gsub(/\$\d+/) { binds.shift }
+        end
+
+        def import(path)
+          # Test if psql is installed
+          `psql --version`
+
+          tasks = ActiveRecord::Tasks::PostgreSQLDatabaseTasks.new(conn.pool.spec.config.with_indifferent_access)
+
+          while_disconnected do
+            tasks.structure_load(path, "--output=/dev/null")
+          end
+
+          true
+        rescue Errno::ENOENT
+          false
+        end
+
+        def setup_env
+          # Mock UUID generating functions to provide consistent results
+          quoted_functions = UUID_FUNCTIONS.map { |func| "'#{func}'" }.join(", ")
+
+          @uuid_funcs = execute <<~SQL
+            SELECT
+              pp.proname, pn.nspname,
+              pg_get_functiondef(pp.oid) AS definition
+            FROM pg_proc pp
+            JOIN pg_namespace pn
+              ON pn.oid = pp.pronamespace
+            WHERE pp.proname in (#{quoted_functions})
+            ORDER BY pp.oid;
+          SQL
+
+          uuid_funcs.each do |(func, ns, _)|
+            execute <<~SQL
+              CREATE OR REPLACE FUNCTION #{ns}.#{func}()
+                RETURNS UUID
+                LANGUAGE SQL
+                AS $$
+                  SELECT md5(random()::TEXT)::UUID;
+                $$; -- any_fixture:dump
+            SQL
+          end
+
+          execute <<~SQL
+            SELECT setseed(#{rand}); -- any_fixture:dump
+          SQL
+        end
+
+        def teardown_env
+          uuid_funcs.each do |(func, ns, definition)|
+            execute "#{definition}; -- any_fixture:dump"
+          end
+        end
+
+        private
+
+        attr_reader :uuid_funcs
+
+        def execute(query)
+          super.values
+        end
+      end
+    end
+  end
+end

data/lib/test_prof/any_fixture/dump/sqlite.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require "test_prof/any_fixture/dump/base_adapter"
+
+module TestProf
+  module AnyFixture
+    class Dump
+      class SQLite < BaseAdapter
+        def reset_sequence!(table_name, start)
+          execute <<~SQL.chomp
+            DELETE FROM sqlite_sequence WHERE name=#{table_name}
+          SQL
+
+          execute <<~SQL.chomp
+            INSERT INTO sqlite_sequence (name, seq)
+            VALUES (#{table_name}, #{start})
+          SQL
+        end
+
+        def compile_sql(sql, binds)
+          sql.gsub(/\?/) { binds.shift }
+        end
+
+        def import(path)
+          db = conn.pool.spec.config[:database]
+          return false if %r{:memory:}.match?(db)
+
+          # Check that sqlite3 is installed
+          `sqlite3 --version`
+
+          while_disconnected do
+            `sqlite3 #{db} < "#{path}"`
+          end
+
+          true
+        rescue Errno::ENOENT
+          false
+        end
+      end
+    end
+  end
+end

data/lib/test_prof/before_all.rb

@@ -24,10 +24,6 @@ module TestProf
         yield
       end
 
-      def within_transaction
-        yield
-      end
-
       def rollback_transaction
         raise AdapterMissing if adapter.nil?
 
@@ -36,6 +32,12 @@ module TestProf
         end
       end
 
+      def setup_fixtures(test_object)
+        raise ArgumentError, "Current adapter doesn't support #setup_fixtures" unless adapter.respond_to?(:setup_fixtures)
+
+        adapter.setup_fixtures(test_object)
+      end
+
       def config
         @config ||= Configuration.new
       end
@@ -64,8 +66,11 @@ module TestProf
     class Configuration
       HOOKS = %i[begin rollback].freeze
 
+      attr_accessor :setup_fixtures
+
       def initialize
         @hooks = Hash.new { |h, k| h[k] = HooksChain.new(k) }
+        @setup_fixtures = false
       end
 
       # Add `before` hook for `begin` or

data/lib/test_prof/before_all/adapters/active_record.rb

@@ -1,10 +1,5 @@
 # frozen_string_literal: true
 
-if ::ActiveRecord::VERSION::MAJOR < 4
-  require "test_prof/ext/active_record_3"
-  using TestProf::ActiveRecord3Transactions
-end
-
 module TestProf
   module BeforeAll
     module Adapters
@@ -23,6 +18,20 @@ module TestProf
             end
             ::ActiveRecord::Base.connection.rollback_transaction
           end
+
+          def setup_fixtures(test_object)
+            test_object.instance_eval do
+              @@already_loaded_fixtures ||= {}
+              @fixture_cache ||= {}
+
+              if @@already_loaded_fixtures[self.class]
+                @loaded_fixtures = @@already_loaded_fixtures[self.class]
+              else
+                @loaded_fixtures = load_fixtures(config)
+                @@already_loaded_fixtures[self.class] = @loaded_fixtures
+              end
+            end
+          end
         end
       end
     end

data/lib/test_prof/cops/rspec/aggregate_examples.rb

@@ -12,7 +12,7 @@ module RuboCop
     module RSpec
       # Checks if example groups contain two or more aggregatable examples.
       #
-      # @see https://github.com/rubocop-hq/rspec-style-guide#expectations-per-example
+      # @see https://github.com/rubocop-hq/rspec-style-guide#expectation-per-example
       #
       # This cop is primarily for reducing the cost of repeated expensive
       # context initialization.
@@ -108,7 +108,7 @@ module RuboCop
       #     expect(number).to be_odd
       #   end
       #
-      class AggregateExamples < Cop
+      class AggregateExamples < ::RuboCop::Cop::Cop
         include LineRangeHelpers
         include MetadataHelpers
         include NodeMatchers

data/lib/test_prof/cops/rspec/aggregate_examples/its.rb

@@ -3,7 +3,7 @@
 module RuboCop
   module Cop
     module RSpec
-      class AggregateExamples < Cop
+      class AggregateExamples < ::RuboCop::Cop::Cop
         # @example `its`
         #
         #   # Supports regular `its` call with an attribute/method name,

data/lib/test_prof/cops/rspec/aggregate_examples/line_range_helpers.rb

@@ -3,7 +3,7 @@
 module RuboCop
   module Cop
     module RSpec
-      class AggregateExamples < Cop
+      class AggregateExamples < ::RuboCop::Cop::Cop
         # @internal Support methods for keeping newlines around examples.
         module LineRangeHelpers
           include RangeHelp

data/lib/test_prof/cops/rspec/aggregate_examples/matchers_with_side_effects.rb

@@ -5,7 +5,7 @@ require_relative "../language"
 module RuboCop
   module Cop
     module RSpec
-      class AggregateExamples < Cop
+      class AggregateExamples < ::RuboCop::Cop::Cop
         # When aggregated, the expectations will fail when not supposed to or
         # have a risk of not failing when expected to. One example is
         # `validate_presence_of :comment` as it leaves an empty comment after

data/lib/test_prof/cops/rspec/aggregate_examples/metadata_helpers.rb

@@ -3,7 +3,7 @@
 module RuboCop
   module Cop
     module RSpec
-      class AggregateExamples < Cop
+      class AggregateExamples < ::RuboCop::Cop::Cop
         # @internal
         #   Support methods for example metadata.
         #   Examples with similar metadata are grouped.

data/lib/test_prof/cops/rspec/aggregate_examples/node_matchers.rb

@@ -5,7 +5,7 @@ require_relative "../language"
 module RuboCop
   module Cop
     module RSpec
-      class AggregateExamples < Cop
+      class AggregateExamples < ::RuboCop::Cop::Cop
         # @internal
         #   Node matchers and searchers.
         module NodeMatchers

data/lib/test_prof/event_prof/instrumentations/active_support.rb

@@ -4,13 +4,31 @@ module TestProf::EventProf
   module Instrumentations
     # Wrapper over ActiveSupport::Notifications
     module ActiveSupport
+      class Subscriber
+        attr_reader :block, :started_at
+
+        def initialize(block)
+          @block = block
+        end
+
+        def start(*)
+          @started_at = TestProf.now
+        end
+
+        def publish(_name, started_at, finished_at, *)
+          block.call(finished_at - started_at)
+        end
+
+        def finish(*)
+          block.call(TestProf.now - started_at)
+        end
+      end
+
       class << self
-        def subscribe(event)
+        def subscribe(event, &block)
           raise ArgumentError, "Block is required!" unless block_given?
 
-          ::ActiveSupport::Notifications.subscribe(event) do |_event, start, finish, *_args|
-            yield (finish - start)
-          end
+          ::ActiveSupport::Notifications.subscribe(event, Subscriber.new(block))
         end
 
         def instrument(event)

data/lib/test_prof/recipes/minitest/before_all.rb

@@ -8,41 +8,59 @@ module TestProf
     # store instance variables
     module Minitest # :nodoc: all
       class Executor
-        attr_reader :active
+        attr_reader :active, :block, :captured_ivars, :teardown_block, :current_test_object,
+          :setup_fixtures
 
         alias active? active
+        alias setup_fixtures? setup_fixtures
 
-        def initialize(&block)
+        def initialize(setup_fixtures: false, &block)
+          @setup_fixtures = setup_fixtures
           @block = block
+          @captured_ivars = []
         end
 
-        def activate!(test_class)
-          return if active?
+        def teardown(&block)
+          @teardown_block = block
+        end
+
+        def activate!(test_object)
+          @current_test_object = test_object
+
+          return restore_ivars(test_object) if active?
           @active = true
-          @examples_left = test_class.runnable_methods.size
+          BeforeAll.setup_fixtures(test_object) if setup_fixtures?
           BeforeAll.begin_transaction do
-            capture!
+            capture!(test_object)
           end
         end
 
-        def try_deactivate!
-          @examples_left -= 1
-          return unless @examples_left.zero?
-
+        def deactivate!
           @active = false
+
+          current_test_object&.instance_eval(&teardown_block) if teardown_block
+
+          @current_test_object = nil
           BeforeAll.rollback_transaction
         end
 
-        def capture!
-          instance_eval(&@block)
+        def capture!(test_object)
+          return unless block
+
+          before_ivars = test_object.instance_variables
+
+          test_object.instance_eval(&block)
+
+          (test_object.instance_variables - before_ivars).each do |ivar|
+            captured_ivars << [ivar, test_object.instance_variable_get(ivar)]
+          end
         end
 
-        def restore_to(test_object)
-          instance_variables.each do |ivar|
-            next if ivar == :@block
+        def restore_ivars(test_object)
+          captured_ivars.each do |(ivar, val)|
             test_object.instance_variable_set(
               ivar,
-              instance_variable_get(ivar)
+              val
             )
           end
         end
@@ -57,22 +75,29 @@ module TestProf
       module ClassMethods
         attr_accessor :before_all_executor
 
-        def before_all(&block)
-          self.before_all_executor = Executor.new(&block)
+        def before_all(setup_fixtures: BeforeAll.config.setup_fixtures, &block)
+          self.before_all_executor = Executor.new(setup_fixtures: setup_fixtures, &block)
 
           prepend(Module.new do
-            def setup
-              self.class.before_all_executor.activate!(self.class)
-              self.class.before_all_executor.restore_to(self)
+            def before_setup
+              self.class.before_all_executor.activate!(self)
               super
             end
+          end)
 
-            def teardown
+          singleton_class.prepend(Module.new do
+            def run(*)
               super
-              self.class.before_all_executor.try_deactivate!
+            ensure
+              before_all_executor&.deactivate!
             end
           end)
         end
+
+        def after_all(&block)
+          self.before_all_executor ||= Executor.new
+          before_all_executor.teardown(&block)
+        end
       end
     end
   end