pecorino 0.7.2 → 0.7.4

data/test/adapters/postgres_adapter_test.rb

@@ -23,7 +23,7 @@ class PostgresAdapterTest < ActiveSupport::TestCase
 
   def create_postgres_database_if_none
     self.class.establish_connection(encoding: "unicode", database: SEED_DB_NAME.call)
-    ActiveRecord::Base.connection.execute("SELECT 1 FROM pecorino_leaky_buckets")
+    ActiveRecord::Base.connection_pool.with_connection { |connection| connection.execute("SELECT 1 FROM pecorino_leaky_buckets") }
   rescue ActiveRecord::NoDatabaseError, ActiveRecord::ConnectionNotEstablished
     create_postgres_database
     retry
@@ -38,20 +38,24 @@ class PostgresAdapterTest < ActiveSupport::TestCase
   def create_postgres_database
     ActiveRecord::Migration.verbose = false
     self.class.establish_connection(database: "postgres")
-    ActiveRecord::Base.connection.create_database(SEED_DB_NAME.call, charset: :unicode)
+    ActiveRecord::Base.connection_pool.with_connection { |connection| connection.create_database(SEED_DB_NAME.call, charset: :unicode) }
     ActiveRecord::Base.connection.close
     self.class.establish_connection(encoding: "unicode", database: SEED_DB_NAME.call)
   end
 
   def truncate_test_tables
-    ActiveRecord::Base.connection.execute("TRUNCATE TABLE pecorino_leaky_buckets")
-    ActiveRecord::Base.connection.execute("TRUNCATE TABLE pecorino_blocks")
+    ActiveRecord::Base.connection_pool.with_connection do |connection|
+      connection.execute("TRUNCATE TABLE pecorino_leaky_buckets")
+      connection.execute("TRUNCATE TABLE pecorino_blocks")
+    end
   end
 
   def test_create_tables
     ActiveRecord::Base.transaction do
-      ActiveRecord::Base.connection.execute("DROP TABLE pecorino_leaky_buckets")
-      ActiveRecord::Base.connection.execute("DROP TABLE pecorino_blocks")
+      ActiveRecord::Base.connection_pool.with_connection do |connection|
+        connection.execute("DROP TABLE pecorino_leaky_buckets")
+        connection.execute("DROP TABLE pecorino_blocks")
+      end
       # The adapter has to be in a variable as the schema definition is scoped to the migrator, not self
       retained_adapter = create_adapter # the schema define block is run via instance_exec so it does not retain scope
       ActiveRecord::Schema.define(version: 1) do |via_definer|
@@ -64,6 +68,6 @@ class PostgresAdapterTest < ActiveSupport::TestCase
   Minitest.after_run do
     ActiveRecord::Base.connection.close
     establish_connection(database: "postgres")
-    ActiveRecord::Base.connection.drop_database(SEED_DB_NAME.call)
+    ActiveRecord::Base.connection_pool.with_connection { |connection| connection.drop_database(SEED_DB_NAME.call) }
   end
 end

data/test/adapters/sqlite_adapter_test.rb

@@ -33,8 +33,10 @@ class SqliteAdapterTest < ActiveSupport::TestCase
 
   def test_create_tables
     ActiveRecord::Base.transaction do
-      ActiveRecord::Base.connection.execute("DROP TABLE pecorino_leaky_buckets")
-      ActiveRecord::Base.connection.execute("DROP TABLE pecorino_blocks")
+      ActiveRecord::Base.connection_pool.with_connection do |connection|
+        connection.execute("DROP TABLE pecorino_leaky_buckets")
+        connection.execute("DROP TABLE pecorino_blocks")
+      end
       # The adapter has to be in a variable as the schema definition is scoped to the migrator, not self
       retained_adapter = create_adapter # the schema define block is run via instance_exec so it does not retain scope
       ActiveRecord::Schema.define(version: 1) do |via_definer|

data/test/block_test.rb

@@ -4,20 +4,24 @@ require "test_helper"
 require "base64"
 
 class BlockTest < ActiveSupport::TestCase
+  def setup
+    @adapter = Pecorino::Adapters::MemoryAdapter.new
+  end
+
   test "sets a block" do
     k = Base64.strict_encode64(Random.bytes(4))
-    assert_nil Pecorino::Block.blocked_until(key: k)
-    assert Pecorino::Block.set!(key: k, block_for: 30.minutes)
+    assert_nil Pecorino::Block.blocked_until(key: k, adapter: @adapter)
+    assert Pecorino::Block.set!(key: k, block_for: 30.minutes, adapter: @adapter)
 
-    blocked_until = Pecorino::Block.blocked_until(key: k)
+    blocked_until = Pecorino::Block.blocked_until(key: k, adapter: @adapter)
     assert_in_delta Time.now + 30.minutes, blocked_until, 10
   end
 
   test "does not return a block which has lapsed" do
     k = Base64.strict_encode64(Random.bytes(4))
-    assert_nil Pecorino::Block.blocked_until(key: k)
-    Pecorino::Block.set!(key: k, block_for: -30.minutes)
-    blocked_until = Pecorino::Block.blocked_until(key: k)
+    assert_nil Pecorino::Block.blocked_until(key: k, adapter: @adapter)
+    Pecorino::Block.set!(key: k, block_for: -30.minutes, adapter: @adapter)
+    blocked_until = Pecorino::Block.blocked_until(key: k, adapter: @adapter)
     assert_nil blocked_until
   end
 end

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: pecorino
 version: !ruby/object:Gem::Version
-  version: 0.7.2
+  version: 0.7.4
 platform: ruby
 authors:
 - Julik Tarkhanov
-autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-04-01 00:00:00.000000000 Z
+date: 2025-08-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -35,8 +34,10 @@ files:
 - ".github/workflows/ci.yml"
 - ".gitignore"
 - ".standard.yml"
+- ".yardopts"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
+- Gemfile
 - LICENSE.txt
 - README.md
 - Rakefile
@@ -59,6 +60,7 @@ files:
 - lib/pecorino/version.rb
 - pecorino.gemspec
 - rbi/pecorino.rbi
+- rbi/pecorino.rbs
 - test/adapters/adapter_test_methods.rb
 - test/adapters/memory_adapter_test.rb
 - test/adapters/postgres_adapter_test.rb
@@ -77,7 +79,6 @@ metadata:
   homepage_uri: https://github.com/cheddar-me/pecorino
   source_code_uri: https://github.com/cheddar-me/pecorino
   changelog_uri: https://github.com/cheddar-me/pecorino/CHANGELOG.md
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -92,8 +93,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.6
-signing_key:
+rubygems_version: 3.6.2
 specification_version: 4
 summary: Database-based rate limiter using leaky buckets
 test_files: []