activerecord-health 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8ffd824e810c1dc16ebe84ec333bacd29867f80320b278b95ba5f9132e3929e0
-  data.tar.gz: 28599edb6d93b0e21b9cfd1c52f938a4394060e6757cf805195f0e410489548b
+  metadata.gz: c72d2a420a64a52fd3cf223bafb4e84e4dab50b5fa807b512a322a8dc909f62d
+  data.tar.gz: 98962aa4a2527386c888791d9b123967a004175305f985068e475b4f8557e653
 SHA512:
-  metadata.gz: 3c185ce40c985568ad318b5d356cc967e5781a3e5ee98259ebc9edbff3a28f4d409fde79945a00f7b00a8dd31b9df84de3785f66f9993745be08a87861b8c755
-  data.tar.gz: c3199b2e98cfdd1059e2e98bdbd1c640b8a0efa3d4de5ceddd1e70efcb9f8502d2a365ccd934158a2b2a01bddc51a4f7ab8821bfd774ce4aaacb092d6e1164b4
+  metadata.gz: c816d187880de866ee3e48fd972ea3b80543bfeba86d4a16466279a0bb45ae776f976f16f7d39dbdb68dd422f1f3fe0d92b8bf4ba2269e6c462ce6a85337b3c3
+  data.tar.gz: 291b81fb5116df82c6b310d17e7077701648e16b9eacf24cd57447f60bda87e8e0483d13b4cc3b92ba772d878566bf11a098a2a32e4bb80a84e5a357139c596c

data/CHANGELOG.md

@@ -0,0 +1,12 @@
+# Changelog
+
+## 0.2.0
+
+### Bug Fixes
+
+- Fix Rails initialization issue (#2)
+- Remove nonexistent exception class (#1)
+
+## 0.1.0
+
+- Initial release

data/README.md

@@ -137,7 +137,7 @@ class DatabaseHealthMiddleware
 
   def call(_worker, job, _queue)
     if THROTTLED_QUEUES.include?(job["queue"]) && !ActiveRecord::Health.ok?
-      raise ActiveRecord::Health::Unhealthy
+      raise "Database unhealthy, try again later"
     end
     yield
   end
@@ -150,6 +150,8 @@ Sidekiq.configure_server do |config|
 end
 ```
 
+This uses Sidekiq's [exponential backoff retries](https://github.com/sidekiq/sidekiq/wiki/Error-Handling#automatic-job-retry) to push load into the future.
+
 ## Multi-Database Support
 
 Pass the model class that connects to your database:

data/lib/activerecord/health/adapters/mysql_adapter.rb

@@ -8,6 +8,11 @@ module ActiveRecord
 
         attr_reader :version_string
 
+        def self.build(connection)
+          version = connection.select_value("SELECT VERSION()")
+          new(version)
+        end
+
         def initialize(version_string)
           @version_string = version_string
         end
@@ -24,6 +29,13 @@ module ActiveRecord
           !mariadb? && mysql_version >= PERFORMANCE_SCHEMA_MIN_VERSION
         end
 
+        def execute_with_timeout(connection, query, timeout)
+          connection.transaction do
+            connection.execute("SET max_execution_time = #{timeout * 1000}")
+            connection.select_value(query)
+          end
+        end
+
         private
 
         def mariadb?

data/lib/activerecord/health/adapters/postgresql_adapter.rb

@@ -4,6 +4,10 @@ module ActiveRecord
   module Health
     module Adapters
       class PostgreSQLAdapter
+        def self.build(_connection)
+          new
+        end
+
         def name
           :postgresql
         end
@@ -17,6 +21,13 @@ module ActiveRecord
               AND pid != pg_backend_pid()
           SQL
         end
+
+        def execute_with_timeout(connection, query, timeout)
+          connection.transaction do
+            connection.execute("SET LOCAL statement_timeout = '#{timeout}s'")
+            connection.select_value(query)
+          end
+        end
       end
     end
   end

data/lib/activerecord/health/railtie.rb

@@ -3,7 +3,7 @@
 module ActiveRecord
   module Health
     class Railtie < Rails::Railtie
-      initializer "activerecord_health.validate_configuration" do
+      config.after_initialize do
         ActiveRecord::Health.configuration.validate!
       end
     end

data/lib/activerecord/health/version.rb

@@ -2,6 +2,6 @@
 
 module Activerecord
   module Health
-    VERSION = "0.1.0"
+    VERSION = "0.2.0"
   end
 end

data/lib/activerecord/health.rb

@@ -54,68 +54,47 @@ module ActiveRecord
       end
 
       def read_from_cache(cache_key)
-        cached_value = configuration.cache.read(cache_key)
-        return cached_value unless cached_value.nil?
+        configuration.cache.read(cache_key) || write_to_cache(cache_key, yield)
+      rescue
+        0.0
+      end
 
-        value = yield
+      def write_to_cache(cache_key, value)
         configuration.cache.write(cache_key, value, expires_in: configuration.cache_ttl)
         value
-      rescue
-        0.0
       end
 
       def query_load_pct(model)
-        connection = model.connection
-        adapter = adapter_for(connection)
-        config = config_for(model)
-        db_config_name = model.connection_db_config.name
-
-        active_sessions = execute_with_timeout(connection, adapter.active_session_count_query)
-        load_pct = active_sessions.to_f / config.vcpu_count
-
-        instrument(db_config_name, load_pct, active_sessions)
-
-        load_pct
+        active_sessions = fetch_active_sessions(model)
+        calculate_and_instrument_load(model, active_sessions)
       rescue
         1.0
       end
 
-      def execute_with_timeout(connection, query)
-        adapter_name = connection.adapter_name.downcase
-
-        case adapter_name
-        when /postgresql/
-          execute_with_postgresql_timeout(connection, query)
-        when /mysql/
-          execute_with_mysql_timeout(connection, query)
-        else
-          connection.select_value(query)
-        end
+      def fetch_active_sessions(model)
+        adapter = adapter_for(model.connection)
+        execute_with_timeout(model.connection, adapter, adapter.active_session_count_query)
       end
 
-      def execute_with_postgresql_timeout(connection, query)
-        connection.transaction do
-          connection.execute("SET LOCAL statement_timeout = '#{QUERY_TIMEOUT}s'")
-          connection.select_value(query)
-        end
+      def calculate_and_instrument_load(model, active_sessions)
+        load_pct = active_sessions.to_f / config_for(model).vcpu_count
+        instrument(model.connection_db_config.name, load_pct, active_sessions)
+        load_pct
       end
 
-      def execute_with_mysql_timeout(connection, query)
-        connection.transaction do
-          connection.execute("SET max_execution_time = #{QUERY_TIMEOUT * 1000}")
-          connection.select_value(query)
-        end
+      def execute_with_timeout(connection, adapter, query)
+        adapter.execute_with_timeout(connection, query, QUERY_TIMEOUT)
       end
 
       def adapter_for(connection)
+        adapter_class_for(connection).build(connection)
+      end
+
+      def adapter_class_for(connection)
         case connection.adapter_name.downcase
-        when /postgresql/
-          Adapters::PostgreSQLAdapter.new
-        when /mysql/
-          version = connection.select_value("SELECT VERSION()")
-          Adapters::MySQLAdapter.new(version)
-        else
-          raise "Unsupported database adapter: #{connection.adapter_name}"
+        when /postgresql/ then Adapters::PostgreSQLAdapter
+        when /mysql/ then Adapters::MySQLAdapter
+        else raise "Unsupported database adapter: #{connection.adapter_name}"
         end
       end
 

data/test/integration/mysql_integration_test.rb

@@ -17,7 +17,7 @@ class MySQLIntegrationTest < ActiveRecord::Health::TestCase
       database: ENV.fetch("MYSQL_DB", "activerecord_health_test")
     )
 
-    @cache = MockCache.new
+    @cache = ActiveSupport::Cache::MemoryStore.new
     @baseline_sessions = count_active_sessions
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = @baseline_sessions + 4

data/test/integration/postgresql_integration_test.rb

@@ -17,7 +17,7 @@ class PostgreSQLIntegrationTest < ActiveRecord::Health::TestCase
       database: ENV.fetch("POSTGRES_DB", "activerecord_health_test")
     )
 
-    @cache = MockCache.new
+    @cache = ActiveSupport::Cache::MemoryStore.new
     @baseline_sessions = count_active_sessions
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = @baseline_sessions + 4

data/test/integration/rails_app/config/application.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require_relative "boot"
+
+require "rails"
+require "active_record/railtie"
+require "activerecord/health"
+
+module RailsApp
+  class Application < Rails::Application
+    config.load_defaults Rails::VERSION::STRING.to_f
+    config.eager_load = false
+    config.logger = Logger.new(nil)
+  end
+end

data/test/integration/rails_app/config/boot.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+# Set up gems listed in the Gemfile.
+ENV["BUNDLE_GEMFILE"] ||= File.expand_path("../../../../Gemfile", __dir__)
+
+require "bundler/setup" if File.exist?(ENV["BUNDLE_GEMFILE"])
+$LOAD_PATH.unshift File.expand_path("../../../../lib", __dir__)

data/test/integration/rails_app/config/database.yml

@@ -0,0 +1,3 @@
+test:
+  adapter: sqlite3
+  database: ":memory:"

data/test/integration/rails_app/config/environment.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+require_relative "application"
+
+Rails.application.initialize!

data/test/integration/rails_app/config/fixtures/missing_cache.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+ActiveRecord::Health.configure do |config|
+  config.vcpu_count = 4
+end

data/test/integration/rails_app/config/fixtures/missing_vcpu.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+ActiveRecord::Health.configure do |config|
+  config.cache = ActiveSupport::Cache::MemoryStore.new
+end

data/test/integration/rails_app/config/fixtures/valid.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+ActiveRecord::Health.configure do |config|
+  config.vcpu_count = 4
+  config.cache = ActiveSupport::Cache::MemoryStore.new
+end

data/test/integration/rails_app/config/initializers/activerecord_health.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+fixture = ENV.fetch("HEALTH_CONFIG_FIXTURE", "valid")
+require_relative "../fixtures/#{fixture}"

data/test/integration/rails_integration_test.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require "test_helper"
+require "open3"
+
+class RailsIntegrationTest < Minitest::Test
+  def test_boots_successfully_with_valid_initializer_configuration
+    output, status = run_rails_runner("puts ActiveRecord::Health.configuration.vcpu_count")
+
+    assert status.success?, "Expected success but got: #{output}"
+    assert_equal "4", output.strip
+  end
+
+  def test_configuration_from_initializer_is_applied
+    output, status = run_rails_runner(<<~RUBY)
+      puts ActiveRecord::Health.configuration.vcpu_count
+      puts ActiveRecord::Health.configuration.cache.class.name
+    RUBY
+
+    assert status.success?, "Expected success but got: #{output}"
+    lines = output.strip.split("\n")
+    assert_equal "4", lines[0]
+    assert_equal "ActiveSupport::Cache::MemoryStore", lines[1]
+  end
+
+  def test_raises_configuration_error_when_vcpu_count_missing
+    output, status = run_rails_runner("puts 'ok'", fixture: "missing_vcpu")
+
+    refute status.success?
+    assert_match(/vcpu_count must be configured/, output)
+  end
+
+  def test_raises_configuration_error_when_cache_missing
+    output, status = run_rails_runner("puts 'ok'", fixture: "missing_cache")
+
+    refute status.success?
+    assert_match(/cache must be configured/, output)
+  end
+
+  def test_configuration_defaults_are_preserved
+    output, status = run_rails_runner(<<~RUBY)
+      puts ActiveRecord::Health.configuration.threshold
+      puts ActiveRecord::Health.configuration.cache_ttl
+    RUBY
+
+    assert status.success?, "Expected success but got: #{output}"
+    lines = output.strip.split("\n")
+    assert_equal "0.75", lines[0]
+    assert_equal "60", lines[1]
+  end
+
+  private
+
+  def run_rails_runner(code, fixture: "valid")
+    rails_app_path = File.expand_path("rails_app", __dir__)
+
+    env = {
+      "BUNDLE_GEMFILE" => File.expand_path("../../Gemfile", __dir__),
+      "HEALTH_CONFIG_FIXTURE" => fixture
+    }
+    cmd = "bundle exec ruby -e \"require '#{rails_app_path}/config/environment'; #{code.gsub('"', '\\"').gsub("\n", "; ")}\""
+
+    Open3.capture2e(env, cmd, chdir: rails_app_path)
+  end
+end

data/test/test_helper.rb

@@ -14,19 +14,9 @@ require "minitest/reporters"
 Minitest::Reporters.use! Minitest::Reporters::DefaultReporter.new
 
 if ENV["FAIL_ON_SKIP"]
-  module Minitest
-    class << self
-      alias_method :original_run, :run
-
-      def run(args = [])
-        result = original_run(args)
-        reporter = Minitest::Reporters.reporters.first
-        if reporter && reporter.skips > 0
-          warn "\nCI failed: #{reporter.skips} tests were skipped"
-          exit 1
-        end
-        result
-      end
+  class Minitest::Test
+    def skip(msg = nil, _bt = caller)
+      flunk(msg || "Test was skipped")
     end
   end
 end
@@ -41,28 +31,6 @@ class ActiveRecord::Health::TestCase < Minitest::Test
   end
 end
 
-class MockCache
-  def initialize
-    @store = {}
-  end
-
-  def read(key)
-    @store[key]
-  end
-
-  def write(key, value, options = {})
-    @store[key] = value
-  end
-
-  def delete(key)
-    @store.delete(key)
-  end
-
-  def clear
-    @store.clear
-  end
-end
-
 class FailingCache
   def read(key)
     raise "Cache connection failed"

data/test/unit/configuration_test.rb

@@ -6,7 +6,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
   def test_configure_sets_vcpu_count
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
     end
 
     assert_equal 16, ActiveRecord::Health.configuration.vcpu_count
@@ -15,7 +15,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
   def test_configure_sets_threshold_with_default
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
     end
 
     assert_equal 0.75, ActiveRecord::Health.configuration.threshold
@@ -25,14 +25,14 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
       config.threshold = 0.5
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
     end
 
     assert_equal 0.5, ActiveRecord::Health.configuration.threshold
   end
 
   def test_configure_sets_cache
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
       config.cache = cache
@@ -44,7 +44,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
   def test_configure_sets_cache_ttl_with_default
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
     end
 
     assert_equal 60, ActiveRecord::Health.configuration.cache_ttl
@@ -53,7 +53,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
   def test_configure_sets_custom_cache_ttl
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
       config.cache_ttl = 120
     end
 
@@ -62,7 +62,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
 
   def test_raises_without_vcpu_count
     ActiveRecord::Health.configure do |config|
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
     end
 
     error = assert_raises(ActiveRecord::Health::ConfigurationError) do
@@ -85,7 +85,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
   def test_for_model_configures_specific_database
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
 
       config.for_model(AnimalsRecord) do |db|
         db.vcpu_count = 8
@@ -99,7 +99,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
   end
 
   def test_for_model_inherits_cache_from_default
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
       config.cache = cache
@@ -115,7 +115,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
   def test_for_model_inherits_threshold_from_default_when_not_specified
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
 
       config.for_model(AnimalsRecord) do |db|
         db.vcpu_count = 8
@@ -129,7 +129,7 @@ class ConfigurationTest < ActiveRecord::Health::TestCase
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
       config.threshold = 0.75
-      config.cache = MockCache.new
+      config.cache = ActiveSupport::Cache::MemoryStore.new
     end
 
     assert_equal 12, ActiveRecord::Health.configuration.max_healthy_sessions

data/test/unit/extensions_test.rb

@@ -5,7 +5,7 @@ require "activerecord/health/extensions"
 
 class ConnectionExtensionTest < ActiveRecord::Health::TestCase
   def test_healthy_returns_true_when_below_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.5)
 
     ActiveRecord::Health.configure do |config|
@@ -19,7 +19,7 @@ class ConnectionExtensionTest < ActiveRecord::Health::TestCase
   end
 
   def test_healthy_returns_false_when_above_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.9)
 
     ActiveRecord::Health.configure do |config|
@@ -33,7 +33,7 @@ class ConnectionExtensionTest < ActiveRecord::Health::TestCase
   end
 
   def test_load_pct_returns_cached_value
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.625)
 
     ActiveRecord::Health.configure do |config|
@@ -48,7 +48,7 @@ end
 
 class ModelExtensionTest < ActiveRecord::Health::TestCase
   def test_database_healthy_returns_true_when_below_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.5)
 
     ActiveRecord::Health.configure do |config|
@@ -61,7 +61,7 @@ class ModelExtensionTest < ActiveRecord::Health::TestCase
   end
 
   def test_database_healthy_returns_false_when_above_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.9)
 
     ActiveRecord::Health.configure do |config|

data/test/unit/health_test.rb

@@ -4,7 +4,7 @@ require "test_helper"
 
 class HealthTest < ActiveRecord::Health::TestCase
   def test_ok_returns_true_when_load_below_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.5)
 
     ActiveRecord::Health.configure do |config|
@@ -18,7 +18,7 @@ class HealthTest < ActiveRecord::Health::TestCase
   end
 
   def test_ok_returns_false_when_load_above_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.9)
 
     ActiveRecord::Health.configure do |config|
@@ -32,7 +32,7 @@ class HealthTest < ActiveRecord::Health::TestCase
   end
 
   def test_ok_returns_true_when_load_equals_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.75)
 
     ActiveRecord::Health.configure do |config|
@@ -56,7 +56,7 @@ class HealthTest < ActiveRecord::Health::TestCase
   end
 
   def test_load_pct_returns_cached_value
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.625)
 
     ActiveRecord::Health.configure do |config|
@@ -79,7 +79,7 @@ class HealthTest < ActiveRecord::Health::TestCase
   end
 
   def test_load_pct_queries_database_when_not_cached
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
 
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
@@ -94,7 +94,7 @@ class HealthTest < ActiveRecord::Health::TestCase
   end
 
   def test_load_pct_returns_1_0_when_database_query_fails
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
 
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
@@ -108,7 +108,7 @@ class HealthTest < ActiveRecord::Health::TestCase
   end
 
   def test_ok_returns_false_when_database_query_fails
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
 
     ActiveRecord::Health.configure do |config|
       config.vcpu_count = 16
@@ -123,7 +123,7 @@ class HealthTest < ActiveRecord::Health::TestCase
   end
 
   def test_uses_per_model_configuration
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:animals", 0.6)
 
     ActiveRecord::Health.configure do |config|

data/test/unit/sheddable_test.rb

@@ -4,7 +4,7 @@ require "test_helper"
 
 class SheddableTest < ActiveRecord::Health::TestCase
   def test_sheddable_executes_block_and_returns_true_when_healthy
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.5)
 
     ActiveRecord::Health.configure do |config|
@@ -22,7 +22,7 @@ class SheddableTest < ActiveRecord::Health::TestCase
   end
 
   def test_sheddable_returns_false_and_skips_block_when_overloaded
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.9)
 
     ActiveRecord::Health.configure do |config|
@@ -40,7 +40,7 @@ class SheddableTest < ActiveRecord::Health::TestCase
   end
 
   def test_sheddable_pct_executes_block_and_returns_true_when_below_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.4)
 
     ActiveRecord::Health.configure do |config|
@@ -57,7 +57,7 @@ class SheddableTest < ActiveRecord::Health::TestCase
   end
 
   def test_sheddable_pct_returns_false_and_skips_block_when_above_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.6)
 
     ActiveRecord::Health.configure do |config|
@@ -74,7 +74,7 @@ class SheddableTest < ActiveRecord::Health::TestCase
   end
 
   def test_sheddable_pct_executes_block_and_returns_true_when_at_threshold
-    cache = MockCache.new
+    cache = ActiveSupport::Cache::MemoryStore.new
     cache.write("activerecord_health:load_pct:primary", 0.5)
 
     ActiveRecord::Health.configure do |config|

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-health
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Nate Berkopec
@@ -45,6 +45,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- CHANGELOG.md
 - CONTRIBUTING.md
 - LICENSE.txt
 - README.md
@@ -60,6 +61,15 @@ files:
 - lib/activerecord/health/version.rb
 - test/integration/mysql_integration_test.rb
 - test/integration/postgresql_integration_test.rb
+- test/integration/rails_app/config/application.rb
+- test/integration/rails_app/config/boot.rb
+- test/integration/rails_app/config/database.yml
+- test/integration/rails_app/config/environment.rb
+- test/integration/rails_app/config/fixtures/missing_cache.rb
+- test/integration/rails_app/config/fixtures/missing_vcpu.rb
+- test/integration/rails_app/config/fixtures/valid.rb
+- test/integration/rails_app/config/initializers/activerecord_health.rb
+- test/integration/rails_integration_test.rb
 - test/test_helper.rb
 - test/unit/adapters/mysql_adapter_test.rb
 - test/unit/adapters/postgresql_adapter_test.rb